FedMomentum提升联邦微调中LoRA的更新效率与精度。
FedMomentum: Preserving LoRA Training Momentum in Federated Fine-Tuning
- 通过SVD提取主更新方向,实现结构化且保动量的LoRA聚合。
- 收敛速度更快,最终准确率超越现有最优方法。
- 适合需要高效隐私保护微调大模型的场景。
在大型语言模型的联邦微调中使用低秩适配(LoRA)可实现通信高效且保护隐私的任务定制。直接独立平均降维和升维矩阵会引入噪声,而现有无噪声聚合策略不可避免地损害LoRA的结构表达能力,导致无法有效保留客户端特有适应。我们识别出此问题为训练动量丢失,使更新无法在多轮间有效累积,导致收敛慢、性能差。为此,提出FedMomentum框架,通过奇异值分解(SVD)实现结构化且保动量的LoRA聚合。具体地,在数学正确聚合低秩更新后,利用SVD提取主导成分以重建同秩的LoRA模块,残余成分可保留并后续合并至主干网络,以保持语义信息和鲁棒性。多任务实验证明,FedMomentum在收敛速度与最终准确率上均持续优于现有最先进方法。
原文摘要 · Abstract (English)
Federated fine-tuning of large language models (LLMs) with low-rank adaptation (LoRA) offers a communication-efficient and privacy-preserving solution for task-specific adaptation. Naive aggregation of LoRA modules introduces noise due to mathematical incorrectness when averaging the downsampling and upsampling matrices independently. However, existing noise-free aggregation strategies inevitably compromise the structural expressiveness of LoRA, limiting its ability to retain client-specific adaptations by either improperly reconstructing the low-rank structure or excluding partially trainable components. We identify this problem as loss of training momentum, where LoRA updates fail to accumulate effectively across rounds, resulting in slower convergence and suboptimal performance. To address this, we propose FedMomentum, a novel framework that enables structured and momentum-preserving LoRA aggregation via singular value decomposition (SVD). Specifically, after aggregating low-rank updates in a mathematically correct manner, FedMomentum applies SVD to extract the dominant components that capture the main update directions. These components are used to reconstruct the LoRA modules with the same rank, while residual components can be retained and later merged into the backbone to preserve semantic information and ensure robustness. Extensive experiments across multiple tasks demonstrate that FedMomentum consistently outperforms prior state-of-the-art methods in convergence speed and final accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。