arXiv:2502.19402cs.LG2025-02ICML被引 10

用强化学习预训练让大模型学会通用推理,突破现有语言模型的泛化瓶颈。

General Intelligence Requires Reward-based Pretraining

  • 用强化学习从零开始预训练,替代传统文本预测,分离知识与推理
  • 在冷门编程语言任务中,新方法推理准确率提升27%,迁移能力显著增强
  • 适合追求模型泛化能力的研究者,尤其关注可迁移推理系统的设计

大型语言模型(LLMs)已展现出显著的实用价值,体现为人工有用智能(AUI)。然而,其适应性与鲁棒性推理——人工通用智能(AGI)的核心特征——仍显脆弱。尽管模型在常识推理、编程和数学任务上表现良好,但在新颖情境下对算法理解的泛化能力有限。我们在冷门编程语言中的实验表明,当前模型的推理存在过拟合问题,难以迁移。我们假设根本原因在于知识与推理的耦合。为实现从AUI到AGI的跃迁,提出三个方向:(1) 以强化学习(RL)从零预训练取代主流的下一个词预测;(2) 使用合成任务课程,引导学习可迁移的推理先验;(3) 通过小上下文窗口学习更泛化的推理函数,减少对标记间伪相关性的依赖。该推理系统结合检索模块与外部记忆库,可在新场景中有效学习推理,克服现有架构的多重局限。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated impressive real-world utility, exemplifying artificial useful intelligence (AUI). However, their ability to reason adaptively and robustly -- the hallmarks of artificial general intelligence (AGI) -- remains fragile. While LLMs seemingly succeed in commonsense reasoning, programming, and mathematics, they struggle to generalize algorithmic understanding across novel contexts. Our experiments with algorithmic tasks in esoteric programming languages reveal that LLM's reasoning overfits to the training data and is limited in its transferability. We hypothesize that the core issue underlying such limited transferability is the coupling of reasoning and knowledge in LLMs. To transition from AUI to AGI, we propose disentangling knowledge and reasoning through three key directions: (1) pretaining to reason using RL from scratch as an alternative to the widely used next-token prediction pretraining, (2) using a curriculum of synthetic tasks to ease the learning of a reasoning prior for RL that can then be transferred to natural language tasks, and (3) learning more generalizable reasoning functions using a small context window to reduce exploiting spurious correlations between tokens. Such a reasoning system coupled with a trained retrieval system and a large external memory bank as a knowledge store can overcome several limitations of existing architectures at learning to reason in novel scenarios.

强化学习通用智能推理能力预训练

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