arXiv:2409.01585cs.LGcs.DC2024-09被引 5

用本地缓冲区和梯度聚合缓解联邦持续学习中的遗忘问题。

Buffer-based Gradient Projection for Continual Federated Learning

  • 通过本地缓冲区和聚合梯度投影,防止模型遗忘旧知识。
  • 在CIFAR-100上任务增量场景下准确率提升最高达27%。
  • 适合资源受限、数据异构的分布式持续学习场景。

持续联邦学习(CFL)对于支持多客户端从连续数据流中自适应学习至关重要。其主要挑战是灾难性遗忘,即模型在学习新信息时会丢失已有知识。现有方法常受限于设备存储容量及客户端间数据分布的异质性。部分算法虽已解决这些问题,但通常依赖于对任务边界(即新任务开始时间)的不现实假设。为此,我们提出Fed-A-GEM,一种对A-GEM方法(Chaudhry et al., 2019)的联邦适配,采用基于缓冲区的梯度投影策略。该方法通过利用本地缓冲区样本与聚合缓冲区梯度,有效缓解灾难性遗忘,实现跨多个客户端的知识保留。我们的方法可与现有CFL技术结合,在标准基准测试中展现一致性能提升。例如,在使用CIFAR-100数据集的任务增量学习场景中,准确率最高可提升27%。代码已开源:https://github.com/shenghongdai/Fed-A-GEM。

原文摘要 · Abstract (English)

Continual Federated Learning (CFL) is essential for enabling real-world applications where multiple decentralized clients adaptively learn from continuous data streams. A significant challenge in CFL is mitigating catastrophic forgetting, where models lose previously acquired knowledge when learning new information. Existing approaches often face difficulties due to the constraints of device storage capacities and the heterogeneous nature of data distributions among clients. While some CFL algorithms have addressed these challenges, they frequently rely on unrealistic assumptions about the availability of task boundaries (i.e., knowing when new tasks begin). To address these limitations, we introduce Fed-A-GEM, a federated adaptation of the A-GEM method (Chaudhry et al., 2019), which employs a buffer-based gradient projection approach. Fed-A-GEM alleviates catastrophic forgetting by leveraging local buffer samples and aggregated buffer gradients, thus preserving knowledge across multiple clients. Our method is combined with existing CFL techniques, enhancing their performance in the CFL context. Our experiments on standard benchmarks show consistent performance improvements across diverse scenarios. For example, in a task-incremental learning scenario using the CIFAR-100 dataset, our method can increase the accuracy by up to 27%. Our code is available at https://github.com/shenghongdai/Fed-A-GEM.

联邦学习持续学习梯度投影缓冲区机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。