English

FedRot-LoRA: Mitigating Rotational Misalignment in Federated LoRA

Machine Learning 2026-05-13 v2 Artificial Intelligence

Abstract

Federated LoRA provides a communication-efficient mechanism for fine-tuning large language models on decentralized data. In practice, however, a discrepancy between the factor-wise averaging used to preserve low rank and the mathematically correct aggregation of local updates can cause significant aggregation error and unstable training. We argue that a major source of this problem is rotational misalignment, arising from the rotational invariance of low-rank factorizations -- semantically equivalent updates can be represented in different latent subspaces across clients since (BiRi)(RiAi)=BiAi(B_i R_i)(R_i^\top A_i) = B_i A_i. When such misaligned factors are averaged directly, they interfere destructively and degrade the global update. To address this issue, we propose FedRot-LoRA, a federated LoRA framework that aligns client updates via orthogonal transformations prior to aggregation. This alignment preserves the semantic update while reducing cross-client subspace mismatch, without increasing communication cost or restricting model expressivity. We provide a convergence analysis that examines the aggregation error induced by factor-wise averaging and shows how rotational alignment yields a tighter upper bound on this error. Extensive experiments on natural language understanding and generative tasks demonstrate that FedRot-LoRA consistently outperforms existing federated LoRA baselines across a range of heterogeneity levels and LoRA ranks.

Keywords

Cite

@article{arxiv.2602.23638,
  title  = {FedRot-LoRA: Mitigating Rotational Misalignment in Federated LoRA},
  author = {Haoran Zhang and Dongjun Kim and Seohyeon Cha and Haris Vikalo},
  journal= {arXiv preprint arXiv:2602.23638},
  year   = {2026}
}

Comments

ICML 2026

R2 v1 2026-07-01T10:54:51.290Z