arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.01746cs.LGcs.AI
收藏

重新思考LoRA用于数据异质联邦学习:子空间和状态对齐

Rethinking LoRA for Data Heterogeneous Federated Learning: Subspace and State Alignment

College of Computing and Data Science, Nanyang Technological University, Singapore.(南洋理工大学计算机与数据科学学院);Department of Electrical and Computer Engineering, The University of British Columbia, Vancouver, BC, Canada.(不列颠哥伦比亚大学电子与计算机工程系);Vector Institute, Canada(加拿大向量研究所);The Hong Kong Polytechnic University(香港理工大学)

Hongyi Peng, Han Yu, Xiaoxiao Li, Qiang Yang

AI总结:

FedGaLore通过子空间优化与漂移鲁棒同步提升非IID联邦学习中LoRA的鲁棒性和准确性。

摘要:

低秩适应(LoRA)广泛用于联邦微调。然而在非独立同分布(non-IID)设置下,它可能显著劣于全参数微调。通过高概率鲁棒性分析,我们发现这一差距可以归因于两个耦合的不匹配:(i)更新空间不匹配,其中客户端在低秩子空间中优化,但聚合发生在全空间中;(ii)优化器状态不匹配,其中非同步的自适应状态在轮次间放大漂移。我们提出了FedGaLore,它结合了客户端侧GaLore风格的梯度子空间优化与服务器侧通过谱共享信号提取实现的漂移鲁棒同步,以解决这一挑战。在NLU、视觉和NLG基准测试中,FedGaLore在非IID设置下比最先进的联邦LoRA基线在鲁棒性和准确性上都有所提升。

英文摘要:

Low-Rank Adaptation (LoRA) is widely used for federated fine-tuning. Yet under non-IID settings, it can substantially underperform full-parameter fine-tuning. Through with-high-probability robustness analysis, we uncover that this gap can be attributed to two coupled mismatches: (i) update-space mismatch, where clients optimize in a low-rank subspace but aggregation occurs in the full space; and (ii) optimizer-state mismatch, where unsynchronized adaptive states amplify drift across rounds. We propose FedGaLore, which combines client-side GaLore-style gradient-subspace optimization with server-side drift-robust synchronization of projected second-moment states via spectral shared-signal extraction, to address this challenge. Across NLU, vision, and NLG benchmarks, FedGaLore improves robustness and accuracy over state-of-the-art federated LoRA baselines in non-IID settings.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑