arXiv:2411.02908cs.LGcs.DC2024-11

首个支持全球分布式训练大模型的系统,通信量减少512倍且效率更高。

Photon: Federated LLM Pre-Training

  • 采用联邦平均+小批量高学习率策略,降低通信开销。
  • 70亿参数模型在联邦训练下困惑度优于中心化训练。
  • 适合资源受限但需大规模协作训练的团队使用。

扩展大语言模型(LLM)需要大量数据与算力,传统分布式训练受限于高带宽需求,仅能在数据中心内进行。低带宽的联邦学习(FL)若能用于预训练,则可在弱连接设备间实现更大规模的协同训练。为此,我们提出Photon——首个完整的联邦端到端LLM训练系统,利用跨站点联邦学习实现全局规模训练,通信开销极低。基于Photon,我们首次从零训练出联邦式解码器单向LLM家族。实验表明:(1) Photon可在联邦模式下训练最大达70亿参数的模型,并达到比中心化预训练更低的困惑度;(2) 训练时间随可用算力增加而下降,表现出与集中式训练相似的算力-时间权衡;(3) 相较于基线分布式训练方法,通信量减少64至512倍,节省35%运行时间。该方法对数据异构性鲁棒,收敛速度是前序方法如DiLoCo的两倍。这一惊人数据效率源于小客户端批次与极高学习率结合,得益于联邦平均对超参数的强鲁棒性。Photon因此成为首个经济可行的互联网级大模型预训练系统。

原文摘要 · Abstract (English)

Scaling large language models (LLMs) demands extensive data and computing resources, which are traditionally constrained to data centers by the high-bandwidth requirements of distributed training. Low-bandwidth methods like federated learning (FL) could enable collaborative training of larger models across weakly-connected GPUs if they can effectively be used for pre-training. To achieve this, we introduce Photon, the first complete system for federated end-to-end LLM training, leveraging cross-silo FL for global-scale training with minimal communication overheads. Using Photon, we train the first federated family of decoder-only LLMs from scratch. We show that: (1) Photon can train model sizes up to 7B in a federated fashion while reaching an even better perplexity than centralized pre-training; (2) Photon model training time decreases with available compute, achieving a similar compute-time trade-off to centralized; and (3) Photon outperforms the wall-time of baseline distributed training methods by 35% via communicating 64x-512xless. Our proposal is robust to data heterogeneity and converges twice as fast as previous methods like DiLoCo. This surprising data efficiency stems from a unique approach combining small client batch sizes with extremely high learning rates, enabled by federated averaging's robustness to hyperparameters. Photon thus represents the first economical system for global internet-wide LLM pre-training.

联邦学习大模型训练通信优化高效预训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。