arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2409.06277cs.LGcs.AI
收藏

Ferret:面向大语言模型的大规模联邦全参数微调

Ferret: Federated Full-Parameter Tuning at Scale for Large Language Models

SAP(SAP公司);Guangdong Laboratory of Artificial Intelligence and Digital Economy (SZ)(广东人工智能与数字经济实验室(深圳))

Yao Shu, Wenyang Hu, See-Kiong Ng, Bryan Kian Hsiang Low, Fei Richard Yu

AI总结:

本文提出 Ferret,一种结合一阶本地更新、低维投影与共享随机重构的联邦全参数微调方法,在降低通信和计算开销的同时保持 LLM 精度与快速收敛。

摘要:

大语言模型(LLM)已在众多实际应用中变得不可或缺。然而,对这些模型进行大规模微调面临重大挑战,尤其是在数据隐私和通信效率至关重要的联邦场景中。现有方法通常借助参数高效微调(PEFT)来降低通信开销,但这往往以牺牲模型准确性为代价。为此,我们提出面向 LLM 的大规模联邦全参数微调方法 Ferret,这是首个采用共享随机性的一阶方法,能够在去中心化数据源上实现可扩展的 LLM 全参数微调,同时保持具有竞争力的模型准确性。Ferret 通过三个方面实现这一目标:(i)采用广泛使用的一阶方法进行高效的本地更新;(ii)将这些更新投影到低维空间,以显著降低通信开销;(iii)利用共享随机性从该低维空间重构本地更新,从而实现有效的全参数全局聚合,确保快速收敛和具有竞争力的最终性能。我们严谨的理论分析与洞见以及大量实验表明,Ferret 显著提升了现有联邦全参数微调方法的可扩展性,实现了高计算效率、更低的通信开销和快速收敛,同时保持了具有竞争力的模型准确性。我们的实现可在 https://github.com/allen4747/Ferret 获取。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑