通过低秩压缩显著降低设备端大模型微调通信开销
SplitLite: Low-Rank Residual Compression for Split Learning

- 利用相邻轮次激活与梯度残差的低秩特性,仅传输压缩后的奇异值分解因子
- 在GLUE基准上实现93.5%的激活上传量减少和83.7%的总通信成本降低
- 适用于资源受限设备上的高效联邦微调,尤其适合高阶LoRA场景
设备端大语言模型(LLM)的联邦微调面临巨大计算负担。为解决此问题,分割学习(SL)将主要训练任务卸载至强大服务器,但需在客户端与服务器间交换高维激活和梯度,导致通信成本高昂。本文提出SplitLite,一种通信高效的分割联邦LoRA微调方法,利用连续轮次激活与梯度残差在有效秩结构上的特性。关键发现:当LoRA在参数空间使用秩 $r$ 更新时,同一数据样本在相邻轮次间的激活与梯度残差分别具有有效秩 $2r$ 与 $4r$。基于此,SplitLite仅传输量化截断的奇异值分解(SVD)残差因子,大幅降低激活上行与梯度下行流量。在一系列先进设备端LLM上的GLUE基准实验表明,该方法可使激活上行通信成本降低最高达93.5%,总通信成本降低最高达83.7%,且无性能损失。
原文摘要 · Abstract (English)
Federated fine-tuning of on-device large language models (LLMs) faces a significant computing burden. To overcome this limitation, split learning (SL) has emerged as a promising solution, which offloads the primary training workload to a powerful server. However, SL requires exchanging high-dimensional activations and gradients between clients and the server, resulting in prohibitive communication costs. To overcome this challenge, we propose SplitLite, a communication-efficient split federated LoRA fine-tuning method that exploits the low effective rank structure of consecutive-epoch activation and gradient residuals. Our key finding is that, when LoRA uses rank $r$ updates in parameter space, the activation and gradient residuals of the same data sample between adjacent epochs also exhibit effective rank-$2r$ and rank-$4r$ structures, respectively. By revealing this property, SplitLite transmits only quantized truncated singular value decomposition (SVD) residual factors, thereby significantly reducing both activation uplink and gradient downlink traffic. Extensive experiments on the GLUE benchmark across a series of advanced on-device LLMs demonstrate that our method reduces activation uplink communication costs by up to 93.5\% and total communication costs by up to 83.7\%, without performance degradation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。