arXiv:2510.00206cs.LGcs.AI2025-10中稿 · EuroSys 2026被引 7

优化大模型微调效率,减少内存访问并支持多适配器并发训练。

LoRAFusion: Efficient LoRA Fine-Tuning for LLMs

  • 通过图拆分融合内存密集型操作,减少冗余内存读写。
  • 多任务调度采用自适应批处理,提升显卡负载均衡与通信重叠。
  • 在多个基准上实现1.47倍平均加速,可直接替换现有LoRA系统。

低秩适配(LoRA)已成为大型语言模型参数高效微调的主流方法,显著降低GPU内存占用并保持下游任务性能。然而,现有LoRA微调系统存在两大效率瓶颈:一是对大规模激活张量产生大量冗余内存访问,导致运行时开销高;二是未能充分利用同一基模型上多独立LoRA适配器的并发训练机会,错失减少流水线空洞、改善通信重叠和提升显卡负载平衡的性能增益。为此,我们提出LoRAFusion,一种高效的LLM LoRA微调系统。在内核层面,提出图拆分方法融合内存密集型操作,消除不必要的内存访问,同时不增加计算量或同步开销。在调度层面,引入自适应批处理算法,先将LoRA适配器分组以有意错开各任务的批次执行时间,再在每组内求解带依赖关系的装箱问题,生成负载均衡的微批次。与Megatron-LM相比,LoRAFusion实现最高1.96倍(平均1.47倍)端到端加速;相较于当前最优的mLoRA系统,最高提升1.46倍(平均1.29倍)。其融合内核最高提升1.39倍(平均1.27倍)性能,可作为即插即用模块集成至现有LoRA系统。项目已开源于https://github.com/CentML/lorafusion。

原文摘要 · Abstract (English)

Low-Rank Adaptation (LoRA) has become the leading Parameter-Efficient Fine-Tuning (PEFT) method for Large Language Models (LLMs), as it significantly reduces GPU memory usage while maintaining competitive fine-tuned model quality on downstream tasks. Despite these benefits, we identify two key inefficiencies in existing LoRA fine-tuning systems. First, they incur substantial runtime overhead due to redundant memory accesses on large activation tensors. Second, they miss the opportunity to concurrently fine-tune multiple independent LoRA adapters that share the same base model on the same set of GPUs. This leads to missed performance gains such as reduced pipeline bubbles, better communication overlap, and improved GPU load balance. To address these issues, we introduce LoRAFusion, an efficient LoRA fine-tuning system for LLMs. At the kernel level, we propose a graph-splitting method that fuses memory-bound operations. This design eliminates unnecessary memory accesses and preserves the performance of compute-bound GEMMs without incurring the cost of recomputation or synchronization. At the scheduling level, LoRAFusion introduces an adaptive batching algorithm for multi-job fine-tuning. It first splits LoRA adapters into groups to intentionally stagger batch execution across jobs, and then solves a bin-packing problem within each group to generate balanced, dependency-aware microbatches. LoRAFusion achieves up to $1.96\times$ ($1.47\times$ on average) end-to-end speedup compared to Megatron-LM, and up to $1.46\times$ ($1.29\times$ on average) improvement over mLoRA, the state-of-the-art multi-LoRA fine-tuning system. Our fused kernel achieves up to $1.39\times$ ($1.27\times$ on average) kernel performance improvement and can directly serve as a plug-and-play replacement in existing LoRA systems. We open-source LoRAFusion at https://github.com/CentML/lorafusion.

大模型微调LoRAGPU优化高效训练

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