让大模型在分散数据上全参数微调,又快又省通信
Ferret: Federated Full-Parameter Tuning at Scale for Large Language Models
- 用低维投影压缩更新信息,降低通信开销
- 共享随机性重建参数,实现全局有效聚合
- 支持大规模分布式训练,精度媲美传统方法
大型语言模型(LLMs)已在众多实际应用中不可或缺。然而,在数据隐私与通信效率至关重要的联邦设置下进行大规模微调仍面临重大挑战。现有方法多采用参数高效微调(PEFT)以减少通信开销,但通常以牺牲模型精度为代价。为此,我们提出联邦全参数微调框架Ferret,是首个基于一阶优化且具备共享随机性的方法,可在去中心化数据源上实现大规模LLM的全参数微调,同时保持竞争性模型性能。Ferret通过三方面实现:(i) 使用广泛的一阶优化方法实现高效的本地更新;(ii) 将这些更新投影到低维空间,显著降低通信开销;(iii) 利用共享随机性从低维空间重构本地更新,支持有效的全参数全局聚合,确保快速收敛与优异最终表现。严格的理论分析与大量实验表明,Ferret显著提升了现有联邦全参数微调方法的可扩展性,实现高计算效率、低通信开销与快速收敛,同时保持竞争力的模型精度。代码已开源:https://github.com/allen4747/Ferret。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have become indispensable in numerous real-world applications. However, fine-tuning these models at scale, especially in federated settings where data privacy and communication efficiency are critical, presents significant challenges. Existing approaches often resort to parameter-efficient fine-tuning (PEFT) to mitigate communication overhead, but this typically comes at the cost of model accuracy. To this end, we propose federated full-parameter tuning at scale for LLMs (Ferret), the first first-order method with shared randomness to enable scalable full-parameter tuning of LLMs across decentralized data sources while maintaining competitive model accuracy. Ferret accomplishes this through three aspects: (i) it employs widely used first-order methods for efficient local updates; (ii) it projects these updates into a low-dimensional space to considerably reduce communication overhead; and (iii) it reconstructs local updates from this low-dimensional space with shared randomness to facilitate effective full-parameter global aggregation, ensuring fast convergence and competitive final performance. Our rigorous theoretical analyses and insights along with extensive experiments, show that Ferret significantly enhances the scalability of existing federated full-parameter tuning approaches by achieving high computational efficiency, reduced communication overhead, and fast convergence, all while maintaining competitive model accuracy. Our implementation is available at https://github.com/allen4747/Ferret.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。