arXiv:2602.23638cs.LGcs.AI2026-02被引 7

解决联邦学习中低秩微调的旋转错位问题,提升模型聚合稳定性。

FedRot-LoRA: Mitigating Rotational Misalignment in Federated LoRA

  • 通过正交变换对齐客户端更新的潜在空间,减少聚合误差。
  • 在多种数据异构性和低秩配置下,性能优于现有基线方法。
  • 无需增加通信开销,适合资源受限的分布式场景。

联邦低秩微调(Federated LoRA)为在分散数据上高效微调大语言模型提供了通信高效的机制。然而,实践中基于因子平均的聚合方式与数学上正确的局部更新聚合之间存在差异,导致显著的聚合误差和训练不稳定。我们指出,主要原因是低秩分解的旋转不变性——语义等价的更新可能在不同客户端表现为不同的潜在子空间,因为 $(B_i R_i)(R_i^ op A_i) = B_i A_i$。当这些错位的因子直接平均时,会产生破坏性干扰,降低全局更新质量。为此,我们提出 FedRot-LoRA,一种在聚合前通过正交变换对齐客户端更新的联邦 LoRA 框架。该方法在不增加通信成本或限制模型表达能力的前提下,保留语义更新并减少跨客户端子空间不匹配。我们提供了收敛性分析,揭示了因子平均引起的聚合误差,并证明旋转对齐可获得更紧的误差上界。在自然语言理解与生成任务上的大量实验表明,FedRot-LoRA 在多种数据异构性和低秩设置下均一致优于现有联邦 LoRA 基线。

原文摘要 · Abstract (English)

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 $(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.

联邦学习低秩微调模型对齐通信效率

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