arXiv:2602.11543cs.CL2026-02

用少量显存GPU训练大模型,实现高效分布式预训练。

Pretraining A Large Language Model using Distributed GPUs: A Memory-Efficient Decentralized Paradigm

  • 每节点只训练部分专家,大幅降低显存占用。
  • 16块48GB显卡即可训练20亿参数MoE模型,性能媲美中心化训练。
  • 适合资源有限但想训练大模型的研究者或小型团队。

预训练大语言模型通常依赖数千块高显存GPU的集中式集群(如H100/A100)。现有去中心化训练方法虽减少通信开销,但仍需在每个节点上完整训练模型,受显存限制。本文提出SPES(稀疏专家同步)框架,用于高效去中心化预训练混合专家(MoE)模型。该方法让每节点仅训练部分专家,显著降低内存需求;节点本地更新专家,并周期性与其他节点同步,避免传输全部参数,实现知识高效共享。为缓解稀疏更新导致的专家数据利用不足问题,引入专家合并预热策略,早期即促进专家间知识交换,快速建立基础能力。使用SPES,我们仅用16块独立的48GB GPU通过互联网连接训练出一个20亿参数的MoE模型,在相同计算预算下性能接近中心化训练模型。进一步验证了可扩展性:从头训练70亿参数模型,以及将90亿参数密集模型转换为稀疏模型,均达到先前集中式基线水平。代码已开源:https://github.com/zjr2000/SPES。

原文摘要 · Abstract (English)

Pretraining large language models (LLMs) typically requires centralized clusters with thousands of high-memory GPUs (e.g., H100/A100). Recent decentralized training methods reduce communication overhead by employing federated optimization; however, they still need to train the entire model on each node, remaining constrained by GPU memory limitations. In this work, we propose SParse Expert Synchronization (SPES), a memory-efficient decentralized framework for pretraining mixture-of-experts (MoE) LLMs. SPES trains only a subset of experts per node, substantially lowering the memory footprint. Each node updates its local experts and periodically synchronizes with other nodes, eliminating full-parameter transmission while ensuring efficient knowledge sharing. To mitigate limited per-expert data utilization under sparse expert updates, we introduce an expert-merging warm-up strategy, where experts exchange knowledge early in training, to rapidly establish foundational capabilities. With SPES, we train a 2B-parameter MoE LLM using 16 standalone 48GB GPUs over internet connections, which achieves competitive performance with centrally trained LLMs under similar computational budgets. We further demonstrate scalability by training a 7B model from scratch and a 9B model upcycled from a dense checkpoint, both of which match prior centralized baselines. Our code is available at https://github.com/zjr2000/SPES.

MoE分布式训练显存优化大模型

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