Process Rewards with Learned Reliability
带有学习可靠性的过程奖励模型
🧭 过程奖励可靠性建模与自适应推理
Washington University in St. Louis · Singapore University of Technology and Design
BetaPRM 预测步骤正确概率和该预测的可靠性,并利用可靠性信号实现自适应计算分配,减少推理 token 消耗
📌 工作参考价值(4 点)
- 在 Agent 多步决策中,使用 Beta 分布建模每步奖励的可靠性,避免单点奖励误导决策,增强动作选择的可信度。
- 对于生成式推荐候选集选择的 Best-of-N 推理,可借鉴自适应计算分配(ACA):根据可靠性提前终止高置信候选,将计算资源集中到不确定的候选上,显著减少 LLM 调用成本。
- 训练 PRM 时,不再回归有限样本成功率,而是采用 Beta-Binomial 似然直接学习 Beta 信念,输出不确定性,这对排名、过滤等下游任务更友好。
- 在需要对中间结果(如 query 重写、解释生成)打分的系统中,可引入可靠性阈值控制,避免过早丢弃潜在正确但暂时评分偏低的步骤。