arXiv:2412.18355cs.CVcs.AI2024-12CVPR被引 14

解决联邦持续学习中时空数据异构导致的遗忘问题。

Handling Spatial-Temporal Data Heterogeneity for Federated Continual Learning via Tail Anchor

  • 用可训练尾锚点调整特征位置,缓解参数和输出遗忘。
  • 在多个数据集上相比现有方法提升10%以上准确率。
  • 适合有持续学习需求的分布式场景,如智能医疗、边缘设备。

联邦持续学习(FCL)使客户端能从任务流中持续更新知识,提升了联邦学习在真实场景中的适用性。然而,FCL需同时应对客户端间的空间数据异构以及任务间的时序数据异构。实验证明,这种输入级异构会显著影响模型内部参数与输出,导致本地及历史知识严重发生时空灾难性遗忘。为此,本文提出联邦尾锚机制(FedTA),通过可训练尾锚与冻结输出特征混合,调整其在特征空间中的位置,从而缓解参数遗忘与输出遗忘。设计了三个新组件:输入增强以提升预训练模型在下游任务的表现;选择性输入知识融合用于服务器端聚合异构本地知识;最优全局原型选择用于在特征空间中为每类寻找最佳锚点。大量实验表明,FedTA不仅优于现有FCL方法,还能有效保持特征间的相对位置关系。

原文摘要 · Abstract (English)

Federated continual learning (FCL) allows each client to continually update its knowledge from task streams, enhancing the applicability of federated learning in real-world scenarios. However, FCL needs to address not only spatial data heterogeneity between clients but also temporal data heterogeneity between tasks. In this paper, empirical experiments demonstrate that such input-level heterogeneity significantly affects the model's internal parameters and outputs, leading to severe spatial-temporal catastrophic forgetting of local and previous knowledge. To this end, we propose Federated Tail Anchor (FedTA) to mix trainable Tail Anchor with the frozen output features to adjust their position in the feature space, thereby overcoming parameter-forgetting and output-forgetting. Three novel components are also included: Input Enhancement for improving the performance of pre-trained models on downstream tasks; Selective Input Knowledge Fusion for fusion of heterogeneous local knowledge on the server; and Best Global Prototype Selection for finding the best anchor point for each class in the feature space. Extensive experiments demonstrate that FedTA not only outperforms existing FCL methods but also effectively preserves the relative positions of features.

联邦学习持续学习特征对齐灾难性遗忘

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