DECA让大模型在分散数据下高效全参数微调,兼顾性能与资源节省。
DECA: Decentralizing Block-Wise Adam for Efficient LLM Full-Parameter Fine-Tuning on Non-IID Data
- 分块顺序更新参数,降低资源消耗
- 收敛快,下游任务表现强,比现有方法提升12%以上
- 适合隐私敏感、算力有限的分布式场景
在隐私敏感且资源受限的环境中微调大语言模型仍具挑战。由于训练数据通常分布在多个客户端,去中心化微调提供了一种无需中央服务器的协作适应范式。然而,在去中心化环境下实现全参数微调(FPFT)困难重重:尽管FPFT具备强大适应能力,但对百亿规模模型而言资源开销巨大。现有去中心化大模型微调方法主要依赖参数高效更新,虽提升效率但可能限制下游性能。此外,客户端数据通常非独立同分布(non-IID),使去中心化优化更易受客户端漂移和收敛不稳影响。为此,我们提出DECA,一种针对非IID数据的资源高效去中心化全参数微调框架。DECA将模型参数划分为互斥块,采用顺序分块Adam优化,在保持去中心化全参数适配的同时降低资源消耗。为稳定训练,DECA引入一阶与二阶分块动量估计,结合新鲜本地梯度统计与共识衍生的差异信号。我们提供了严格的理论分析和大量实验,结果表明DECA实现了快速收敛、强下游性能及显著资源效率。
原文摘要 · Abstract (English)
Fine-tuning large language models (LLMs) in privacy-sensitive and resource-constrained environments remains challenging. Since training data are often distributed across multiple clients, decentralized fine-tuning offers a natural paradigm for collaborative adaptation without a central server. However, enabling full-parameter fine-tuning (FPFT) in this decentralized setting is difficult: FPFT provides strong adaptation capacity but incurs prohibitive resource consumption for billion-scale models. Existing decentralized LLM fine-tuning methods therefore mainly rely on parameter-efficient updates, which improve efficiency but may restrict downstream performance. Moreover, client data are typically non-IID, making decentralized optimization more vulnerable to client drift and unstable convergence. To address these challenges, we propose DECA, a resource-efficient decentralized FPFT framework for LLMs on non-IID data. DECA partitions model parameters into disjoint blocks and performs sequential block-wise Adam optimization, reducing resource consumption while preserving decentralized full-parameter adaptation. To stabilize training, DECA further introduces first- and second-order block-wise moment estimates with fresh local gradient statistics and consensus-derived discrepancy signals. We provide rigorous theoretical analysis and extensive experiments, showing that DECA achieves fast convergence, strong downstream performance, and significant resource efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。