用不到5100美元在消费级显卡上训练出接近Qwen2-1.5B性能的20亿参数模型
Puro-2B: Poor Lab's Qwen2-1.5B Trained on RTX 5090 within $5090
- 用RTX 5090+FP8精度在1.4万亿词上从零训练,成本低于6.9万美元
- 最佳模型性能逼近Qwen2-1.5B,成本约4.4万美元即可达成同等水平
- 开源完整训练流程,适合预算有限的研究者复现和研究数据课程影响
语言模型预训练因高昂成本被视作难以企及,使学术与开源社区望而却步。尽管已有开源模型与训练方案,但低成本、可访问硬件且完全开源的预训练方法仍缺失。即使小规模训练如Llama-3.2-3B也需超150万美元,而SmolLM3-3B需逾70万美元。本文提出一个开放的预训练方案,利用该方案在消费级RTX 5090 GPU上从零训练了系列Puro-2B模型,最高使用1.4万亿令牌,采用FP8精度。最佳模型训练成本低于6.9万美元,性能接近Qwen2.5-1.5B。成本优势源于硬件选择、低精度训练、超球优化、课程模型平均与数据策略。我们还推导出Puro成本缩放定律,表明约4.4万美元即可达到Qwen2-1.5B水平。此外,通过端到端分析发现预训练数据课程显著影响下游性能。代码、数据与模型权重已开源,许可为Apache 2.0。
原文摘要 · Abstract (English)
Language model pretraining has become almost synonymous with prohibitive cost, placing it out of reach for much of the academic and open-source communities. Although strong open-source efforts already exist, including open-weight models and open-source training recipes, a cost-efficient, hardware-accessible, and open-source pretraining recipe has long been missing. Even at a small scale, training Llama-3.2-3B costs over \$1.5M, and reproducing SmolLM3-3B needs over \$700K. In this report, we present an open pretraining recipe designed to lower this barrier. Using this recipe, we train a collection of Puro-2B models from scratch on up to 1.4 trillion tokens with FP8 precision on consumer-grade RTX 5090 GPUs. The models in the collection differ in token budgets and selected recipe variants. Our best model is trained at a compute cost of less than \$6.9K and approaches Qwen2.5-1.5B performance under our evaluation protocol. This cost efficiency is enabled by a combination of approaches, including hardware selection, low-precision training, hyperball optimization, curriculum model averaging, and the data recipe. Beyond the recipe itself, we provide two additional results. First, across the Puro-2B collection, we derive a Puro Cost Scaling Law that relates training cost to average model performance; the fitted law suggests that about \$4.4K, less than \$5,090, is sufficient to reach the performance of Qwen2-1.5B. Second, as an end-to-end case study, we examine how pretraining data curricula shape downstream performance after post-training. Such controlled studies are enabled by having access to the full pretraining pipeline rather than model weights alone. We release the full training recipe for Puro-2B, including data, code, and model weights under Apache 2.0 at https://huggingface.co/collections/thu-pacman/puro-2b.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。