Communication-Efficient and Accurate Approach for Aggregation in Federated Low-Rank Adaptation
Authors:
- Le-Tuan NguyenLNLe-Tuan Nguyen
- Minh-Duong NguyenMNMinh-Duong Nguyen
- Dung D. LeDLDung D. Le
- Quoc-Viet PhamQPQuoc-Viet Pham
Preprint
Abstract
With the rapid emergence of foundation models and the increasing need for fine-tuning across distributed environments, Federated Low-Rank Adaptation (FedLoRA) has recently gained significant attention. Despite enormous potential, current FedLoRA methods face notable challenges due to inexact updates. Existing approaches have attempted to mitigate this issue, but they often introduce a \emph{local-global generalization gap} and incur \emph{substantial communication overhead}, limiting their scalability and effectiveness. To address these limitations, we propose \textbf{F}ederated \textbf{Lo}w-\textbf{R}ank \textbf{A}ggregation with \textbf{N}early \textbf{A}ccurate Estimation (FLoRA-NA). FLoRA-NA leverages the local LoRA matrices on the server to estimate the aggregated matrices $\hat{A}$ and $\hat{B}$, which are then distributed to clients for local updates. This surrogated aggregated matrices minimizes the divergence between ideal $\nabla \Bar{W} = \sum^{U}_{u=1}B_u A_u$ and practical updates $\nabla \hat{W} = \hat{B}\hat{A}$ without adding communication cost beyond vanilla FedLoRA. By doing so, FLoRA-NA achieves communication efficiency and bridges the gap between local personalization and global generalization, addressing a key limitation of prior personalized FedLoRA approaches. We conduct extensive evaluations across diverse tasks, including natural language understanding, mathematical reasoning, and code-solving ability using various foundation models. Experimental results consistently demonstrate that FLoRA-NA achieves state-of-the-art global performance while maintaining low communication overhead.
Keywords
All Related Versions
This publication has 9 references indexed in Scilit:
- Federated Fine-Tuning on Heterogeneous LoRAs With Error-Compensated AggregationIEEE Transactions on Neural Networks and Learning Systems, 2025
- Differentially Private Low-Rank Adaptation of Large Language Model Using Federated LearningACM Transactions on Management Information Systems, 2025
- Federated Learning for Generalization, Robustness, Fairness: A Survey and BenchmarkIEEE Transactions on Pattern Analysis and Machine Intelligence, 2024
- Efficient Multi-task LLM Quantization and Serving for Multiple LoRA Adapters2024
- FLoRA: Federated Fine-Tuning Large Language Models with Heterogeneous Low-Rank Adaptations2024
- QLoRA: Efficient Finetuning of Quantized LLMs2023
- FedBERT: When Federated Learning Meets Pre-trainingACM Transactions on Intelligent Systems and Technology, 2022
- HCFL: A High Compression Approach for Communication-Efficient Federated Learning in Very Large Scale IoT NetworksIEEE Transactions on Mobile Computing, 2022
- GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understanding2018