FedVaccine: Knowledge Recall after Spatial-Temporal Catastrophic Forgetting in Federated Continual Learning via Gradient-Based Vaccine
Abstract
Federated Continual Learning (FCL) suffers from spatial-temporal catastrophic forgetting caused by sequential task learning on individual clients and the aggregation of heterogeneous client knowledge. However, most existing methods either overemphasize preserving old knowledge, which restricts model plasticity for new tasks, or rely on training complex generative models for replay, incurring substantial computational overhead. In the human immune system, immune memory is preserved implicitly and can be reactivated through vaccination even after long periods of dormancy. Motivated by this, we propose FedVaccine, a novel framework that explicitly permits forgetting to preserve high plasticity for new tasks, while efficiently recovering past knowledge through re-learning a compact vaccine set. Specifically, the vaccine set for each task is synthesized by compress task gradient information under the guidance of the fine-tuned local model, capturing representative data characteristics and task-specific discriminative patterns. Two novel components are also introduced: Priming & Boosting, which encodes task-specific knowledge in a compact spectral form, priming the model with essential information and enabling efficient memory boosting via vaccine set replay after forgetting; and Vaccination Aggregation, which aggregates vaccine sets from all clients to train a globally generalized model without accessing any raw private data. Extensive experiments on three benchmarks demonstrate that FedVaccine achieves competitive performance, enables rapid knowledge recovery after forgetting, and incurs no additional overhead from training generative models.