arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.14101cs.CVcs.CLcs.LG

用于联邦类增量学习的文本增强无数据方法

Text-Enhanced Data-free Approach for Federated Class-Incremental Learning

Monash University(莫纳什大学);University of Melbourne(墨尔本大学);VinAI Research(VinAI研究院)

Minh-Tuan Tran, Trung Le, Xuan-May Le, Mehrtash Harandi, Dinh Phung

AI总结:

本文提出LANDER方法,通过预训练语言模型生成的标签文本嵌入作为锚点,在训练阶段约束特征嵌入,在无数据知识迁移阶段合成高质量样本,有效解决联邦类增量学习中的灾难性遗忘问题,并在多个数据集上达到最先进性能。

摘要:

联邦类增量学习(FCIL)是一个研究不足但至关重要的问题,涉及在联邦学习背景下动态添加新类。在该领域,无数据知识迁移(DFKT)在解决灾难性遗忘和数据隐私问题中发挥着关键作用。然而,先前的方法缺乏DFKT与模型训练阶段之间关键的协同作用,导致DFKT在从旧任务模型的非锚定潜在空间生成高质量数据时遇到困难。在本文中,我们引入了LANDER(以标签文本为中心的无数据知识迁移),通过利用预训练语言模型生成的标签文本嵌入(LTE)来解决这一问题。具体而言,在模型训练阶段,我们的方法将LTE作为锚点,并将相应训练样本的特征嵌入约束在其周围,用更多有意义的信息丰富周围区域。在DFKT阶段,通过使用这些LTE锚点,LANDER能够合成更有意义的样本,从而有效解决遗忘问题。此外,引入了边界损失,而不是将嵌入严格约束向锚点,以鼓励样本嵌入在定义的半径内保持灵活性。这种方法保留了样本嵌入的自然差异,并缓解了由异构联邦设置引起的嵌入重叠问题。在CIFAR100、Tiny-ImageNet和ImageNet上进行的广泛实验表明,LANDER显著优于先前的方法,并在FCIL中实现了最先进的性能。代码可在https://github.com/tmtuan1307/lander获取。

英文摘要:

Federated Class-Incremental Learning (FCIL) is an underexplored yet pivotal issue, involving the dynamic addition of new classes in the context of federated learning. In this field, Data-Free Knowledge Transfer (DFKT) plays a crucial role in addressing catastrophic forgetting and data privacy problems. However, prior approaches lack the crucial synergy between DFKT and the model training phases, causing DFKT to encounter difficulties in generating high-quality data from a non-anchored latent space of the old task model. In this paper, we introduce LANDER (Label Text Centered Data-Free Knowledge Transfer) to address this issue by utilizing label text embeddings (LTE) produced by pretrained language models. Specifically, during the model training phase, our approach treats LTE as anchor points and constrains the feature embeddings of corresponding training samples around them, enriching the surrounding area with more meaningful information. In the DFKT phase, by using these LTE anchors, LANDER can synthesize more meaningful samples, thereby effectively addressing the forgetting problem. Additionally, instead of tightly constraining embeddings toward the anchor, the Bounding Loss is introduced to encourage sample embeddings to remain flexible within a defined radius. This approach preserves the natural differences in sample embeddings and mitigates the embedding overlap caused by heterogeneous federated settings. Extensive experiments conducted on CIFAR100, Tiny-ImageNet, and ImageNet demonstrate that LANDER significantly outperforms previous methods and achieves state-of-the-art performance in FCIL. The code is available at https://github.com/tmtuan1307/lander.

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

↑