arXiv:2509.24945cs.CLcs.AI2025-09被引 9

用不到十分之一的数据训练出强推理小模型,打破大模型需海量数据的固有认知

MobileLLM-R1: Exploring the Limits of Sub-Billion Language Model Reasoners with Open Training Recipes

  • 精选2万亿高质量开源数据,通过重采样构建高效训练集
  • 仅用4.2万亿token预训练,9.5亿参数模型在推理任务上达15.5分AIME
  • 开源完整训练配方,适合研究轻量级推理模型的开发者复现

大型语言模型从直觉式回应转向链式思维(CoT)推理的范式转变,催生了两个普遍假设:(1)推理能力仅在足够大的模型中出现;(2)此类能力需依赖极大规模数据集训练。尽管前者已被如Qwen3-0.6B和DeepSeek精简版等子十亿参数模型挑战,后者仍未被质疑。本文重新审视推理能力是否必须依赖超大规模语料(>10万亿词元)。通过设计评估指标并筛选有益的开源数据集,我们发现仅约2万亿高质量词元即可支撑强推理能力涌现。具体而言,基于这些数据重采样后进行4.2万亿词元预训练,并结合成熟后训练流程,我们开发出MobileLLM-R1系列子十亿参数推理模型,显著优于此前完全开源数据训练的模型。例如,MobileLLM-R1-950M在推理任务上取得15.5的AIME得分,远超OLMo-2-1.48B的0.6和SmolLM-2-1.7B的0.3。尤为突出的是,尽管仅使用了Qwen3私有36万亿词元语料的11.7%进行预训练,MobileLLM-R1-950M在多个推理基准上达到或超越Qwen3-0.6B表现。为推动该方向研究,我们已公开模型(https://huggingface.co/collections/facebook/mobilellm-r1)、代码(https://github.com/facebookresearch/MobileLLM-R1)及完整训练配方、数据来源与混合比例。

原文摘要 · Abstract (English)

The paradigm shift in large language models (LLMs) from instinctive responses to chain-of-thought (CoT) reasoning has fueled two prevailing assumptions: (1) reasoning capabilities only emerge in sufficiently large models, and (2) such capabilities require training on massive datasets. While the first assumption has already been challenged by recent sub-billion-parameter reasoning models such as Qwen3-0.6B and DeepSeek distilled variants, the second remains largely unquestioned. In this work, we revisit the necessity of scaling to extremely large corpora (>10T tokens) for reasoning emergence. By carefully curating and resampling open-source datasets that we identify as beneficial under our designed metrics, we demonstrate that strong reasoning abilities can emerge with far less data. Specifically, we show that only ~2T tokens of high-quality data are sufficient, and pre-training with 4.2T tokens on the dataset resampled from these ~2T tokens, followed by a established post-training procedure, enables the development of MobileLLM-R1, a series of sub-billion-parameter reasoning models that substantially outperform prior models trained on fully open-sourced data. For example, MobileLLM-R1-950M achieves an AIME score of 15.5, compared to just 0.6 for OLMo-2-1.48B and 0.3 for SmolLM-2-1.7B. Remarkably, despite being trained on only 11.7% of the tokens compared to Qwen3's proprietary 36T-token corpus for pretraining, MobileLLM-R1-950M matches or surpasses Qwen3-0.6B across multiple reasoning benchmarks. To facilitate further research in this direction, we have made the models (https://huggingface.co/collections/facebook/mobilellm-r1) and code (https://github.com/facebookresearch/MobileLLM-R1) publicly available, along with the complete training recipe, data sources, and data mixing ratios.

小模型推理数据效率开源训练轻量化模型

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