arXiv:2410.09671cs.AIcs.CL2024-10被引 73

开源框架OpenR提升大模型推理能力,支持强化学习与分步思考。

OpenR: An Open Source Framework for Advanced Reasoning with Large Language Models

  • 整合测试时计算、强化学习与过程监督,增强大模型推理
  • 在MATH数据集上实现显著性能提升,相对改进明显
  • 适合研究大模型推理与强化学习的开发者与研究人员

本文介绍OpenR,一个开源框架,旨在集成关键组件以提升大语言模型(LLMs)的推理能力。OpenR将数据获取、强化学习训练(在线与离线)及非自回归解码统一为一个软件平台。受OpenAI o1模型成功启发,该框架融合测试时计算、强化学习与过程监督,推动大模型推理进步。这是首个开源探索o1核心方法的框架,超越传统自回归方式,实现高级推理。我们在MATH数据集上验证其有效性,使用公开数据与搜索方法,初步实验显示,通过过程奖励模型实现的测试时计算和强化学习带来显著性能提升。OpenR代码、模型与数据集已开放:https://openreasoner.github.io。

原文摘要 · Abstract (English)

In this technical report, we introduce OpenR, an open-source framework designed to integrate key components for enhancing the reasoning capabilities of large language models (LLMs). OpenR unifies data acquisition, reinforcement learning training (both online and offline), and non-autoregressive decoding into a cohesive software platform. Our goal is to establish an open-source platform and community to accelerate the development of LLM reasoning. Inspired by the success of OpenAI's o1 model, which demonstrated improved reasoning abilities through step-by-step reasoning and reinforcement learning, OpenR integrates test-time compute, reinforcement learning, and process supervision to improve reasoning in LLMs. Our work is the first to provide an open-source framework that explores the core techniques of OpenAI's o1 model with reinforcement learning, achieving advanced reasoning capabilities beyond traditional autoregressive methods. We demonstrate the efficacy of OpenR by evaluating it on the MATH dataset, utilising publicly available data and search methods. Our initial experiments confirm substantial gains, with relative improvements in reasoning and performance driven by test-time computation and reinforcement learning through process reward models. The OpenR framework, including code, models, and datasets, is accessible at https://openreasoner.github.io.

大模型推理强化学习开源框架

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