FedUP: One-Shot Federated Unlearning via Centroid-Guided Plug-in Filters


Quan Wen\(^{1}\) Ji Wang\(^{1}\)
1
\(^1\)National Key Laboratory of Big Data and Decision, National University of Defense Technology, China.
feihongnan178@outlook.com, {zhongzhengyi20, wangpan19, wdbao, zhangxiongtao14, wangji}nudt.edu.cn?, weixingw1@sina.com


Abstract

Federated unlearning (FU) is critical for complying with legal mandates like the right to be forgotten in decentralized systems, yet current methods face a persistent dilemma between non-target knowledge loss and high request latency. To resolve these issues, we propose FedUP, a one-shot federated unlearning framework utilizing lightweight pluggable filters that act as a “knowledge funnel" to screen out target data while preserving original model performance. By freezing original model parameters and training filters at the server side using differentially private (DP)-protected class centroid samples, FedUP bypasses the need for multi-round client-server communication and complex retraining, reducing unlearning latency from minutes to mere seconds. Additionally, the framework’s pluggable architecture ensures inherent reversibility, enabling the seamless restoration of forgotten knowledge by simply removing the filters. Extensive experiments on diverse image and text tasks demonstrate that FedUP effectively reduces non-target knowledge loss and achieves superior unlearning precision and efficiency across various scenarios. Code is available at: https://github.com/suows/FedUP-code.

1 Introduction↩︎

Background. As a distributed machine learning paradigm, federated learning (FL) [1][7] has gained significant attention in privacy-sensitive scenarios recently because it eliminates the need for centralizing raw data during training. However, during the training process, the global model internalizes client information into its parameters through multiple rounds of parameter aggregation, which exposes novel privacy risks when the model is confronted with legal mandates such as the right to be forgotten under GDPR [8]. To address this, researchers have focused on Federated Unlearning (FU), aiming to remove specific knowledge without retraining the global model from scratch. Existing FU methods generally follow two main technical paradigms: server-side FU methods [9][11], which implement approximate unlearning [12] on the global model at the server side; and client-side FU methods [13][16], which achieve exact unlearning [17] through iterative client-side model retraining procedures.

a
b

Figure 1: FedEraser, a representative server-side method, shows noticeable non-target knowledge loss after unlearning across multiple tasks. Client-side methods like FUSED experience significantly longer convergence times as task difficulty increases, with responses up to 5 minutes indicating severe unlearning request latency.. a — Non-target knowledge loss of server-side FU method., b — Unlearning request latency of client-side FU method.

Existing Challenges. While current Federated Unlearning (FU) methods facilitate knowledge removal to various extents, challenges regarding non-target knowledge loss and unlearning request latency still remain. As depicted in Figure 1, server-side FU methods lack precise control over the unlearning scope, frequently incurring non-target knowledge loss, where the model performance unrelated to the unlearning request is inevitably impaired during knowledge removal. Conversely, although client-side FU methods achieve exact unlearning via retraining-based approaches, they are heavily constrained by multi-round training and communication, leading to high request latency. To date, it is hard to find a solution that simultaneously ensures rapid response and prevents non-target knowledge loss. Moreover, both server-side and client-side FU paradigms typically rely on direct modification of original model parameters, which leads to the irreversibility of unlearning. Once the unlearning process is finalized, restoring previously removed knowledge necessitates further parameter tuning or retraining, thereby imposing complexity and additional overhead on practical deployment.

Proposed Solution. To this end, we propose FedUP, a one-shot federated unlearning framework based on lightweight pluggable filters. In terms of mitigating non-target knowledge loss, the framework avoids performing direct, large-scale parameter updates on the global model; instead, it implements unlearning by introducing independent pluggable filters while keeping original model parameters frozen (shown in Figure 2). These filters serve as a “knowledge funnel" that screens out target knowledge and permits only non-target knowledge to pass, thereby reducing the interference of unlearning on non-target knowledge. To lower unlearning request latency, FedUP only needs to perform a few rounds of fine-tuning on the lightweight filters at the server side using differential privacy (DP)-protected class centroid samples to complete the unlearning task. This bypasses multi-round retraining and frequent communication with clients, significantly accelerating the response speed. Moreover, since the filters are pluggable, the framework inherently supports reversibility. As depicted in Figure 2, when restoration of forgotten knowledge is required, it can be achieved simply by removing the filters. Overall, FedUP provides a solution that balances precision, efficiency, and recoverability.

Figure 2: Lightweight plug-in filters.

Contributions. The main contributions are as follows:

  • We design FedUP, a one-shot FU framework utilizing DP-protected class centroids, mitigating non-target knowledge loss and reducing unlearning request latency from minutes to seconds.

  • We propose a reversible unlearning mechanism via lightweight pluggable filters without altering original model parameters while ensuring rapid transitions between unlearned and pre-unlearning states.

  • We conduct differential privacy analysis showing that appropriate noise protects class centroid samples while minimally impacting data utility. Extensive experiments on image and text tasks confirm FedUP’s effectiveness across various scenarios.

2 sec:Related32Work↩︎

Machine Unlearning. Simply removing training data from storage fails to purge its influence on deployed models. Machine Unlearning (MU) [18] is thus introduced to erase such learned knowledge. Existing methods are categorized as either exact [19] or approximate unlearning [20]. Exact unlearning mandates that the post-forgetting model be statistically indistinguishable from one retrained from scratch without the deleted data. It retains the guarantees of complete retraining but lowers its cost via algorithmic shortcuts. For classical models, the training process can be expressed in an invertible additive form, enabling point removal by subtracting its closed-form term [19]. For complex architectures, exact unlearning employs data sharding [21], localized retraining [22], or intermediate checkpointing [23] to reduce computation while preserving retraining-level guarantees. Approximate unlearning, in contrast, substitutes full retraining with lightweight fine-tuning, permitting a bounded, residual influence from the deleted data. Related work is typically grouped into data-driven and model-driven approaches [24]. Data-driven methods involve relabeling retained samples [25] or partitioning data shards [26] before fine-tuning. Model-driven approaches adjust parameters directly via influence functions [27], fisher-based regularization [20], or knowledge distillation [28], countering the gradient contributions of the data to be erased.

Federated Unlearning. Though federated learning maintains client privacy through local retention, knowledge from distributed datasets persists in the aggregated global model. This requires integrating MU techniques into FL, named federated unlearning (FU) [29]. Based on the execution location, FU methods can be categorized into server-side and client-side methods [30], [31]. Server-side methods are intrinsically based on approximate unlearning to expunge client contributions without client involvement. FedEraser [32] calibrates update trajectories but still requires auxiliary retraining. Conversely, Wu et al. [9] bypass client-side computation by subtracting historical updates and employing knowledge distillation. To optimize efficiency, Huynh et al. [10] utilize selective retention of influential updates to reduce memory overhead, while Pan et al. [11] resolve parameter conflicts via Orthogonal Steepest Descent to accelerate the unlearning process. Client-side FU methods perform unlearning locally, striving to achieve exact unlearning while balancing computational efficiency with global model utility. Mora et al. [33] propose FedUNRAN, utilizing local random label perturbations to disperse target gradients and attenuate their influence without server-side intervention. Wang et al. [13] employ variational Bayesian inference for parameter self-sharing to erase target data while preserving performance. To accelerate unlearning, Liu et al. [14] approximate the Hessian via a diagonal empirical Fisher Information Matrix for quasi-Newton optimization, while Zhu et al. [15] combine inverse perturbation with passive decay, propagating updates via knowledge distillation. Furthermore, Deng et al. [34] introduce model contrastive unlearning (MCU) to regularize the feature space. Zhong et al. [16] implement reversible unlearning through local fine-tuning.

In summary, existing unlearning approaches typically involve trade-offs between unlearning precision and computational efficiency, and often suffer from limited generalization. Moreover, reversibility is rarely considered in current studies. Achieving a unified balance among unlearning precision, efficiency, and reversibility therefore remains an open challenge.

3 sec:Methodology↩︎

3.1 Problem Description↩︎

In the FL framework, a set of clients denoted as \(\mathcal{C} = \{C_1, C_2, \ldots, C_N\}\) collaboratively train a global model \(\mathcal{M}_G\) without sharing local data. Each client \(C_n\) trains a local model \(\mathcal{M}_n\) using its local dataset \(\mathcal{D}_n\) and uploads the model parameters to a server. The server aggregates these local models via weighted averaging based on dataset sizes to obtain a global model \(\mathcal{M}_G\). This process iterates over \(W\) global rounds, where each global round comprises \(e\) local training epochs on the clients with a local learning rate \(l_c\). \(\mathcal{M}_G\) is structurally decomposed into a feature extractor \(\mathcal{M}_E\) and a classifier \(\mathcal{M}_{cl}\). After aggregation, \(\mathcal{M}_G\) is distributed back to all clients, where the feature extractor \(\mathcal{M}_E\) is leveraged to extract features from local data. Additionally, a pluggable filter \(\mathcal{M}_{Fi}\) is constructed to implement the filtering of knowledge that needs to be forgotten. The overall dataset is denoted as \(\mathcal{D} = \{\mathcal{D}_n\}_{n=1}^{N}\), where \(\mathcal{D}_n\) represents the local dataset of client \(C_n\). The total data volume across the federation is \(\| \mathcal{D} \|= \sum_{n=1}^{N} \| \mathcal{D}_n \|\). To facilitate data management, particularly for future unlearning operations, each local dataset \(\mathcal{D}_n\) is partitioned into two disjoint subsets: a retained subset \(\mathcal{D}_n^R\) for model training, and an unlearning subset \(\mathcal{D}_n^U\) designated to be removed in compliance with privacy or regulatory constraints. Let \(k \in \{1, 2, \ldots, K\}\) index the data categories, where \(K\) is the total number of classes.

In the FL phase, the training objective is defined as: \[\min_{\theta_{\mathcal{M}_G}} F(\theta_{\mathcal{M}_G}) = \sum_{n=1}^{N} \frac{|\mathcal{D}_n|}{|\mathcal{D}|} \sum_{(x_i^k, y_i^k) \in \mathcal{D}_n} \mathcal{L}\bigl(f(x_i^k; \theta_{\mathcal{M}_G}), y_i^k\bigr),\] where \(\mathcal{L}\) denotes the loss function. During the FU phase, the objective is to maximize the loss on the unlearning set, minimize the loss on the retained set, and keep the training cost low, described as: \[\min_{\theta_{\mathcal{M}_G'}} F^R(\theta_{\mathcal{M}_G'}) = \sum_{n=1}^{N} \frac{|\mathcal{D}_n^R|}{|\mathcal{D}^R|} \sum_{(x_i^k, y_i^k) \in \mathcal{D}_n^R} \mathcal{L}\bigl(f(x_i^k; \theta_{\mathcal{M}_G'}), y_i^k\bigr),\]

\[\max_{\theta_{\mathcal{M}_G'}} F^U(\theta_{\mathcal{M}_G'}) = \sum_{n=1}^{N} \frac{|\mathcal{D}_n^U|}{|\mathcal{D}^U|} \sum_{(x_i^k, y_i^k) \in \mathcal{D}_n^U} \mathcal{L}\bigl(f(x_i^k; \theta_{\mathcal{M}_G'}), y_i^k\bigr),\]

\[\min F^T(\theta_{\mathcal{M}_G'}) = \sum_{w=1}^{W} \text{Comm}^{(w)}(\mathcal{M}_G, \mathcal{C}\bigr),\] where Comm represents the communication resource between the server and the clients \(\mathcal{C}\) during each round.

3.2 Method Overview↩︎

As shown in Figure 3, our method comprises four phases: federated learning, generation of class centroid samples, one-shot federated unlearning, and inference, with each stage’s main procedures and key formulas shown in the diagram.

Figure 3: FedUP follows a structured workflow: it begins with federated learning to obtain a global model. Upon request, clients produce differentially‑private class centroid samples from retained data and upload them for server aggregation. The server then fine‑tunes a pluggable filter, blocking forgotten knowledge without modifying the base model.

3.2.1 Feature Extraction↩︎

Each client \(C_n\) performs \(e\) rounds of training on its local dataset \(\mathcal{D}_n\), resulting in an updated local model \(\mathcal{M}_n^{w,e}\). The local training process can be expressed as: \[\theta_{\mathcal{M}_n}^{w,e} = \theta_{\mathcal{M}_n}^{w,e-1} - l_c \cdot \nabla_{\theta} \mathcal{L}(\mathcal{M}_n^{w,e-1}, \mathcal{D}_n),\] where \(\mathcal{L}\) represents the local loss function and \(\nabla_{\theta} \mathcal{L}\) denotes its gradient with respect to the model parameters.Subsequently, the server collects the local models \(\mathcal{M}_n^{w,e}\) from selected clients \(C_n\) and updates the global model through a weighted average: \[\theta_{\mathcal{M}_G}^{w+1} = \sum_{n=1}^{N} \frac{|\mathcal{D}_n|}{|\mathcal{D}|} \theta_{\mathcal{M}_n}^{w,e}.\] The trained global model \(\mathcal{M}_G\) is regarded as a combination of a feature extractor \(\mathcal{M}_E\) and a classifier \(\mathcal{M}_{cl}\). After freezing this composite model, it is deployed to the clients. The features of all data \(\mathcal{D}_n\) from each client \(C_n\) are extracted by the feature extractor \(\mathcal{M}_E\), which can be represented as follows: \[V_n^k = \left\{ \mathcal{M}_E(x_i^k) \mid x_i^k \in \mathcal{D}_n^k \right\},\] where \(V_n^k\) denotes the feature embedding of the data belonging to class \(k\) on client \(C_n\).

3.2.2 Generation of Class Centroid Samples↩︎

Upon receiving an unlearning request, client \(C_n\) removes the features associated with the unlearning data. The remaining feature embeddings belonging to class \(k\) are denoted as: \[V_n^{R,k} = V_n^k \setminus \left\{ \mathcal{M}_E(x_i) \mid x_i \in \mathcal{D}_n^{U,k} \right\}.\]

To reduce communication cost, the retained features are compressed via KMeans clustering with \[K_{n,k} = \lceil \rho \, |V_n^{R,k}| \rceil,\] the number of clusters is set proportionally according to the scenario by \(\rho\), yielding a compact set of centroids: \[\mu_n^{R,k} = \mathrm{KMeans}(V_n^{R,k}, K_{n,k}).\]

For privacy preservation, add noise to each centroid: \[\label{noise} \tilde{\mu}_{n}^{R,k} = \left\{ \mu + \mathbf{z} \;\middle|\; \mu \in \mu_{n}^{R,k}, \; \mathbf{z} \sim \mathcal{N}(0,\sigma^{2}I) \right\}.\tag{1}\] Here, \(\mathcal{N}(0, \sigma^2 \mathbf{I})\) represents a multivariate Gaussian distribution. These mechanisms ensure that the release of \(\tilde{\mu}_n^{R,k}\) does not reveal excessive information about any individual data point. Client \(C_n\) then uploads all perturbed class centroids \(\tilde{\mu}_n^{R,k}\) to the server. Using the local class centroids \(\tilde{\mu}_n^{R,k}\), the server aggregates them to generate the global class centroids \(\tilde{\mu}_\mathcal{G}^{R,k}\). The global class centroid is denoted as: \[\tilde{\mu}_\mathcal{G}^{R,k}= \left[ \tilde{\mu}_1^{R,k}; \tilde{\mu}_2^{R,k}; \ldots; \tilde{\mu}_N^{R,k}\right].\]

3.2.3 One-shot Federated Unlearning↩︎

The objective of the FU phase is to train the filter \(\mathcal{M}_{Fi}\) such that it effectively blocks the flow of unlearning knowledge while allowing the retained learning knowledge to pass. The specific steps and formulas are as follows.

A filter \(\mathcal{M}_{Fi}\) is inserted into the global model \(\mathcal{M}_G=\mathcal{M}_E \oplus \mathcal{M}_{cl}\), yielding a new model structure: \[\mathcal{M}_G' = \mathcal{M}_E \oplus \mathcal{M}_{Fi} \oplus \mathcal{M}_{cl}.\]

Subsequently, the parameters of the global model \(\mathcal{M}_{G}\) including both \(\mathcal{M}_{E}\) and \(\mathcal{M}_{cl}\) are frozen: \[\theta_{\mathcal{M_G}} = \theta_{\mathcal{M_G}}^{*}.\]

The filter \(\mathcal{M}_{Fi}\) is trained using the global class centroids \(\tilde{\mu}_\mathcal{G}^{R,k}\). The filtered class centroid prediction is defined as: \[\{\hat{y_k}\} = \{\mathcal{M}_{\text{cl}}(\mathcal{M}_{\text{Fi}}(\tilde{\mu})) \mid \tilde{\mu} \in \tilde{\mu}_\mathcal{G}^{R,k}, k \in \mathcal{K}_R\}.\]

We propose a composite loss function for the filter consisting of cross-entropy and reconstruction losses. It preserves discriminative capability on non-target knowledge while enforcing structural consistency in the feature space, thereby improving the stability of the unlearning process without compromising knowledge selectivity. The cross-entropy loss, measuring the difference between predicted class probabilities and true labels, is defined as: \[\mathcal{L}_{\text{CE}} =-\sum_{k} y_k \log(\hat{y}_k),\] where \(y_k\) represents the true label while \(\hat{y}_k\) denotes the corresponding predicted probability for class \(k\). The reconstruction loss, which measures how well the filter reconstructs its input, is defined as: \[\mathcal{L}_{\text{RE}} = \frac{1}{d} \sum_{i=1}^{d} {|\tilde{\mu} - \mathcal{M}_{Fi}(\tilde{\mu})|^2 },\] where \(d\) is the input and output dimensionality of the filter. The total loss function used to train the filter \(\mathcal{M}_{Fi}\) is a weighted sum of these two losses: \[\label{filter32loss} \mathcal{L}_{\text{total}} = \alpha \mathcal{L}_{\text{CE}} + (1 - \alpha) \mathcal{L}_{\text{RE}},\tag{2}\] where \(\alpha\) is a weighting factor that balances the importance of the cross-entropy loss and the reconstruction loss. Train the filter for \(W_a\) rounds with a learning rate \(l_a\). The update process is as follows: \[\label{update32parameters} \theta_{\mathcal{M}_{Fi}}^w = \theta_{\mathcal{M}_{Fi}}^{w-1} - {l_a} \cdot \nabla_{\theta} \mathcal{L}_{\text{total}}\left( \theta_{\mathcal{M}_{Fi}}^{w-1}; \tilde{\mu}_\mathcal{G}^{R,k} \right) \quad w = 1, \dots, W_a.\tag{3}\]

When a client requests the restoration of forgotten knowledge, the filter \(\mathcal{M}_{Fi}\) can be removed, thereby reverting to the original global model \(\mathcal{M}_G\): \[\mathcal{M}_G = \mathcal{M}_E \oplus \mathcal{M}_{cl}.\]

3.2.4 Differential Privacy Guarantee↩︎

Our framework injects Gaussian noise for privacy guarantee during client-side centroid uploads, satisfying (\(\varepsilon,\delta\))-DP. The key is to calibrate the noise scale \(\sigma\) according to the desired privacy budget and the \(\ell_2\)-sensitivity, which is bounded by: \[\Delta_2 f_i = \max_{p,q} \frac{ \| d_p^{(i)} - d_q^{(i)} \|_{2} }{n_i}.\]

Table 1: Average privacy budget for different datasets.
\(\sigma\)
0.001 0.005 0.01 0.05 0.1
MNIST 11.09 2.22 1.11 0.22 0.11
CIFAR-10 16.73 3.35 1.67 0.33 0.17
AG News 12.27 2.45 1.23 0.25 0.12
CIFAR-100 141.40 28.28 14.14 2.83 1.41

9pt

We set \(\sigma \geq \frac{\sqrt{2\ln(1.25/\delta_i)} \cdot \Delta_2 f_i}{\varepsilon_i}\) with \(\delta_i = 1/n_i\). Empirical \(\varepsilon_i\) values (Table 1) confirm moderate privacy budgets \(\varepsilon \approx 10\) [35] for all datasets under specified \(\sigma\). We calculate our differentially private guarantee as:

\[\varepsilon_{i} = \frac{\sqrt{2 \ln(1.25 n_{i})} \cdot \Delta_{2} f_{i}}{ \sigma}.\] The final choice of the optimal \(\sigma\) is determined via \(\sigma = q \cdot s\), with the corresponding values of \(q\) and \(s\) established accordingly. Detailed analysis and supporting experiments are relegated to Section 2.1 and 2.2 of the supplementary material.

3.3 Algorithm↩︎

Figure 4: FedUP

The algorithm, as illustrated in Algorithm 4, consists of three stages: feature extraction, generation of class centroid samples \(\mu_n^{R,k}\) and one-shot federated unlearning. Upon receiving an unlearning request, each client first removes the corresponding data points from its local feature set and computes class centroid samples using the retained data. These centroids are then protected by a differential privacy mechanism and uploaded to the server. The server aggregates the uploaded centroids from all clients to obtain global retained class centroids as \(\tilde{\mu}_\mathcal{G}^{R,k}\). Then the server freezes the parameters of the original global model \(\mathcal{M}_\mathcal{G}\) and randomly initializes a lightweight, pluggable filter \(\mathcal{M}_{Fi}\), which is trained solely using the global differentially private centroids. By jointly optimizing a cross-entropy loss and a reconstruction loss \(\mathcal{L}_{\text{total}}\), the filter blocks the propagation of forgotten knowledge while preserving the discriminative capability of retained knowledge. Finally, when restoration of forgotten knowledge is required, the filter can be removed to revert to the original global model structure, enabling efficient and reversible federated unlearning.

4 Experiment↩︎

4.1 Experimental Setup↩︎

We conducted experiments on diverse datasets, including MNIST[36], CIFAR-10, CIFAR-100 [37], and AG News[38]. We partitioned the dataset using the Dirichlet distribution [39] with a concentration parameter of 0.5. These experiments use various network architectures, including LeNet-5, ResNet-18 [40], ResNet-34, Transformer [41] and TinyBert[42], covering three unlearning scenarios: client unlearning, class unlearning and sample unlearning. We evaluated five baselines: EraseClient [43], Federaser [32], Exact-Fun [44], Retrain and FUSED [16]. The experimental framework is implemented using PyTorch 2.3.1 and CUDA 12.1. For hardware acceleration, an NVIDIA RTX 3080 Ti GPU is utilized. We employ SGD and Adam optimizers. The balancing factor \(\alpha\) for the loss of filters is set to 0.5. During the generation of sample centroids via clustering, we set different sampling ratios \(\rho\) to accommodate various scenarios. In the client, class, and sample scenarios, the sampling ratios are 0.8, 0.1, and 0.5, respectively.

Table 2: Evaluation metrics of FU.
Metric Description
R-A(%)\(\uparrow\) Accuracy of retained knowledge.
F-A(%)\(\downarrow\) Accuracy of unlearning knowledge.
0A(%)\(\uparrow\) Accuracy of class 0.
PS(%)\(\uparrow\) Prediction precision of class 0.
MIA(%)\(\downarrow\) Post-unlearning privacy risk quantified via membership inference attacks.
Time(s)\(\downarrow\) Time required to achieve the desired effect.
Comm(MB)\(\downarrow\) Communication resource consumption between client and server.

0pt 0pt 1pt

0pt 0pt 1pt

c c c c c c c c c c c c c c c c c c & & & &
& & E-C & Federaser & E-F & FUSED & Retrain & FedUP & FUSED & Retrain & FedUP & & FUSED & Retrain & FedUP
& R-A & 0.99 & 0.97 & 0.97 & 0.99 & 0.99 & 0.99 & 0.97 & 0.99 & 1.00 & 0A & 0.95 & 1.00 & 0.95
& F-A & 0.00 & 0.00 & 0.00 & 0.00 & 0.00 & 0.00 & 0.00 & 0.00 & 0.00 & PS & 1.00 & 1.00 & 1.00
& MIA & 0.80 & 0.60 & 0.76 & 0.69 & 0.67 & 0.70 & 0.99 & 0.97 & 0.99 & MIA & 0.96 & 0.68 & 0.97
& R-A & 0.57 & 0.46 & 0.56 & 0.56 & 0.58 & 0.57 & 0.71 & 0.72 & 0.71 & 0A & 0.60 & 0.73 & 0.70
& F-A & 0.04 & 0.08 & 0.04 & 0.05 & 0.03 & 0.03 & 0.00 & 0.00 & 0.00 & PS & 0.52 & 0.60 & 0.53
& MIA & 0.67 & 0.70 & 0.62 & 0.68 & 0.57 & 0.63 & 0.81 & 0.63 & 0.78 & MIA & 0.95 & 0.98 & 0.96
& R-A & 0.48 & 0.51 & 0.50 & 0.49 & 0.52 & 0.52 & 0.59 & 0.60 & 0.59 & 0A & 0.35 & 0.64 & 0.57
& F-A & 0.05 & 0.05 & 0.05 & 0.05 & 0.04 & 0.04 & 0.00 & 0.00 & 0.00 & PS & 0.50 & 0.63 & 0.52
& MIA & 0.54 & 0.35 & 0.59 & 0.56 & 0.64 & 0.52 & 0.81 & 0.78 & 0.90 & MIA & 0.96 & 0.91 & 0.76
& R-A & 0.89 & 0.88 & 0.88 & 0.89 & 0.89 & 0.89 & 0.89 & 0.93 & 0.93 & 0A & 0.92 & 0.93 & 0.93
& F-A & 0.03 & 0.05 & 0.04 & 0.04 & 0.03 & 0.05 & 0.00 & 0.00 & 0.00 & PS & 0.87 & 0.89 & 0.88
& MIA & 0.63 & 0.63 & 0.79 & 0.68 & 0.68 & 0.77 & 0.44 & 0.63 & 0.50 & MIA & 0.77 & 0.54 & 0.70

& R-A & 0.13 & 0.15 & 0.27 & 0.26 & 0.27 & 0.26 & 0.36 & 0.37 & 0.37 & 0A & 0.50 & 0.57 & 0.55
& F-A & 0.01 & 0.01 & 0.01 & 0.01 & 0.01 & 0.02 & 0.00 & 0.00 & 0.00 & PS & 0.55 & 0.56 & 0.54
& MIA & 0.32 & 0.49 & 0.43 & 0.40 & 0.26 & 0.69 & 0.91 & 0.38 & 0.67 & MIA & 0.98 & 0.98 & 0.99

Evaluation Metrics. As shown in Table 2, we evaluate our proposed method and the baselines using multiple metrics.

4.2 sec:Experimental32Results↩︎

Main Results. In the client unlearning setting, Byzantine attacks are introduced, where label flipping is applied to construct inverse feature prototypes [45], [46]. Class unlearning randomly remaps the feature labels of the target class to several classes. Sample unlearning is implemented via backdoor attacks: fixed triggers are injected into the bottom-right pixels of images or appended as specific trigger tokens to the end of text sequences. During the unlearning process, all triggered samples are consistently predicted as class 0, thereby yielding accuracy of class 0 (0A) on the forgotten samples.

As shown in Table ¿tbl:tab:results?, our method achieves superior performance across a wide range of datasets, model architectures, and unlearning scenarios. Among them, CIFAR-100 is the most challenging benchmark, while MNIST is the least. Benefiting from the pluggable filters, FedUP reduces the accuracy on the unlearning knowledge to a minimal level (F-A) while maintaining high accuracy on the retained data (R-A). Owing to its centroid-based unlearning mechanism, FedUP exhibits particularly strong performance in class unlearning scenarios. Overall, the proposed method supports single-round communication and reversible unlearning, achieving the desirable triad of high retention accuracy, low forgetting accuracy, and privacy guarantees across all evaluated datasets.

Analysis of Non-target Knowledge Loss. To validate the effectiveness of our method in mitigating non-target knowledge loss, we compare the accuracy on retained knowledge before and after executing the unlearning operation. As shown in Figure 5, FedUP achieves an effect comparable to the Retrain method, characterized by minimal non-target knowledge loss and stable model performance before and after the unlearning operation.

Figure 5: Non-target knowledge loss of methods.

Unlearning Response Time. As shown in Table 3, when achieving the desired forgetting effect, our method requires only 6.33 seconds for image unlearning and 6.45 seconds for text unlearning, which is less than 10% of the latency of the second-best method and less than 1% of that of the Retrain approach. It can promptly respond to diverse unlearning requests across different modalities while maintaining consistently low latency, effectively minimizing processing delays.

Communication Cost. Communication cost is defined as the wireless resources consumed for transmitting model parameters or gradients between local clients and the central server. As shown in Table 3, although the centroid uploading phase introduces additional communication overhead, FedUP requires only a single communication round, meaning its communication cost does not accumulate with rounds. In contrast, model transmission methods gradually converge as rounds increase, causing the communication overhead to escalate to the order of \(10^3\). Specifically, the lightweight filter incurs an overhead of merely 11.50MB for image unlearning, approximately one-third of that of the second-best method and 6.73MB for text unlearning, which is second only to the FUSED method.

Table 3: Comparing time and communication costs.
CIFAR10 AG News
Time Comm Time Comm
E-C 1627.76 1110.98 567.56 268.64
Federaser 1586.59 2820.18 634.67 738.76
E-F 898.70 1452.82 629.83 369.38
FUSED 151.21 31.36 88.31 0.55
Retrain 935.54 1538.25 859.28 537.28
FedUP 6.33 11.50 6.45 6.73

0pt 0pt 1pt

Figure 6: Ablation of loss functions.

4.3 Analysis of Hyper-parameters↩︎

Bottleneck Dimension. In this section, we analyse the dimension of the filter. The pluggable filter adopts a straightforward encoder-decoder architecture. Its input dimension is configured to match the output dimension of the feature extractor. Our empirical investigation focuses on determining the optimal bottleneck dimension within this encoder-decoder structure. As shown in Table 4, the remember accuracy consistently reaches its optimum across both image and text datasets when the bottleneck dimension is set to 32.

Table 4: Exploration of filter. “M" stands for memory (KB).
4 8 16 32 64
R-A M R-A M R-A M R-A M R-A M
MNIST 0.98 16 0.99 32 0.99 64 0.99 128 0.99 256
CIFAR10 0.55 16 0.58 32 0.58 64 0.60 128 0.59 256
CIFAR10-T 0.43 16 0.47 32 0.46 64 0.51 128 0.46 256
AG News 0.89 4 0.90 8 0.90 16 0.91 32 0.90 64
CIFAR100 0.06 16 0.18 32 0.24 64 0.25 128 0.23 256

0pt 0pt 1pt

Sensitivity Analysis of the Loss Function. The filter is trained with joint reconstruction and cross-entropy losses. We performed a sensitivity analysis on \(\alpha\) of \(\mathcal{L}_{\text{total}}\) to determine its optimal value. As shown in Figure 6, using either loss alone hinders convergence and degrades the accuracy of retained knowledge, whereas their combination maximizes the reduction of non-target knowledge loss. This is further illustrated in Table 5, which shows that increasing the weight of the cross-entropy loss lowers the recognition accuracy on retained data, while assigning a higher weight to the reconstruction loss deteriorates the unlearning effectiveness.

Table 5: Impact of \(\alpha\)-CE on R-A and F-A. The filter shows optimal and well-balenced performance at \(\alpha = 0.5\).
\(\alpha\)-CE 0.10 0.30 0.50 0.70 0.90
metric R-A F-A R-A F-A R-A F-A R-A F-A R-A F-A
MNIST 0.99 0.67 1.00 0.00 0.99 0.00 0.99 0.00 0.99 0.00
CIFAR10 0.70 0.00 0.68 0.00 0.71 0.00 0.68 0.00 0.64 0.00
CIFAR10-T 0.51 0.22 0.56 0.00 0.59 0.00 0.51 0.00 0.45 0.00
AG News 0.92 0.00 0.93 0.00 0.93 0.00 0.92 0.00 0.87 0.00
CIFAR100 0.35 0.11 0.37 0.00 0.37 0.00 0.31 0.00 0.30 0.00

0pt 0pt 1pt

4.4 Ablation Study↩︎

Analysis of Differential Privacy Effects. To ensure privacy preservation, Gaussian noise is injected into the generated class centroids. We conducted ablation studies to evaluate its impact, as shown in Table 6. The results demonstrate that an appropriately calibrated noise scale does not significantly impede the model’s performance on non-target knowledge. Instead, it enhances model robustness, achieving a synergistic improvement in both privacy protection and model utility.

Table 6: Differential privacy effects.
CIFAR10 AG News
DP w/o DP DP w/o DP
R-A 0.5814 0.5833 0.9132 0.9114
F-A 0.0438 0.0460 0.0490 0.0303
MIA 0.5239 0.6373 0.5654 0.6196

0pt 0pt 1pt

Table 7: Ablation of class centroid samples.
R-A F-A
Feature Centroid Feature Centroid
MNIST 0.9900 0.9921 0.0001 0.0005
CIFAR10 0.6011 0.6020 0.0498 0.0331
CIFAR10-T 0.5264 0.5671 0.0632 0.0490
AG News 0.9153 0.9118 0.0331 0.0482
CIFAR100 0.2537 0.2508 0.0076 0.0172

0pt 0pt 1pt

Effectiveness of Class Centroid Samples. Class centroids are obtained by clustering original features and privately aggregated on the server side. To validate the effectiveness of class centroid samples, we additionally conduct experiments in which the filter is trained using only the original features. As illustrated in Table 7, the model trained on class centroid samples achieves comparable performance on the accuracy of retained and unlearning knowledge to that trained on the original features, demonstrating that class centroid samples are as effective as the original features.

5 Conclusion and Discussion↩︎

Conclusion. In the field of FU, server-side methods face non-target knowledge loss, whereas client‑side methods incur high request latency. Both are limited by the irreversibility of the unlearning operation. To address these issues, this paper proposes a one-shot federated unlearning framework. By employing differentially private class centroid samples on the server, our approach achieves approximate unlearning that surpasses exact unlearning in effect, reducing non‑target knowledge loss and high resource overhead. Through fine-tuning a lightweight plug-in filter in a single round, the desired unlearning effect is achieved, significantly reducing the latency of unlearning responses. Removing the filter allows the model to revert to its pre-unlearning state, thus realizing the reversibility of unlearning. Extensive experiments across diverse datasets, scenarios, and models demonstrate that FedUP demonstrates excellent performance.

Discussion. While FedUP advances unlearning efficiency, reversibility, and responsiveness, two fundamental limitations still persist: Class centroid fidelity depends on federated feature extraction. Low-quality global models propagate bias into aggregated class centroids. It is expected that these limitations can be effectively addressed by utilizing more powerful pre-trained backbones for local feature extraction to ensure reliable class centroids.

5.0.1 Contribution Statement↩︎

Feihong Nan and Zhengyi Zhong contributed equally.

References↩︎

[1]
H. B. McMahan, E. Moore, D. Ramage, S. Hampson, and B. Arcas, “Communication-efficient learning of deep networks from decentralized data,” arXiv: Learning,arXiv: Learning, Feb. 2016.
[2]
N. Truong, K. Sun, S. Wang, F. Guitton, and Y. Guo, “Privacy preservation in federated learning: An insightful survey from the GDPR perspective,” Computers & Security, vol. 110, p. 102402, 2021.
[3]
Z. Zhong, W. Bao, J. Wang, X. Zhu, and X. Zhang, “Flee: A hierarchical federated learning framework for distributed deep neural network over cloud, edge, and end device,” ACM Transactions on Intelligent Systems and Technology (TIST), vol. 13, no. 5, pp. 1–24, 2022.
[4]
Z. Qi et al., “Cross-training with multi-view knowledge fusion for heterogenous federated learning,” arXiv e-prints, pp. arXiv–2405, 2024.
[5]
Z. Zhong, W. Bao, J. Wang, J. Chen, L. Lyu, and W. Y. B. Lim, “Sacfl: Self-adaptive federated continual learning for resource-constrained end devices,” IEEE Transactions on Neural Networks and Learning Systems, 2025.
[6]
L. Fu, S. Huang, Y. Li, C. Chen, C. Zhang, and Z. Zheng, “Learn the global prompt in the low-rank tensor space for heterogeneous federated learning,” Neural Networks, vol. 187, p. 107319, 2025.
[7]
W. Jiang et al., “Unveiling and mitigating untargeted poisoning attacks on federated knowledge graph embedding,” in Proceedings of the ACM web conference 2026, 2026, pp. 2569–2580.
[8]
L. de la Torre, “A guide to the california consumer privacy act of 2018,” SSRN Electronic Journal, Dec. 2018, doi: 10.2139/ssrn.3275571.
[9]
C. Wu, S. Zhu, and P. Mitra, “Federated unlearning with knowledge distillation,” arXiv preprint arXiv:2201.09441, 2022.
[10]
T. T. Huynh et al., “Certified unlearning for federated recommendation,” ACM Transactions on Information Systems, vol. 43, no. 2, pp. 1–29, 2025.
[11]
Z. Pan et al., “Federated unlearning with gradient descent and conflict mitigation,” in Proceedings of the AAAI conference on artificial intelligence, 2025, vol. 39, pp. 19804–19812.
[12]
Z.-R. Yang, J. Han, C.-D. Wang, and H. Liu, “Erase then rectify: A training-free parameter editing approach for cost-effective graph unlearning,” in Proceedings of the AAAI conference on artificial intelligence, 2025, vol. 39, pp. 13044–13051.
[13]
W. Wang, Z. Tian, C. Zhang, A. Liu, and S. Yu, “Bfu: Bayesian federated unlearning with parameter self-sharing,” in Proceedings of the 2023 ACM asia conference on computer and communications security, 2023, pp. 567–578.
[14]
Y. Liu, L. Xu, X. Yuan, C. Wang, and B. Li, “The right to be forgotten in federated learning: An efficient realization with rapid retraining,” in IEEE INFOCOM 2022-IEEE conference on computer communications, 2022, pp. 1749–1758.
[15]
X. Zhu, G. Li, and W. Hu, “Heterogeneous federated knowledge graph embedding learning and unlearning,” in Proceedings of the ACM web conference 2023, 2023, pp. 2444–2454.
[16]
Z. Zhong et al., “Unlearning through knowledge overwriting: Reversible federated unlearning via selective sparse adapter,” in Proceedings of the computer vision and pattern recognition conference, 2025, pp. 30661–30670.
[17]
K. Kuo, A. Setlur, K. Srinivas, A. Raghunathan, and V. Smith, “Exact unlearning of finetuning data via model merging at scale,” arXiv preprint arXiv:2504.04626, 2025.
[18]
Z. Ma, Y. Liu, X. Liu, J. Liu, J. Ma, and K. Ren, “Learn to forget: Machine unlearning via neuron masking,” IEEE Transactions on Dependable and Secure Computing, vol. 20, no. 4, pp. 3194–3207, 2022.
[19]
Y. Cao, A. F. Yu, A. Aday, E. Stahl, J. Merwine, and J. Yang, “Efficient repair of polluted machine learning systems via causal unlearning,” in ASIACCS ’18: Proceedings of the 2018 on asia conference on computer and communications security, 2018.
[20]
A. Golatkar, A. Achille, and S. Soatto, “Eternal sunshine of the spotless net: Selective forgetting in deep networks,” in Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, 2020, pp. 9304–9312.
[21]
L. Bourtoule et al., “Machine unlearning,” in 2021 IEEE symposium on security and privacy (SP), 2021, pp. 141–159.
[22]
M. Chen, Z. Zhang, T. Wang, M. Backes, M. Humbert, and Y. Zhang, “Graph unlearning,” in Proceedings of the 2022 ACM SIGSAC conference on computer and communications security, 2022, pp. 499–513.
[23]
C.-L. Wang, M. Huai, and D. Wang, “Inductive graph unlearning,” in 32nd USENIX security symposium (USENIX security 23), 2023, pp. 3205–3222.
[24]
T. T. Nguyen et al., “A survey of machine unlearning,” ACM Transactions on Intelligent Systems and Technology, vol. 16, no. 5, pp. 1–46, 2025.
[25]
L. Graves, V. Nagisetty, and V. Ganesh, “Amnesiac machine learning,” in Proceedings of the AAAI conference on artificial intelligence, 2021, vol. 35, pp. 11516–11524.
[26]
V. Gupta, C. Jung, S. Neel, A. Roth, S. Sharifi-Malvajerdi, and C. Waites, “Adaptive machine unlearning,” Advances in Neural Information Processing Systems, vol. 34, pp. 16319–16330, 2021.
[27]
C. Guo, T. Goldstein, A. Hannun, and L. Van Der Maaten, “Certified data removal from machine learning models,” arXiv preprint arXiv:1911.03030, 2019.
[28]
M. Kurmanji, P. Triantafillou, J. Hayes, and E. Triantafillou, “Towards unbounded machine unlearning,” Advances in neural information processing systems, vol. 36, pp. 1957–1987, 2023.
[29]
J. Wang, S. Guo, X. Xie, and H. Qi, “Federated unlearning via class-discriminative pruning,” in Proceedings of the ACM web conference 2022, 2022, pp. 622–632.
[30]
Z. Liu et al., “A survey on federated unlearning: Challenges, methods, and future directions,” ACM Computing Surveys, vol. 57, no. 1, pp. 1–38, 2024.
[31]
N. Li et al., “Machine unlearning: Taxonomy, metrics, applications, challenges, and prospects,” IEEE Transactions on Neural Networks and Learning Systems, 2025.
[32]
G. Liu, X. Ma, Y. Yang, C. Wang, and J. Liu, “Federaser: Enabling efficient client-level data removal from federated learning models,” in 2021 IEEE/ACM 29th international symposium on quality of service (IWQOS), 2021, pp. 1–10.
[33]
A. Mora, L. Dominici, and P. Bellavista, “Fedunran: On-device federated unlearning via random labels,” in 2024 IEEE international conference on big data (BigData), 2024, pp. 7955–7960.
[34]
Z. Deng, L. Luo, and H. Chen, “Enable the right to be forgotten with federated client unlearning in medical imaging,” in International conference on medical image computing and computer-assisted intervention, 2024, pp. 240–250.
[35]
K. Wei et al., “Federated learning with differential privacy: Algorithms and performance analysis,” IEEE transactions on information forensics and security, vol. 15, pp. 3454–3469, 2020.
[36]
Y. LeCun, L. Bottou, Y. Bengio, and P. Haffner, “Gradient-based learning applied to document recognition,” Proceedings of the IEEE, vol. 86, no. 11, pp. 2278–2324, 2002.
[37]
A. Krizhevsky, G. Hinton, et al., “Learning multiple layers of features from tiny images,” 2009.
[38]
X. Zhang, J. Zhao, and Y. LeCun, “Character-level convolutional networks for text classification,” Advances in neural information processing systems, vol. 28, 2015.
[39]
Q. Li, Y. Diao, Q. Chen, and B. He, “Federated learning on non-iid data silos: An experimental study,” in 2022 IEEE 38th international conference on data engineering (ICDE), 2022, pp. 965–978.
[40]
K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2016, pp. 770–778.
[41]
A. Vaswani et al., “Attention is all you need,” Advances in neural information processing systems, vol. 30, 2017.
[42]
X. Jiao et al., “Tinybert: Distilling bert for natural language understanding,” in Findings of the association for computational linguistics: EMNLP 2020, 2020, pp. 4163–4174.
[43]
A. Halimi, S. Kadhe, A. Rawat, and N. Baracaldo, “Federated unlearning: How to efficiently erase a client in FL?” Jul. 2022.
[44]
Z. Xiong, W. Li, Y. Li, and Z. Cai, “Exact-fun: An exact and efficient federated unlearning approach,” in 2023 IEEE international conference on data mining (ICDM), 2023, pp. 1439–1444.
[45]
L. Fu et al., “Federated domain-independent prototype learning with alignments of representation and parameter spaces for feature shift,” IEEE Transactions on Mobile Computing, 2025.
[46]
Z. Qi, L. Meng, Z. Chen, H. Hu, H. Lin, and X. Meng, “Cross-silo prototypical calibration for federated learning with non-iid data,” in Proceedings of the 31st ACM international conference on multimedia, 2023, pp. 3099–3107.

  1. Corresponding Author↩︎