arXiv:2505.13718cs.AIcs.CL2025-05EMNLP被引 1

用简单逻辑题预训练,让小数据也能练出强推理能力

Warm Up Before You Train: Unlocking General Reasoning in Resource-Constrained Settings

  • 先用少量逻辑谜题预训练模型,学通用推理技能
  • 在仅100个样本下,预训练后模型性能显著提升
  • 适合数据少但需强推理的场景,如低资源AI研发

构建具备推理能力的大模型通常依赖强化学习结合可验证奖励(RLVR)或精心设计的长思维链(CoT)蒸馏,二者均需大量高质量训练数据。当数据稀缺时,这一需求成为瓶颈。本文提出一种两阶段、样本高效的训练策略:第一阶段通过在小型逻辑谜题(骑士与说谎者,K&K)领域蒸馏长思维链,对模型进行“预热”,使其掌握通用推理能力;第二阶段在有限目标领域数据上对预热后的模型应用RLVR。实验表明:(i) 预热阶段本身即可提升跨任务泛化能力,在MATH、HumanEval⁺和MMLU-Pro等任务中表现更优;(ii) 当基线模型与预热模型均在≤100例数据上进行RLVR训练时,预热模型始终优于基线;(iii) 预热可保持模型跨域泛化性,即使经过特定领域训练也未退化;(iv) 预热显著提升整体样本效率,不仅提高准确率,还减少训练所需数据量。结果表明,预热策略在数据受限环境下具有显著潜力。

原文摘要 · Abstract (English)

Designing effective reasoning-capable LLMs typically requires training using Reinforcement Learning with Verifiable Rewards (RLVR) or distillation with carefully curated Long Chain of Thoughts (CoT), both of which depend heavily on extensive training data. This creates a major challenge when the amount of quality training data is scarce. We propose a sample-efficient, two-stage training strategy to develop reasoning LLMs under limited supervision. In the first stage, we "warm up" the model by distilling Long CoTs from a toy domain, namely, Knights \& Knaves (K\&K) logic puzzles to acquire general reasoning skills. In the second stage, we apply RLVR to the warmed-up model using a limited set of target-domain examples. Our experiments demonstrate that this two-phase approach offers several benefits: $(i)$ the warmup phase alone facilitates generalized reasoning, leading to performance improvements across a range of tasks, including MATH, HumanEval$^{+}$, and MMLU-Pro; $(ii)$ When both the base model and the warmed-up model are RLVR trained on the same small dataset ($\leq100$ examples), the warmed-up model consistently outperforms the base model; $(iii)$ Warming up before RLVR training allows a model to maintain cross-domain generalizability even after training on a specific domain; $(iv)$ Introducing warmup in the pipeline improves not only accuracy but also overall sample efficiency during RLVR training. The results in this paper highlight the promise of warmup for building robust reasoning LLMs in data-scarce environments.

推理增强小样本学习模型预热强化学习

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