arXiv:2506.05568cs.LGcs.AI2025-06NeurIPS被引 9

Ravan通过多头低秩适配提升联邦微调效果,兼顾效率与精度。

Ravan: Multi-Head Low-Rank Adaptation for Federated Fine-Tuning

  • 用多个可训练缩放因子的低秩头重构权重更新,增强表达能力。
  • 在视觉与语言基准上比现有方法准确率提升2-8%。
  • 适合资源受限设备上的大模型联邦微调,尤其数据异构场景。

大型语言模型(LLMs)尚未有效利用边缘设备的海量数据,而联邦学习(FL)为在不传输私有数据的前提下协同微调LLMs提供了可行方案。为适应边缘设备的计算与通信限制,近期研究采用低秩适配(LoRA)等参数高效方法进行联邦微调。然而,基于LoRA的方法在联邦设置中常因客户端间的数据与计算异构性导致性能下降。本文提出Ravan,一种自适应多头LoRA方法,通过将权重更新重参数化为多个LoRA头 $s_i extbf{B}_i extbf{H}_i extbf{A}_i$,仅训练核心矩阵 $ extbf{H}_i$ 及其轻量级缩放因子 $s_i$。这些可训练缩放因子使优化聚焦于最有用的头,无需增加通信参数量即可实现更接近全秩的更新。在视觉与语言基准上的实验表明,Ravan相比现有参数高效基线测试准确率提升2-8%,为大模型联邦微调提供了鲁棒且可扩展的解决方案。

原文摘要 · Abstract (English)

Large language models (LLMs) have not yet effectively leveraged the vast amounts of edge-device data, and federated learning (FL) offers a promising paradigm to collaboratively fine-tune LLMs without transferring private edge data to the cloud. To operate within the computation and communication constraints of edge devices, recent literature on federated fine-tuning of LLMs proposes the use of low-rank adaptation (LoRA) and similar parameter-efficient methods. However, LoRA-based methods suffer from accuracy degradation in FL settings, primarily because of data and computational heterogeneity across clients. We propose Ravan, an adaptive multi-head LoRA method that balances parameter efficiency and model expressivity by reparameterizing the weight updates as the sum of multiple LoRA heads $s_i\textbf{B}_i\textbf{H}_i\textbf{A}_i$ in which only the core matrices $\textbf{H}_i$ and their lightweight scaling factors $s_i$ are trained. These trainable scaling factors let the optimization focus on the most useful heads, recovering a higher-rank approximation of the full update without increasing the number of communicated parameters since clients upload $s_i\textbf{H}_i$ directly. Experiments on vision and language benchmarks show that Ravan improves test accuracy by $2-8\%$ over prior parameter-efficient baselines, making it a robust and scalable solution for federated fine-tuning of LLMs.

联邦学习低秩适配大模型微调

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