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

按类别平衡的数据回放用于联邦类增量学习

Class-wise Balancing Data Replay for Federated Class-Incremental Learning

School of Software, Shandong University(山东大学软件学院);College of Computing and Data Science, Nanyang Technological University(南洋理工大学计算机与数据科学学院);Department of Electrical and Computer Engineering, University of British Columbia(不列颠哥伦比亚大学电气与计算机工程系);Vector Institute(向量研究所)

Zhuang Qi, Ying-Peng Tang, Lei Meng, Han Yu, Xiaoxiao Li, Xiangxu Meng

AI总结:

FedCBDR通过全局协调机制和任务感知温度缩放模块,解决联邦类增量学习中的类别不平衡问题,提升模型在异构数据分布和任务不平衡下的泛化能力。

摘要:

联邦类增量学习(FCIL)旨在跨多个客户端协同处理持续增加的 incoming 任务。在各种方法中,数据回放已成为一种有前途的解决方案,可以通过重新引入先前任务的代表性样本来缓解遗忘。然而,其性能通常受到类别不平衡的限制,这既体现在回放缓冲区内的有限全局意识,也体现在回放类与新到达类之间的不平衡。为了解决这个问题,我们提出了一种用于FCIL的按类别平衡的数据回放方法(FedCBDR),该方法采用全局协调机制进行类别级内存构建,并重新加权学习目标以缓解上述不平衡。具体来说,FedCBDR有两个关键组成部分:1)全局视角的数据回放模块以隐私保护的方式重建先前任务的全局表示,然后引导一种具有类别意识和重要性敏感的采样策略以实现平衡回放;2)随后,为了处理跨任务的类别不平衡,任务感知的温度缩放模块根据任务动态自适应调整logits的温度,既在类别层面又在实例层面,从而减少模型对多数类的过度自信,同时增强其对少数类的敏感性。实验结果验证了FedCBDR在异构数据分布下实现了平衡的类别采样,并在早期和近期任务之间的任务不平衡情况下提高了泛化能力,相对于六个最先进的方法,Top-1准确率提高了2%-15%。

英文摘要:

Federated Class Incremental Learning (FCIL) aims to collaboratively process continuously increasing incoming tasks across multiple clients. Among various approaches, data replay has become a promising solution, which can alleviate forgetting by reintroducing representative samples from previous tasks. However, their performance is typically limited by class imbalance, both within the replay buffer due to limited global awareness and between replayed and newly arrived classes. To address this issue, we propose a class wise balancing data replay method for FCIL (FedCBDR), which employs a global coordination mechanism for class-level memory construction and reweights the learning objective to alleviate the aforementioned imbalances. Specifically, FedCBDR has two key components: 1) the global-perspective data replay module reconstructs global representations of prior task in a privacy-preserving manner, which then guides a class-aware and importance-sensitive sampling strategy to achieve balanced replay; 2) Subsequently, to handle class imbalance across tasks, the task aware temperature scaling module adaptively adjusts the temperature of logits at both class and instance levels based on task dynamics, which reduces the model's overconfidence in majority classes while enhancing its sensitivity to minority classes. Experimental results verified that FedCBDR achieves balanced class-wise sampling under heterogeneous data distributions and improves generalization under task imbalance between earlier and recent tasks, yielding a 2%-15% Top-1 accuracy improvement over six state-of-the-art methods.

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

↑