arXiv:2503.23869cs.LG2025-03被引 3

通过三矩阵适配降低通信开销,实现个性化联邦微调

Communication-Efficient and Personalized Federated Foundation Model Fine-Tuning via Tri-Matrix Adaptation

  • 用小尺寸稠密矩阵重构LoRA参数,大幅减少通信量
  • 在非独立同分布数据下,性能优于传统方法
  • 结合客户端相似性,提升隐私保护并防梯度反演攻击

在联邦学习中,微调预训练基础模型面临通信开销大和数据异构导致性能不佳的挑战。本文提出通信高效联邦LoRA适配(CE-LoRA),采用三因子低秩适配方法并实现个性化参数聚合。通过引入小尺寸稠密矩阵进行新型LoRA参数分解,显著降低通信成本,且在实验中达到与现有方法相当的性能。服务器在不泄露数据隐私的前提下,同时考虑客户端在训练数据集和模型参数空间中的相似性,学习个性化聚合权重。在多种LLM和VLM微调任务上的实验表明,CE-LoRA不仅显著减少通信开销,还在非独立同分布数据条件下提升模型性能。此外,该方法增强了数据隐私保护,有效抵御基于梯度的数据重构攻击。

原文摘要 · Abstract (English)

In federated learning, fine-tuning pre-trained foundation models poses significant challenges, particularly regarding high communication cost and suboptimal model performance due to data heterogeneity between the clients. To address these issues, this paper introduces communication-efficient federated LoRA adaption (CE-LoRA), a method that employs a tri-factorization low-rank adaptation approach with personalized model parameter aggregation. We first presents a novel LoRA parameter factorization by introducing a small-size dense matrix, which can significantly reduce the communication cost and achieve comparable empirical performance than transferring the low-rank parameter matrix used by existing methods. Without violating data privacy, the server considers the client similarity in both training dataset and model parameter space, and learns personalized weights for model aggregation. Our experiments on various LLM and VLM fine-tuning tasks demonstrate that CE-LoRA not only significantly reduces communication overhead but also improves performance under not independently and identically distributed data conditions. In addition, CE-LoRA improves data privacy protection, effectively mitigating gradient-based data reconstruction attacks.

联邦学习模型微调隐私保护低秩适配

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