用流程挖掘技术量化推理步骤结构质量,提升数学解题模型表现
Rewarding Structural Conformance of Reasoning using Process Mining
- 通过流程挖掘对比教师与模型的推理路径差异
- 生成0-1区间奖励值,显著提升解题结构质量
- 无需人工标注,适配现有强化学习框架
稀疏奖励策略梯度方法在基于语言模型的后训练中取得进展,但在数学推理等任务中,二元结果奖励无法提供中间步骤的有效反馈。尽管已有研究尝试评估整体推理质量,但其可靠性仍不明确。本文将推理视为结构化过程,提出TACReward——一种可无缝集成至稀疏奖励策略梯度方法的奖励模型,无需额外人工标注或架构修改。该模型利用流程挖掘技术,聚合教师与策略模型在推理步骤上的结构偏差,输出[0,1]范围的标量奖励,反映推理质量。在多个数学推理基准上的实验表明,引入TACReward能有效提升策略模型推理结构质量,带来一致性能提升。代码与模型已公开于https://github.com/Thrillcrazyer/TACReward和https://huggingface.co/Thrillcrazyer/TACReward7B。
原文摘要 · Abstract (English)
Recent advances in sparse reward policy gradient methods have enabled effective reinforcement learning (RL)-based language model post-training. However, for reasoning tasks such as mathematical problem solving, binarized outcome rewards provide limited feedback on intermediate reasoning steps. While some studies have attempted to address this issue by estimating overall reasoning quality, it remains unclear whether these rewards are reliable proxies for the quality of stepwise reasoning. In this study, we consider reasoning as a structured process and propose TACReward, the reward model that can be seamlessly integrated into sparse reward policy gradient methods without additional human annotation costs or architectural modifications. TACReward aggregates stepwise structural deviations between teacher and policy reasoning using process mining techniques, producing a scalar output reward range of [0, 1] to indicate reasoning quality. Experiments on multiple mathematical reasoning benchmarks demonstrate that integrating the TACReward into sparse reward frameworks encourages the policy model to improve the structural quality of reasoning. Consequently, this leads to consistent performance improvements over existing sparse reward frameworks. Our code and checkpoints are publicly available at https://github.com/Thrillcrazyer/TACReward and https://huggingface.co/Thrillcrazyer/TACReward7B.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。