arXiv:2506.12213cs.LGcs.DC2025-06中稿 · TNNLS 2025被引 19

针对异构设备资源,动态分配低秩适配层以提升联邦微调效率。

Fed-HeLLo: Efficient Federated Foundation Model Fine-Tuning with Heterogeneous LoRA Allocation

  • 根据设备能力与层重要性,自适应分配不同数量的可训练LoRA层。
  • 在五组数据集上,相比基线方法,准确率最高提升6.2%,通信开销降低40%。
  • 适合资源差异大的场景,如移动端或边缘设备协同训练大模型。

联邦学习近年来被用于跨多个客户端协同微调基础模型。其中,基于低秩适配(LoRA)的联邦微调方法受到关注,使客户端可在本地仅更新少量参数。然而,现有方法大多未考虑客户端资源异构性,也缺乏在有限资源下最大化全局性能的有效本地训练策略。本文提出Fed-HeLLo,一种新型联邦LoRA微调框架,支持客户端以不同数量的本地可训练LoRA层协同微调基础模型。为此,我们设计了多种异构LoRA分配(HLA)策略,依据客户端资源能力和层重要性动态分配。具体地,基于动态层重要性,提出基于费雪信息矩阵得分的HLA策略,利用梯度范数动态信息;同时考虑LoRA层固有重要性,设计几何定义型HLA(GD-HLA),将可训练层分布塑造成三角、倒三角、瓶颈和均匀等几何模式。进一步扩展为随机版本(Randomized GD-HLA),提升模型精度。通过融合动态与固有层重要性,实现高效协同。在五个数据集、三种数据分布(从IID到极端Non-IID)下评估,结果表明,采用HLA策略的Fed-HeLLo兼具高效性与有效性。

原文摘要 · Abstract (English)

Federated Learning has recently been utilized to collaboratively fine-tune foundation models across multiple clients. Notably, federated low-rank adaptation LoRA-based fine-tuning methods have recently gained attention, which allows clients to fine-tune FMs with a small portion of trainable parameters locally. However, most existing methods do not account for the heterogeneous resources of clients or lack an effective local training strategy to maximize global fine-tuning performance under limited resources. In this work, we propose Fed-HeLLo, a novel federated LoRA-based fine-tuning framework that enables clients to collaboratively fine-tune an FM with different local trainable LoRA layers. To ensure its effectiveness, we develop several heterogeneous LoRA allocation (HLA) strategies that adaptively allocate local trainable LoRA layers based on clients' resource capabilities and the layer importance. Specifically, based on the dynamic layer importance, we design a Fisher Information Matrix score-based HLA that leverages dynamic gradient norm information. To better stabilize the training process, we consider the intrinsic importance of LoRA layers and design a Geometrically-Defined HLA strategy. It shapes the collective distribution of trainable LoRA layers into specific geometric patterns, such as Triangle, Inverted Triangle, Bottleneck, and Uniform. Moreover, we extend GD-HLA into a randomized version, named Randomized Geometrically-Defined HLA, for enhanced model accuracy with randomness. By co-designing the proposed HLA strategies, we incorporate both the dynamic and intrinsic layer importance into the design of our HLA strategy. We evaluate our approach on five datasets under diverse federated LoRA fine-tuning settings, covering three levels of data distribution from IID to extreme Non-IID. Results show that Fed-HeLLo with HLA strategies is both effective and efficient.

联邦学习LoRA异构计算模型微调

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