提出可自适应调整更新范围的联邦微调框架,解决异构设备训练大模型难题。
Federated Sketching LoRA: A Flexible Framework for Heterogeneous Collaborative Fine-Tuning of LLMs
- 客户端仅更新全局LoRA模块的子矩阵,通过调节采样比例灵活适配算力
- 在多个数据集上性能优于基线,通信量减少30%且收敛稳定
- 适合资源差异大的边缘设备协同微调大模型,尤其适合移动端部署
在资源受限客户端上微调大语言模型仍具挑战性。现有工作将低秩适配(LoRA)与联邦微调结合以缓解模型尺寸和数据稀缺问题,但客户端异构性仍是关键瓶颈:高阶模块虽提升性能,但不同客户端能力限制了可选的LoRA秩范围。现有方法或缺乏理论支撑,或增加额外计算开销。为此,我们提出联邦草图LoRA(FSLoRA),利用草图机制使客户端选择性更新由服务器维护的全局LoRA模块的子矩阵。通过调整决定子矩阵秩的草图比例,FSLoRA可灵活适应客户端的通信与计算约束。我们提供了严格的收敛性分析,揭示草图比例如何影响收敛速度。大量实验表明,FSLoRA优于基线,在显著提升训练效率的同时保持稳定收敛。
原文摘要 · Abstract (English)
Fine-tuning large language models (LLMs) on resource-constrained clients remains a challenging problem. Recent works have fused low-rank adaptation (LoRA) techniques with federated fine-tuning to mitigate challenges associated with client model sizes and data scarcity. Still, the heterogeneity of resources remains a critical bottleneck: while higher-rank modules generally enhance performance, varying client capabilities constrain LoRA's feasible rank range. Existing approaches attempting to resolve this issue either lack analytical justification or impose additional computational overhead, leaving a wide gap for efficient and theoretically-grounded solutions. To address these challenges, we propose federated sketching LoRA (FSLoRA), which leverages a sketching mechanism to enable clients to selectively update submatrices of global LoRA modules maintained by the server. By adjusting the sketching ratios, which determine the ranks of the submatrices on the clients, FSLoRA flexibly adapts to client-specific communication and computational constraints. We provide a rigorous convergence analysis of FSLoRA that characterizes how the sketching ratios affect the convergence rate. Through extensive experiments, we demonstrate that FSLoRA outperforms baselines and significantly improves training efficiency while preserving stable convergence.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。