FLoRG通过聚合低秩矩阵的内积矩阵,解决联邦微调中的误差与漂移问题。
FLoRG: Federated Fine-tuning with Low-rank Gram Matrices and Procrustes Alignment
- 用单一低秩矩阵的内积矩阵替代两个分离矩阵,减少聚合误差。
- 在多个基准上提升下游任务准确率,通信开销降低2041倍。
- 引入Procrustes对齐,稳定更新过程,适合隐私敏感场景的模型微调。
参数高效的微调技术如低秩适配(LoRA)可高效地将大语言模型(LLMs)适配至下游任务。联邦学习(FL)进一步支持在不共享私有数据的前提下,跨分布式客户端协同微调。然而,LoRA中使用两个独立的低秩矩阵进行联邦微调会带来两类挑战:其一,分别聚合两个低秩矩阵会产生聚合误差;其二,即使服务器聚合了两矩阵乘积,仍需将其重新分解为低秩形式,而分解不唯一,导致分解漂移。为此,我们提出联邦低秩格拉姆矩阵聚合(FLoRG),一种仅使用单个低秩矩阵进行微调并聚合其格拉姆矩阵(即列向量内积构成的矩阵)的框架。该方法可消除聚合误差、降低通信开销,并通过引入普罗克鲁斯特斯对齐方法,在连续训练轮次间对齐分解矩阵,从而最小化分解漂移。我们从理论上分析了FLoRG的收敛性,证明采用普罗克鲁斯特斯对齐可获得更紧的收敛界。在多个大语言模型微调基准上的实验表明,FLoRG优于五种前沿基线方法,在下游任务准确率上表现更优,且通信开销最高可降低2041×。
原文摘要 · Abstract (English)
Parameter-efficient fine-tuning techniques such as low-rank adaptation (LoRA) enable large language models (LLMs) to adapt to downstream tasks efficiently. Federated learning (FL) further facilitates this process by enabling collaborative fine-tuning across distributed clients without sharing private data. However, the use of two separate low-rank matrices in LoRA for federated fine-tuning introduces two types of challenges. First, aggregation error can arise from separately aggregating the two low-rank matrices. Second, even if the server aggregates the product of two low-rank matrices, it needs to decompose the aggregated matrix back into low-rank matrices. Since the decomposition is not unique, it can lead to decomposition drift. To tackle the aforementioned challenges, we propose federated low-rank Gram-matrix aggregation (FLoRG), a federated fine-tuning framework which employs a single low-rank matrix for fine-tuning and aggregates its Gram matrix (i.e., the matrix of inner products of its column vectors). FLoRG can eliminate the aggregation error and reduce the communication overhead. It also minimizes the decomposition drift by introducing a Procrustes alignment approach which aligns the decomposed matrix between consecutive fine-tuning rounds for consistent updates. We theoretically analyze the convergence of FLoRG and prove that adopting the Procrustes alignment results in a tighter convergence bound. Experimental results across multiple LLM fine-tuning benchmarks demonstrate that FLoRG outperforms five state-of-the-art baseline schemes by providing higher downstream task accuracy and can reduce the communication overhead by up to 2041$\times$.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。