解决大模型在长尾数据下微调导致性能下降的问题
Fine-Tuning Impairs the Balancedness of Foundation Models in Long-tailed Personalized Federated Learning

- 用零样本预测净化本地梯度,保持全局模型类平衡
- 将个性化视为对冻结全局模型的残差修正,避免偏差传播
- 在多种长尾场景中同时提升全局与个性化模型性能
基于大模型的个性化联邦学习(PFL)能适应客户端异构数据分布,但现实场景常同时存在非独立同分布数据与长尾类别分布,现有方法面临两大挑战:(i) 微调会削弱大模型固有的类别平衡性,导致性能低于零样本基线;(ii) 传统个性化技术通过参数或特征级融合进一步将偏差传递至本地模型。为此,本文提出联邦学习梯度净化与残差学习框架(FedPuReL),在保持全局模型类平衡的同时实现无偏个性化。具体而言,利用零样本预测净化本地梯度以维持全局模型的平衡性,并将个性化建模为在冻结全局模型上的残差修正。大量实验表明,FedPuReL在不同长尾场景下均显著优于现有方法,在全局与个性化模型上表现更优。
原文摘要 · Abstract (English)
Personalized federated learning (PFL) with foundation models has emerged as a promising paradigm enabling clients to adapt to heterogeneous data distributions. However, real-world scenarios often face the co-occurrence of non-IID data and long-tailed class distributions, presenting unique challenges that remain underexplored in PFL. In this paper, we investigate this long-tailed personalized federated learning and observe that current methods suffer from two limitations: (i) fine-tuning degrades performance below zero-shot baselines due to the erosion of inherent class balance in foundation models; (ii) conventional personalization techniques further transfer this bias to local models through parameter or feature-level fusion. To address these challenges, we propose Federated Learning via Gradient Purification and Residual Learning (FedPuReL), which preserves balanced knowledge in the global model while enabling unbiased personalization. Specifically, we purify local gradients using zero-shot predictions to maintain a class-balanced global model, and model personalization as residual correction atop the frozen global model. Extensive experiments demonstrate that FedPuReL consistently outperforms state-of-the-art methods, achieving superior performance on both global and personalized models across diverse long-tailed scenarios. The code is available at https://github.com/shihaohou/FedPuReL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。