arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.02447cs.LG
收藏

联邦类增量学习与层次化生成原型

Federated Class-Incremental Learning with Hierarchical Generative Prototypes

AImageLab - University of Modena and Reggio Emilia(AImageLab - Modena和Reggio Emilia大学)

Riccardo Salami, Pietro Buzzega, Matteo Mosconi, Mattia Verasani, Simone Calderara

AI总结:

针对联邦持续学习中的增量偏差与联邦偏差问题,提出利用可学习提示微调预训练骨干并采用层次化生成原型平衡预测,平均准确率提升7.8%。

摘要:

联邦学习(FL)旨在通过将计算分布到多个设备(客户端)上来减轻深度模型训练的负担,同时保护数据隐私。在此基础上,联邦持续学习(FCL)还考虑了随时间演变的数据分布,反映了真实世界环境的动态特性。虽然先前的研究已将灾难性遗忘和客户端漂移确定为联邦持续学习中性能下降的主要原因,但我们揭示了增量偏差和联邦偏差的重要性,这两种偏差分别导致模型优先考虑最近引入的类别或本地占主导地位的类别。我们的方法通过在最后一层约束这两种偏差,利用可学习的提示高效微调预训练骨干网络,从而使客户端产生偏差较小的表示和偏差较大的分类器。因此,我们不单纯依赖参数聚合,而是利用生成原型有效平衡全局模型的预测。我们的方法显著提升了当前最先进的性能,平均准确率提高了+7.8%。

英文摘要:

Federated Learning (FL) aims at unburdening the training of deep models by distributing computation across multiple devices (clients) while safeguarding data privacy. On top of that, Federated Continual Learning (FCL) also accounts for data distribution evolving over time, mirroring the dynamic nature of real-world environments. While previous studies have identified Catastrophic Forgetting and Client Drift as major factors of performance degradation in FCL, we shed light on the importance of Incremental Bias and Federated Bias, which cause models to prioritize classes that are recently introduced or locally predominant, respectively. Our proposal constrains both biases to the last layer by efficiently fine-tuning a pre-trained backbone using learnable prompts, resulting in clients that produce less biased representations and more biased classifiers. Therefore, instead of solely relying on parameter aggregation, we leverage generative prototypes to effectively balance the predictions of the global model. Our proposed methodology significantly improves the current state of the art across six datasets, each including three different scenarios.

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

↑