Kimi k1.5:通过强化学习扩展大语言模型

Kimi k1.5: Scaling Reinforcement Learning with LLMs

杨植麟 Zhilin Yang · Moonshot AI · 2025-01-22 · arXiv:2501.12599 ↗ · 被引 992

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

基于下一个词预测的语言模型预训练在扩展计算方面已被证明有效,但受限于可用训练数据量。扩展强化学习为人工智能的持续改进开辟了新维度,其前景在于大语言模型可以通过学习基于奖励进行探索来扩展训练数据。然而,先前已发表的工作并未产生有竞争力的结果。鉴于此,我们报告了 Kimi k1.5(我们最新的多模态大语言模型,通过强化学习训练)的训练实践,包括其强化学习训练技术、多模态数据方案和基础设施优化。长上下文扩展和改进的策略优化方法是我们方法的关键要素,它建立了一个简单有效的强化学习框架,无需依赖更复杂的技术,如蒙特卡洛树搜索、价值函数和过程奖励模型。值得注意的是,我们的系统在多个基准和模态上实现了最先进的推理性能——例如,AIME 77.5,MATH 500 96.2,Codeforces 94 百分位,MathVista 74.9——与 OpenAI 的 o1 相当。此外,我们提出了有效的长转短方法,利用长思维链技术改进短思维链模型,产生了最先进的短思维链推理结果——例如,AIME 60.8,MATH500 94.6,LiveCodeBench 47.3——大幅超越现有的短思维链模型,如 GPT-4o 和 Claude Sonnet 3.5(提升高达+550%)。

Language model pretraining with next token prediction has proved effective for scaling compute but is limited to the amount of available training data. Scaling reinforcement learning (RL) unlocks a new axis for the continued improvement of artificial intelligence, with the promise that large language models (LLMs) can scale their training data by learning to explore with rewards. However, prior published work has not produced competitive results. In light of this, we report on the training practice of Kimi k1.5, our latest multi-modal LLM trained with RL, including its RL training techniques, multi-modal data recipes, and infrastructure optimization. Long context scaling and improved policy optimization methods are key ingredients of our approach, which establishes a simplistic, effective RL framework without relying on more complex techniques such as Monte Carlo tree search, value functions, and process reward models. Notably, our system achieves state-of-the-art reasoning performance across multiple benchmarks and modalities -- e.g., 77.5 on AIME, 96.2 on MATH 500, 94-th percentile on Codeforces, 74.9 on MathVista -- matching OpenAI's o1. Moreover, we present effective long2short methods that use long-CoT techniques to improve short-CoT models, yielding state-of-the-art short-CoT reasoning results -- e.g., 60.8 on AIME, 94.6 on MATH500, 47.3 on LiveCodeBench -- outperforming existing short-CoT models such as GPT-4o and Claude Sonnet 3.5 by a large margin (up to +550%).

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 15)

阅读逐段中英对照全文 →