用鲁棒PCA分离共性与个性,提升联邦LoRA的聚合效果
FedRPCA: Enhancing Federated LoRA Aggregation Using Robust PCA
- 通过鲁棒PCA分解客户端更新为共性低秩项与个性稀疏项
- 在多任务视觉语言模型上实现更快收敛和更高精度
- 适合资源受限下的联邦微调场景,尤其数据异构时
LoRA作为一种高效的微调技术,在资源受限的联邦学习中广受关注,能显著降低通信与计算开销。然而,数据异构性仍是基于LoRA的联邦学习主要挑战,传统基于FedAvg的聚合策略存在收敛慢、精度不足的问题。受近期模型融合方法(如Task Arithmetic)启发,我们探索使用缩放平均法聚合客户端LoRA参数。发现直接应用该方法效果不佳,因客户端更新间余弦相似度高,表明存在大量共性知识。为此,我们提出通过鲁棒主成分分析(Robust-PCA)将客户端LoRA更新分解为共同的低秩成分与客户端特异的稀疏成分。算法对低秩成分进行平均以整合共性知识,对稀疏成分采用缩放平均以强化个性化知识。在多个视觉与语言任务上评估表明,相比基线方法,该方法实现更高最终精度和更快收敛速度。
原文摘要 · Abstract (English)
LoRA has emerged as one of the most promising fine-tuning techniques, especially for federated learning (FL), since it significantly reduces communication and computation costs at resource-constrained clients. However, data heterogeneity remains a significant challenge for LoRA-based FL, and the conventional aggregation strategy based on FedAvg suffers from slow convergence and suboptimal accuracy. Motivated by recent advances in model merging, particularly Task Arithmetic, we explore the idea of aggregating client LoRA parameters using scaled averaging. We first observe that a naive application of Task Arithmetic is ineffective due to the high cosine similarity between client updates, indicating significant common knowledge in the updates across clients. To address this issue, we propose decomposing client LoRA updates via Robust Principal Component Analysis (Robust-PCA) into a common low-rank component and client-specific sparse components. Our proposed algorithm FedRPCA aggregates the low-rank components through averaging, consolidating common knowledge, and applies scaled averaging to the sparse components to amplify client-specific knowledge. We evaluate our approach across a variety of vision and language tasks and demonstrate that it achieves higher final accuracy and faster convergence compared to competing baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。