Diffusion-Driven Data Replay: A Novel Approach to Combat Forgetting in Federated Class Continual Learning
<|reference_start|>Diffusion-Driven Data Replay: A Novel Approach to Combat Forgetting in Federated Class Continual Learning: Federated Class Continual Learning (FCCL) merges the challenges of distributed client learning with the need for seamless adaptation to new classes without forgetting old ones. The key challenge in FCCL is catastrophic forgetting, an issue that has been explored to some extent in Continual Learning (CL). However, due to privacy preservation requirements, some conventional methods, such as experience replay, are not directly applicable to FCCL. Existing FCCL methods mitigate forgetting by generating historical data through federated training of GANs or data-free knowledge distillation. However, these approaches often suffer from unstable training of generators or low-quality generated data, limiting their guidance for the model. To address this challenge, we propose a novel method of data replay based on diffusion models. Instead of training a diffusion model, we employ a pre-trained conditional diffusion model to reverse-engineer each class, searching the corresponding input conditions for each class within the model's input space, significantly reducing computational resources and time consumption while ensuring effective generation. Furthermore, we enhance the classifier's domain generalization ability on generated and real data through contrastive learning, indirectly improving the representational capability of generated data for real data. Comprehensive experiments demonstrate that our method significantly outperforms existing baselines. Code is available at https://github.com/jinglin-liang/DDDR.<|reference_end|>
@article{liang2024diffusion-driven,
    title={Diffusion-Driven Data Replay: A Novel Approach to Combat Forgetting in
  Federated Class Continual Learning},
    author={Jinglin Liang, Jin Zhong, Hanlin Gu, Zhongqi Lu, Xingxing Tang, Gang
  Dai, Shuangping Huang, Lixin Fan and Qiang Yang},
    journal={arXiv preprint arXiv:2409.01128},
    year={2024},
    archivePrefix={arXiv},
    eprint={2409.01128},
    primaryClass={cs.LG cs.CV}
}