arXiv:2410.08048cs.LGcs.CL2024-10被引 14

用强化学习提升大模型推理能力,让验证器更智能。

VerifierQ: Enhancing LLM Test Time Compute with Q-Learning-based Verifiers

  • 引入离线Q-learning优化验证器,改进推理判断
  • 在数学推理任务中准确率与效率显著优于传统方法
  • 适合关注大模型推理鲁棒性的研究人员

近期测试时计算的进展,特别是通过验证器模型,显著提升了大语言模型(LLMs)的推理能力。该生成-验证框架类似于强化学习中的演员-评论家结构。然而,当前的验证器模型多依赖监督微调,缺乏如Q-learning等时序差分学习。本文提出VerifierQ,将离线Q-learning引入LLM验证器。针对三个关键挑战:(1) 处理话语级马尔可夫决策过程,(2) 管理大规模动作空间,(3) 缓解高估偏差,提出改进贝尔曼更新以限制Q值范围,采用隐式Q-learning(IQL)高效管理动作空间,并结合新型保守Q-learning(CQL)实现平衡的Q值估计。该方法支持并行Q值计算,提升训练效率。尽管已有研究探索使用MCTS等强化学习技术于生成器,但VerifierQ是首个将Q-learning应用于验证器(评论家)的尝试。该整合为生成与评估能力的协同进化提供了新路径,在数学推理任务上表现优于传统监督微调,兼具更高效率、准确率与鲁棒性。

原文摘要 · Abstract (English)

Recent advancements in test time compute, particularly through the use of verifier models, have significantly enhanced the reasoning capabilities of Large Language Models (LLMs). This generator-verifier approach closely resembles the actor-critic framework in reinforcement learning (RL). However, current verifier models in LLMs often rely on supervised fine-tuning without temporal difference learning such as Q-learning. This paper introduces VerifierQ, a novel approach that integrates Offline Q-learning into LLM verifier models. We address three key challenges in applying Q-learning to LLMs: (1) handling utterance-level Markov Decision Processes (MDPs), (2) managing large action spaces, and (3) mitigating overestimation bias. VerifierQ introduces a modified Bellman update for bounded Q-values, incorporates Implicit Q-learning (IQL) for efficient action space management, and integrates a novel Conservative Q-learning (CQL) formulation for balanced Q-value estimation. Our method enables parallel Q-value computation and improving training efficiency. While recent work has explored RL techniques like MCTS for generators, VerifierQ is among the first to investigate the verifier (critic) aspect in LLMs through Q-learning. This integration of RL principles into verifier models complements existing advancements in generator techniques, potentially enabling more robust and adaptive reasoning in LLMs. Experimental results on mathematical reasoning tasks demonstrate VerifierQ's superior performance compared to traditional supervised fine-tuning approaches, with improvements in efficiency, accuracy and robustness. By enhancing the synergy between generation and evaluation capabilities, VerifierQ contributes to the ongoing evolution of AI systems in addressing complex cognitive tasks across various domains.

大模型推理强化学习验证器

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