arXiv:2501.12599cs.AIcs.LG2025-01被引 1.1k

用强化学习提升大模型推理能力,效果媲美顶级模型。

Kimi k1.5: Scaling Reinforcement Learning with LLMs

  • 通过长上下文与优化策略,构建简洁有效的强化学习框架
  • 在AIME、MATH500等多任务上达到顶尖水平,部分超OpenAI o1
  • 提出长思维链转短思维链方法,显著提升短链模型表现

基于下一个词预测的语言模型预训练虽能有效扩展算力,但受限于可用数据量。强化学习(RL)则为人工智能持续提升开辟了新路径,使大语言模型(LLMs)可通过奖励机制自主探索以扩展训练数据。然而此前研究未取得竞争性成果。本文报告了我们最新多模态大模型Kimi k1.5的强化学习训练实践,涵盖其训练技术、多模态数据配方及基础设施优化。长上下文扩展与改进的策略优化是核心,构建了一个无需蒙特卡洛树搜索、价值函数或过程奖励模型的简洁高效框架。系统在多个基准和模态上实现领先性能——如AIME达77.5,MATH 500达96.2,Codeforces第94百分位,MathVista达74.9,媲美OpenAI o1。此外,我们提出有效的long2short方法,利用长思维链技术提升短思维链模型,实现AIME 60.8、MATH500 94.6、LiveCodeBench 47.3的优异结果,相比GPT-4o与Claude Sonnet 3.5提升高达550%。

原文摘要 · Abstract (English)

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%).

强化学习大模型推理长思维链多模态

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