arXiv:2503.05139cs.LGcs.AI2025-03被引 23

用普通设备训练3000亿参数专家混合模型,性能媲美高端设备。

Every FLOP Counts: Scaling a 300B Mixture-of-Experts LING LLM without Premium GPUs

  • 用优化架构与训练流程降低资源消耗
  • 2900亿参数模型仅激活288亿,训练成本降20%
  • 适合资源有限但想做大规模模型的研究者

本文针对大规模专家混合(MoE)模型训练中的成本低效与资源限制问题,提出两个不同规模的MoE大语言模型:Ling-Lite(168亿参数,激活27.5亿)与Ling-Plus(2900亿参数,激活288亿),性能可比业界领先模型。通过优化模型架构、改进训练异常处理、提升评估效率,实现高效训练。利用知识图谱生成的高质量数据,模型在工具使用能力上表现更优。实验表明,3000亿参数的MoE模型可在低性能设备上完成预训练,相较高性能设备节省约20%计算成本,仍保持相当性能。模型开源可访问:https://huggingface.co/inclusionAI。

原文摘要 · Abstract (English)

In this technical report, we tackle the challenges of training large-scale Mixture of Experts (MoE) models, focusing on overcoming cost inefficiency and resource limitations prevalent in such systems. To address these issues, we present two differently sized MoE large language models (LLMs), namely Ling-Lite and Ling-Plus (referred to as "Bailing" in Chinese, spelled Bǎilíng in Pinyin). Ling-Lite contains 16.8 billion parameters with 2.75 billion activated parameters, while Ling-Plus boasts 290 billion parameters with 28.8 billion activated parameters. Both models exhibit comparable performance to leading industry benchmarks. This report offers actionable insights to improve the efficiency and accessibility of AI development in resource-constrained settings, promoting more scalable and sustainable technologies. Specifically, to reduce training costs for large-scale MoE models, we propose innovative methods for (1) optimization of model architecture and training processes, (2) refinement of training anomaly handling, and (3) enhancement of model evaluation efficiency. Additionally, leveraging high-quality data generated from knowledge graphs, our models demonstrate superior capabilities in tool use compared to other models. Ultimately, our experimental findings demonstrate that a 300B MoE LLM can be effectively trained on lower-performance devices while achieving comparable performance to models of a similar scale, including dense and MoE models. Compared to high-performance devices, utilizing a lower-specification hardware system during the pre-training phase demonstrates significant cost savings, reducing computing costs by approximately 20%. The models can be accessed at https://huggingface.co/inclusionAI.

MoE模型低成本训练大模型资源受限

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