arXiv:2503.21213cs.LG2025-03被引 3

针对异构数据下的大模型高效微调,提出分组动态优化框架

Resource-Efficient Federated Fine-Tuning Large Language Models for Heterogeneous Data

  • 按设备数据分布分组,动态调整聚合频率与微调深度
  • 实测提升准确率1.6%~4.2%,提速超2.1倍
  • 适合资源受限且数据异构的边缘部署场景

通过联邦学习微调大语言模型(FedLLM)可在保护隐私的前提下适配多种下游任务。为降低资源受限设备的开销,FedLoRA引入低秩适应(LoRA)仅微调部分参数。然而,除资源限制外,数据异构仍是实际应用中关键挑战。为此,受分组联邦学习启发,本文提出层级化FedLoRA(HierFedLoRA),将设备划分为多个近似同分布的组,并动态调节各组内部聚合频率以缓解非独立同分布数据的负面影响。同时,为降低计算与通信成本,为每组动态分配不同微调深度(即从输出端连续微调的层数)。通过联合优化聚合频率与微调深度,进一步提升性能。在80台商用设备组成的物理平台上进行大量实验,结果表明,相比强基线,HierFedLoRA可提升最终模型准确率1.6%至4.2%,且微调速度至少加快2.1倍。

原文摘要 · Abstract (English)

Fine-tuning large language models (LLMs) via federated learning, i.e., FedLLM, has been proposed to adapt LLMs for various downstream applications in a privacy-preserving way. To reduce the fine-tuning costs on resource-constrained devices, FedLoRA is proposed to fine-tune only a small subset of model parameters by integrating low-rank adaptation (LoRA) into FedLLM. However, apart from resource constraints, there is still another critical challenge, i.e., data heterogeneity, severely hindering the implementation of FedLoRA in practical applications. Herein, inspired by the previous group-based federated learning paradigm, we propose a hierarchical FedLoRA framework, termed HierFedLoRA, to address these challenges. Specifically, HierFedLoRA partitions all devices into multiple near-IID groups and adjusts the intra-group aggregation frequency for each group to eliminate the negative effects of non-IID data. Meanwhile, to reduce the computation and communication cost, HierFedLoRA dynamically assigns diverse and suitable fine-tuning depth (i.e., the number of continuous fine-tuning layers from the output) for each group. HierFedLoRA explores jointly optimizing aggregation frequency and depth upon their coupled relationship to better enhance the performance of FedLoRA. Extensive experiments are conducted on a physical platform with 80 commercial devices. The results show that HierFedLoRA improves the final model accuracy by 1.6% to 4.2%, speeding up the fine-tuning process by at least 2.1$\times$, compared to the strong baselines.

联邦学习大模型微调资源效率异构数据

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