RW-LoRA: Communication-Efficient Decentralized LoRA Fine-Tuning via Random Walks

📅 2026-08-31
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
本文提出RW-LoRA方法,通过随机游走减少通信开销和计算成本,解决分布式环境下LoRA微调的效率问题。
📝 Abstract
Parameter-efficient fine-tuning methods such as LoRA have become a standard approach for adapting large foundation models. Adopting fine-tuning to distributed settings faces several challenges. Most existing distributed LoRA methods rely on centralized aggregation, and gossip-based decentralized LoRA requires repeated synchronization among multiple model copies. Both methods incur significant communication overhead and introduce errors due to simultaneous aggregation of multiple model updates. In this paper, we take a different perspective and propose a random-walk-based LoRA fine-tuning scheme. Instead of maintaining multiple model replicas, a single model token traverses the network and is updated sequentially using local fine-tuning objectives. This design eliminates the need for global synchronization, substantially reduces communication and computation costs, and avoids aggregation errors. We provide rigorous convergence guarantees for non-convex objectives under standard assumptions. Through empirical results on multiple NLP tasks and graph topologies, we show that the proposed method achieves competitive task performance with substantially less communication and computation than gossip-based LoRA.
Problem

Research questions and friction points this paper is trying to address.

Distributed LoRA
Communication Overhead
Synchronization
Aggregation Errors
Innovation

Methods, ideas, or system contributions that make the work stand out.

Random Walks
Decentralized LoRA
Communication-Efficient
🔎 Similar Papers
No similar papers found.
X
Xingran Chen
Engineering Systems and Design Pillar, Singapore University of Technology and Design, Singapore 487372
R
Rohit Bhagat
Department of Electrical and Computer Engineering, Rutgers University, Piscataway Township, NJ 08854, USA
G
Ghadir Ayache
LinkedIn, New York, NY 10118, USA
Rawad Bitar
Rawad Bitar
Research Group Leader (and Habiliation Candidate), Technical University of Munich
Coding theoryDistributed learningDNA-based data storageData privacy and security
Yanmin Gong
Yanmin Gong
Texas A&M University
Trustworthy AISecurity and PrivacyEdge IntelligenceAI in Health
S
Salim El Rouayheb
Department of Electrical and Computer Engineering, Rutgers University, Piscataway Township, NJ 08854, USA