arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2412.18355cs.CVcs.AIcs.LG

通过 Tail Anchor 处理联邦持续学习中的时空数据异构性

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

Southwestern University of Finance and Economics(西南财经大学);University of Electronic Science and Technology of China(电子科技大学);WeBank(微众银行);Southwest Jiaotong University(西南交通大学);Hong Kong University of Science and Technology(香港科技大学)

Hao Yu, Xin Yang, Le Zhang, Hanlin Gu, Tianrui Li, Lixin Fan, Qiang Yang

AI总结:

针对联邦持续学习中的时空数据异构性导致的灾难性遗忘问题,提出FedTA方法,通过混合可训练Tail Anchor与冻结特征调整特征空间位置,并引入输入增强、选择性知识融合和全局原型选择组件,有效克服参数与输出遗忘并优于现有方法。

摘要:

联邦持续学习(FCL)允许每个客户端从任务流中持续更新其知识,增强了联邦学习在现实场景中的适用性。然而,FCL不仅需要解决客户端之间的空间数据异构性,还需应对任务之间的时间数据异构性。本文的实证实验表明,这种输入级异构性显著影响模型的内部参数和输出,导致局部和先前知识的严重时空灾难性遗忘。为此,我们提出联邦尾部锚点(FedTA),将可训练的 Tail Anchor 与冻结的输出特征混合,调整其在特征空间中的位置,从而克服参数遗忘和输出遗忘。此外还包含三个新颖组件:用于提升预训练模型在下游任务性能的输入增强;用于在服务器上融合异构本地知识的选择性输入知识融合;以及用于在特征空间中为每个类别寻找最佳锚点的最佳全局原型选择。大量实验表明,FedTA不仅优于现有FCL方法,还能有效保留特征的相对位置。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑