arXiv:2505.02795cs.LGcs.AI2025-05被引 45

针对异构设备的LLM高效微调,提升精度与速度

HSplitLoRA: A Heterogeneous Split Parameter-Efficient Fine-Tuning Framework for Large Language Models

  • 基于权重重要性动态分配低秩适配器,按设备算力分拆模型
  • 在多个数据集上实现更高训练精度和更快收敛速度
  • 适合资源差异大的边缘设备联合微调场景

大型语言模型(LLMs)在自然语言处理等领域取得显著突破。由于参数量巨大,使用私有数据对这些模型进行下游任务微调已成为主流。尽管联邦学习(FL)可在不共享原始数据的情况下微调LLMs,但高昂的计算成本阻碍其普及。此外,在实际场景中,客户端设备通常具备异构的计算资源,进一步增加了微调难度。为此,我们提出HSplitLoRA,一种基于分片学习(SL)与低秩适应(LoRA)的异构参数高效微调(PEFT)框架,用于在异构客户端设备上高效微调LLMs。HSplitLoRA首先根据权重对训练的贡献度识别重要权重;随后为选定权重动态配置LoRA适配器的分解秩,并依据客户端不同的计算预算确定模型分割点;最后设计无噪声适配器聚合机制,支持异构适配器聚合且不引入噪声。大量实验表明,HSplitLoRA在训练精度和收敛速度上均优于当前最优基准。

原文摘要 · Abstract (English)

Recently, large language models (LLMs) have achieved remarkable breakthroughs, revolutionizing the natural language processing domain and beyond. Due to immense parameter sizes, fine-tuning these models with private data for diverse downstream tasks has become mainstream. Though federated learning (FL) offers a promising solution for fine-tuning LLMs without sharing raw data, substantial computing costs hinder its democratization. Moreover, in real-world scenarios, private client devices often possess heterogeneous computing resources, further complicating LLM fine-tuning. To combat these challenges, we propose HSplitLoRA, a heterogeneous parameter-efficient fine-tuning (PEFT) framework built on split learning (SL) and low-rank adaptation (LoRA) fine-tuning, for efficiently fine-tuning LLMs on heterogeneous client devices. HSplitLoRA first identifies important weights based on their contributions to LLM training. It then dynamically configures the decomposition ranks of LoRA adapters for selected weights and determines the model split point according to varying computing budgets of client devices. Finally, a noise-free adapter aggregation mechanism is devised to support heterogeneous adapter aggregation without introducing noise. Extensive experiments demonstrate that HSplitLoRA outperforms state-of-the-art benchmarks in training accuracy and convergence speed.

LLM微调联邦学习参数高效异构计算

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