arXiv:2501.15361cs.LG2025-01被引 20

提出去中心化LoRA,让大模型在分布式隐私数据上高效微调。

Decentralized Low-Rank Fine-Tuning of Large Language Models

  • 基于LoRA设计去中心化微调算法,客户端直接协作无需中心服务器。
  • 在异构数据和量化条件下性能接近集中式LoRA,准确率损失低于2%。
  • 理论证明算法收敛性,适合医疗、金融等隐私敏感场景使用。

尽管参数高效微调(PEFT)技术如低秩适配(LoRA)能高效调整大语言模型(LLMs),但其实际部署常依赖集中式数据与训练环境。现实场景中,数据分布且涉及隐私,需去中心化解决方案。联邦学习(FL)通过客户端间协调更新保护隐私,但通常依赖中心服务器聚合,易产生瓶颈与通信限制。而去中心化学习通过客户端直接协作,提升可扩展性与效率。尽管优势明显,去中心化大模型微调仍研究不足。本文提出Dec-LoRA,一种基于LoRA的去中心化微调算法。在BERT与LLaMA-2模型上的大量实验表明,Dec-LoRA在数据异构性和量化约束下性能接近集中式LoRA,准确率损失低于2%。此外,我们提供了严格的理论保证,证明算法对非凸平滑损失函数可收敛至稳定点。这些发现凸显了Dec-LoRA在去中心化环境中规模化微调大模型的潜力。

原文摘要 · Abstract (English)

While parameter-efficient fine-tuning (PEFT) techniques like Low-Rank Adaptation (LoRA) offer computationally efficient adaptations of Large Language Models (LLMs), their practical deployment often assumes centralized data and training environments. However, real-world scenarios frequently involve distributed, privacy-sensitive datasets that require decentralized solutions. Federated learning (FL) addresses data privacy by coordinating model updates across clients, but it is typically based on centralized aggregation through a parameter server, which can introduce bottlenecks and communication constraints. Decentralized learning, in contrast, eliminates this dependency by enabling direct collaboration between clients, improving scalability and efficiency in distributed environments. Despite its advantages, decentralized LLM fine-tuning remains underexplored. In this work, we propose Dec-LoRA, a decentralized fine-tuning algorithm for LLMs based on LoRA. Through extensive experiments on BERT and LLaMA-2 models, we demonstrate that Dec-LoRA achieves performance comparable to centralized LoRA under various conditions, including data heterogeneity and quantization constraints. Additionally, we provide a rigorous theoretical guarantee proving the convergence of our algorithm to a stationary point for non-convex and smooth loss functions. These findings highlight the potential of Dec-LoRA for scalable LLM fine-tuning in decentralized environments.

大模型微调去中心化学习LoRA联邦学习

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