arXiv:2501.14406cs.DCcs.AI2025-01被引 9

针对联邦微调中数据异构与通信低效问题,提出动态分配参数秩的新方法。

Adaptive Rank Allocation for Federated Parameter-Efficient Fine-Tuning of Language Models

  • 用截断SVD增强客户端特征表示一致性,缓解数据异构影响
  • 动态分配关键秩,使通信效率提升2.4倍,平均性能提升7%以上
  • 自动剪枝无效模块,降低设备端计算与能耗,最多省48.9%

预训练语言模型(PLMs)在现代自然语言处理中展现出卓越性能,可通过微调适配多种下游任务。联邦参数高效微调(FedPEFT)为资源受限设备上的分布式训练提供了隐私与效率兼顾的解决方案。然而,我们的测量揭示了两个关键局限:设备间数据异构加剧了低秩适应的性能下降,固定参数配置导致通信效率低下。为此,我们提出FedARA,一种面向联邦参数高效微调的语言模型自适应秩分配框架。具体而言,FedARA采用截断奇异值分解(SVD)适应,增强客户端间相似特征表示,显著缓解数据异构的负面影响;随后通过动态秩分配逐步识别关键秩,有效提升通信效率;最后利用基于秩的模块剪枝,自动移除无效模块,持续降低每轮联邦学习的本地计算成本与内存占用。大量实验表明,当面临数据异构时,FedARA在多个数据集和模型上平均性能优于基线6.95%至8.49%,通信效率提升2.40×。在各类边缘设备上的实验还显示,总训练时间与能耗分别最多减少48.90%和46.95%。

原文摘要 · Abstract (English)

Pre-trained Language Models (PLMs) have demonstrated their superiority and versatility in modern Natural Language Processing (NLP), effectively adapting to various downstream tasks through further fine-tuning. Federated Parameter-Efficient Fine-Tuning (FedPEFT) has emerged as a promising solution to address privacy and efficiency challenges in distributed training for PLMs on resource-constrained local devices. However, our measurements reveal two key limitations of FedPEFT: heterogeneous data across devices exacerbates performance degradation of low-rank adaptation, and a fixed parameter configuration results in communication inefficiency. To overcome these limitations, we propose FedARA, a novel adaptive rank allocation framework for federated parameter-efficient fine-tuning of language models. Specifically, FedARA employs truncated Singular Value Decomposition (SVD) adaptation to enhance similar feature representation across clients, significantly mitigating the adverse effects of data heterogeneity. Subsequently, it utilizes dynamic rank allocation to progressively identify critical ranks, effectively improving communication efficiency. Lastly, it leverages rank-based module pruning to automatically remove inactive modules, steadily reducing local computational cost and memory usage in each federated learning round. Extensive experiments show that FedARA consistently outperforms baselines by an average of 6.95% to 8.49% across various datasets and models under heterogeneous data while significantly improving communication efficiency by 2.40$ \times$. Moreover, experiments on various edge devices demonstrate substantial decreases in total training time and energy consumption by up to 48.90% and 46.95%, respectively.

联邦学习参数高效秩分配边缘计算

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