对比发现直接权重平均比LoRA更适配联邦微调大模型。
Exploring Gradient Subspaces: Addressing and Overcoming LoRA's Limitations in Federated Fine-Tuning of Large Language Models
- 用梯度子空间分析揭示LoRA在联邦学习中因低秩约束导致性能受限。
- 实验证明直接权重平均优于LoRA及其变体,跨文本与图像任务均更优。
- 适合关注联邦学习中高效微调方法的科研人员和工程实践者。
大语言模型(LLMs)在多领域展现出卓越能力,尤其在文本与视觉任务上的泛化性能突出。尽管微调能显著提升其下游任务表现,但高质量数据常因隐私问题无法共享。联邦学习(FL)提供了无需直接共享数据的协作训练方案。然而,当前基于低秩适应(LoRA)的参数高效微调方法在联邦设置中存在局限性。本文系统分析了采用LoRA的主流联邦框架的收敛性与性能保障,指出其因低秩矩阵子空间学习受限而表现不佳。通过严格的理论与实证评估,我们发现直接权重平均优于LoRA策略,显著提升微调模型性能。进一步分析显示,结合局部训练中的低秩梯度优化器如GaLore,配合直接权重聚合,相比FlexLoRA、FFA-LoRA等联邦LoRA方法,在文本与图像模态上均表现更优。尽管隐私保护是联邦学习的核心,本研究聚焦于评估不同框架下微调模型的性能表现,从理论与实证角度重新审视对LoRA的依赖,推动更高效的训练范式发展。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have demonstrated remarkable capabilities across various domains, particularly in task generalization for both text and vision data. While fine-tuning these models can significantly enhance their performance on specific downstream tasks, it often requires high-quality data that cannot be shared due to privacy concerns. Federated Learning (FL) offers a promising solution for collaborative training without direct data sharing. However, many parameter-efficient fine-tuning strategies for LLMs in FL, particularly those based on Low-Rank Adaptation (LoRA), face limitations. In this paper, we critically analyze the convergence and performance guarantees of popular FL frameworks utilizing LoRA, highlighting its suboptimal nature due to constrained subspace learning of low-rank matrices. This limitation hinders effective fine-tuning of LLMs in federated settings. Through rigorous analytical and empirical evaluations, we demonstrate that direct weight averaging outperforms LoRA-based strategies, leading to superior performance for fine-tuned models. Our comprehensive comparison unmasks inefficiencies in LoRA approaches and underscores the advantages of direct weight aggregation. We extend our analysis to low-rank gradient-based optimizers, such as GaLore, used during local training steps. Our findings show that GaLore along with direct-weight aggregation is a more effective approach, outperforming federated LoRA methods like FlexLoRA and FFA-LoRA across both text and image modalities. While privacy remains paramount in FL discourse, our focus is on assessing performance outcomes of federated fine-tuned models and evaluating various FL frameworks from both theoretical and empirical perspectives. Our findings advocate reassessing the reliance on LoRA within FL contexts, paving the way for more efficient training methodologies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。