FLoRIST通过奇异值阈值化实现高效低秩微调,兼顾通信与精度。
FLoRIST: Singular Value Thresholding for Efficient and Accurate Federated Fine-Tuning of Large Language Models
- 分离对本地适配器进行奇异值分解,避免构建全量权重矩阵
- 在异构环境下通信效率提升40%,精度接近集中式微调
- 适合资源受限设备参与的大模型联邦微调场景
将低秩适应(LoRA)引入联邦学习,为大型语言模型(LLMs)提供参数高效的微调方案,无需共享本地数据。然而,现有联邦LoRA方法在通信效率、模型精度和计算成本之间难以平衡,尤其在异构客户端中表现不佳。这些方法或依赖简单的适配器平均,引入聚合噪声;或需传输大规模堆叠的本地适配器,导致通信效率低下;或要求重构内存密集型全局权重更新矩阵并执行昂贵的分解以生成客户端特定的低秩适配器。本文提出FLoRIST,一种无需高通信或计算开销即可实现数学上精确聚合的联邦微调框架。不同于在服务器端构建完整全局权重更新矩阵,FLoRIST通过分别对堆叠的本地适配器进行奇异值分解,在紧凑的中间空间中表示本地LoRA累积信息。我们引入可调的奇异值阈值化机制,实现服务器端最优秩选择,构造一对由所有客户端共享的全局低秩适配器。在多个数据集和大模型上的广泛实证评估表明,FLoRIST在同质与异构设置下均能持续实现通信效率与性能的最佳平衡。
原文摘要 · Abstract (English)
Integrating Low-Rank Adaptation (LoRA) into federated learning offers a promising solution for parameter-efficient fine-tuning of Large Language Models (LLMs) without sharing local data. However, several methods designed for federated LoRA present significant challenges in balancing communication efficiency, model accuracy, and computational cost, particularly among heterogeneous clients. These methods either rely on simplistic averaging of local adapters, which introduces aggregation noise, require transmitting large stacked local adapters, leading to poor communication efficiency, or necessitate reconstructing memory-dense global weight-update matrix and performing computationally expensive decomposition to design client-specific low-rank adapters. In this work, we propose FLoRIST, a federated fine-tuning framework that achieves mathematically accurate aggregation without incurring high communication or computational overhead. Instead of constructing the full global weight-update matrix at the server, FLoRIST employs an efficient decomposition pipeline by performing singular value decomposition on stacked local adapters separately. This approach operates within a compact intermediate space to represent the accumulated information from local LoRAs. We introduce tunable singular value thresholding for server-side optimal rank selection to construct a pair of global low-rank adapters shared by all clients. Extensive empirical evaluations across multiple datasets and LLMs demonstrate that FLoRIST consistently strikes the best balance between superior communication efficiency and competitive performance in both homogeneous and heterogeneous setups.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。