轻量级联邦微调方法,提升大模型在异构数据下的个性化性能。
FedMCP: Parameter-Efficient Federated Learning with Model-Contrastive Personalization
- 在冻结大模型基础上加全局与私有适配器,仅上传全局适配器降低通信开销。
- 通过模型对比正则化,使全局适配器学通用知识,私有适配器学客户端特有知识。
- 在跨任务、跨数据中心的异构数据上显著优于现有联邦微调方法。
随着对数据隐私的关注和监管加强,联邦学习(FL)中微调预训练语言模型(PLMs)已成为NLP任务的常见范式。尽管已有广泛研究,现有方法仍面临两大挑战:一是大规模PLM参数量大,导致通信和计算开销过高;二是客户端间数据和任务异构性严重,难以达到理想的微调性能。为此,我们提出FedMCP,一种具有模型对比个性化能力的新型参数高效微调方法。具体而言,FedMCP在客户端冻结的PLM上添加两个轻量级适配器:全局适配器和私有适配器。每轮通信中,各客户端仅将全局适配器发送至服务器进行联邦聚合。此外,FedMCP引入两个适配器间的模型对比正则化项,一方面促使全局适配器吸收通用知识,另一方面使私有适配器捕捉客户端专属知识。通过协同使用两个适配器,FedMCP能有效生成针对个体客户端的个性化微调模型。在高度异构的跨任务、跨存储库数据集上的大量实验表明,相比最先进的联邦微调方法,FedMCP实现了显著的性能提升。
原文摘要 · Abstract (English)
With increasing concerns and regulations on data privacy, fine-tuning pretrained language models (PLMs) in federated learning (FL) has become a common paradigm for NLP tasks. Despite being extensively studied, the existing methods for this problem still face two primary challenges. First, the huge number of parameters in large-scale PLMs leads to excessive communication and computational overhead. Second, the heterogeneity of data and tasks across clients poses a significant obstacle to achieving the desired fine-tuning performance. To address the above problems, we propose FedMCP, a novel parameter-efficient fine-tuning method with model-contrastive personalization for FL. Specifically, FedMCP adds two lightweight adapter modules, i.e., the global adapter and the private adapter, to the frozen PLMs within clients. In a communication round, each client sends only the global adapter to the server for federated aggregation. Furthermore, FedMCP introduces a model-contrastive regularization term between the two adapters. This, on the one hand, encourages the global adapter to assimilate universal knowledge and, on the other hand, the private adapter to capture client-specific knowledge. By leveraging both adapters, FedMCP can effectively provide fine-tuned personalized models tailored to individual clients. Extensive experiments on highly heterogeneous cross-task, cross-silo datasets show that FedMCP achieves substantial performance improvements over state-of-the-art FL fine-tuning approaches for PLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。