Towards Robust and Efficient Federated Low-Rank Adaptation with Heterogeneous Clients

Jabin Koo, Minwoo Jang, Jungseul Ok


Abstract
Federated fine-tuning for Large Language Models (LLMs) has recently gained attention due to the heavy communication overhead of transmitting large model updates. Low Rank Adaptation (LoRA) has been proposed as a solution, yet its application in federated learning is complicated by discordance in aggregation. Existing methods addressing this discordance often suffer from performance degradation at low ranks in heterogeneous data settings. In response, we introduce LoRA-A^2 (Low Rank Adaptation with Alternating freeze and Adaptive rank selection), which demonstrates robustness in challenging settings with low ranks and high data heterogeneity. Our experimental findings reveal that LoRA-A^2 maintains performance even under extreme heterogeneity and low rank conditions, achieving up to a 99.8% reduction in uploaded parameters compared to full fine-tuning without compromising performance. This adaptive mechanism boosts robustness and communication efficiency in federated fine-tuning, enabling the practical deployment of LLMs in resource-constrained environments.
Anthology ID:
2025.acl-long.19
Volume:
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
July
Year:
2025
Address:
Vienna, Austria
Editors:
Wanxiang Che, Joyce Nabende, Ekaterina Shutova, Mohammad Taher Pilehvar
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
416–429
Language:
URL:
https://preview.aclanthology.org/ingestion-acl-25/2025.acl-long.19/
DOI:
Bibkey:
Cite (ACL):
Jabin Koo, Minwoo Jang, and Jungseul Ok. 2025. Towards Robust and Efficient Federated Low-Rank Adaptation with Heterogeneous Clients. In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 416–429, Vienna, Austria. Association for Computational Linguistics.
Cite (Informal):
Towards Robust and Efficient Federated Low-Rank Adaptation with Heterogeneous Clients (Koo et al., ACL 2025)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingestion-acl-25/2025.acl-long.19.pdf