arXiv:2510.11274cs.LG2025-10被引 1

针对异构数据下的联邦LoRA微调,区分方向与幅度敏感性实现全局共享与本地个性化协同优化。

FedLoRA-Optimizer: Federated LoRA Fine-Tuning with Global and Local Optimization in Heterogeneous Data Scenarios

  • 按敏感度分层优化LoRA的A矩阵(方向)和B矩阵(幅度)
  • 在异构数据下提升全局性能0.39%,本地性能0.59%
  • 适合需兼顾模型通用性与个性化适配的联邦学习场景

联邦高效微调利用节点间分布数据与计算资源,应对大规模微调与隐私保护挑战。低秩适应(LoRA)通过引入可训练低秩矩阵,实现大模型高效微调。但在异构数据场景中,客户端漂移削弱全局模型泛化能力,且本地模型难以满足个体客户需求。现有联邦LoRA方法缺乏对微调矩阵的细粒度分析。我们初步实验发现,不同LoRA矩阵对向量方向与幅度变化的敏感度不同。为此提出细粒度联邦LoRA优化方法:通过优化编码共享知识的A矩阵的方向向量,更有效学习跨客户端共享特征,增强全局泛化;同时优化编码个性化知识的B矩阵的幅度向量,更好捕捉个性化知识,实现对本地数据的精细适配。方法采用结合全局与局部优化器的流水线架构,进一步提升本地模型性能,实现全局与局部协同优化。在Databricks-Dolly-15k与Natural Instructions数据集上,基于LLaMA2-7B与Deepseek-7B的实验表明,该方法使全局性能提升0.39%,本地性能提升0.59%。

原文摘要 · Abstract (English)

Federated efficient fine-tuning has emerged as an approach that leverages distributed data and computational resources across nodes to address the challenges of large-scale fine-tuning and privacy preservation. The Low-Rank Adaptation (LoRA) enables efficient fine-tuning of large-scale pre-trained models by introducing trainable low-rank matrices into weight updates.However, in heterogeneous data scenarios, client drift weakens the generalization of the global model, and local models often fail to meet the personalized needs of individual clients.Moreover, existing federated LoRA efficient fine-tuning techniques overlook fine-grained analysis of the tuning matrices. To address this, we conducted preliminary experiments and found that different LoRA matrices exhibit different sensitivity to changes in the direction and magnitude of their vectors.We thus propose a fine-grained federated LoRA tuning method. By fine-tuning the more sensitive directional vectors in the A matrix, which encode shared knowledge, our method learns shared features more effectively across clients and enhances global generalization. Simultaneously, by fine-tuning the more sensitive magnitude vectors in the B matrix, which encode personalized knowledge, our method better captures personalized knowledge, enabling detailed adaptation to local data. The method uses a pipeline combining global and local optimizers. Global optimization further improves local models, achieving collaborative optimization between global and local levels. This improves both the generalization ability of the global model and the personalized adaptation of local models under heterogeneous data scenarios. Experiments on Databricks-Dolly-15k and Natural Instructions with LLaMA2-7B and Deepseek-7B confirm that our method improves global performance by 0.39% and local performance by 0.59%.

联邦学习LoRA微调优化

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