arXiv:2504.16357cs.DCcs.AI2025-04被引 4

用双提示词提升大模型在小数据下的个性化联邦学习效果

DP2FL: Dual Prompt Personalized Federated Learning in Foundation Models

  • 设计双提示词机制,融合全局任务与本地数据特征
  • 在异构数据下实现更高准确率,新客户端无需重训练
  • 适合数据少、需快速接入的新客户端场景

个性化联邦学习(PFL)能应对客户端数据分布异构性并保护隐私。但当本地数据量有限时,深度模型易因训练不足导致性能下降。如CLIP等基础模型具备强特征提取能力,可通过微调缓解此问题。然而,基础模型在联邦学习中应用仍很少,且新客户端集成难题未解。为此,本文提出双提示词个性化联邦学习(DP2FL),引入双提示词与自适应聚合策略,结合全局任务认知与本地数据驱动信息,使本地模型既具泛化能力又适应特定分布。此外,全局模型支持对新数据源预测,并可无缝集成新增客户端而无需重训练。在高度异构环境中的实验验证了DP2FL提示设计与聚合策略的有效性,凸显其对新数据源的预测优势及新客户端的无痛接入能力。

原文摘要 · Abstract (English)

Personalized federated learning (PFL) has garnered significant attention for its ability to address heterogeneous client data distributions while preserving data privacy. However, when local client data is limited, deep learning models often suffer from insufficient training, leading to suboptimal performance. Foundation models, such as CLIP (Contrastive Language-Image Pretraining), exhibit strong feature extraction capabilities and can alleviate this issue by fine-tuning on limited local data. Despite their potential, foundation models are rarely utilized in federated learning scenarios, and challenges related to integrating new clients remain largely unresolved. To address these challenges, we propose the Dual Prompt Personalized Federated Learning (DP2FL) framework, which introduces dual prompts and an adaptive aggregation strategy. DP2FL combines global task awareness with local data-driven insights, enabling local models to achieve effective generalization while remaining adaptable to specific data distributions. Moreover, DP2FL introduces a global model that enables prediction on new data sources and seamlessly integrates newly added clients without requiring retraining. Experimental results in highly heterogeneous environments validate the effectiveness of DP2FL's prompt design and aggregation strategy, underscoring the advantages of prediction on novel data sources and demonstrating the seamless integration of new clients into the federated learning framework.

联邦学习提示词大模型个性化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。