arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.08048cs.LGcs.CL
收藏

VerifierQ:利用基于 Q-learning 的验证器增强 LLM 测试时计算

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

CUNY Graduate Center(纽约城市大学研究生中心);Borough of Manhattan Community College(曼哈顿社区学院);The City College of New York(纽约城市学院)

Jianing Qi, Hao Tang, Zhigang Zhu

AI总结:

本文提出 VerifierQ,首次将离线 Q-learning 集成到 LLM 验证器中,通过改进 Bellman 更新、IQL 和 CQL 解决动作空间与过估计等挑战,显著提升了模型在数学推理任务中的效率、准确性和稳健性。

摘要:

测试时计算的最新进展,特别是通过使用验证器模型,显著增强了大型语言模型(LLMs)的推理能力。这种生成器-验证器方法与强化学习(RL)中的 actor-critic 框架非常相似。然而,当前 LLMs 中的验证器模型通常依赖于监督微调,而没有采用诸如 Q-learning 的时间差分学习。本文提出了 VerifierQ,这是一种将 Offline Q-learning 集成到 LLM 验证器模型中的新颖方法。我们解决了将 Q-learning 应用于 LLMs 时的三个关键挑战:(1)处理话语级马尔可夫决策过程(MDPs),(2)管理大型动作空间,以及(3)缓解过估计偏差。VerifierQ 引入了用于有界 Q 值的修改版 Bellman 更新,结合了 Implicit Q-learning(IQL)以实现高效的动作空间管理,并集成了一种新颖的 Conservative Q-learning(CQL)公式以进行平衡的 Q 值估计。我们的方法支持并行 Q 值计算并提高了训练效率。虽然最近的工作探索了诸如用于生成器的 MCTS 等 RL 技术,但 VerifierQ 是首批通过 Q-learning 研究 LLMs 中验证器(评论家)方面的工作之一。将 RL 原理集成到验证器模型中补充了生成器技术的现有进展,可能使 LLMs 中实现更稳健和自适应的推理。在数学推理任务上的实验结果表明,与传统监督微调方法相比,VerifierQ 表现出卓越的性能,在效率、准确性和稳健性方面均有提升。通过增强生成与评估能力之间的协同作用,VerifierQ 为 AI 系统在解决跨各个领域复杂认知任务的持续演进做出了贡献。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑