arXiv:2609.03241cs.LGcs.AI2026-09

用验证器校准自我改进,让模型从自身推理中更稳定地学习正确解法。

FlowBalance: Verifier-Grounded Self-Improvement from On-Policy Reasoning Experience

论文配图:FlowBalance: Verifier-Grounded Self-Improvement from On-Policy Reasoning Experience
图 1 · 摘自论文原文
  • 通过轨迹平衡机制,用验证结果调整模型自我引导强度。
  • 在Qwen3-4B和8B上提升数学推理准确率,且训练更稳定快速。
  • 适合需要可靠自进化能力的复杂推理系统研发者。

推理模型可从自身在线策略经验中自我提升,但该内循环脆弱:终态验证器提供可靠但稀疏的监督,而同模型密集指导可能强化错误信心或过度集中于单一解法模式。我们提出FlowBalance,一种基于验证器校准的自改进方法,学习完整响应的归一化分布。对每条在线策略轨迹,使用训练时冻结的同一策略,利用特权上下文生成词元级概率增益,并聚合为轨迹级自我引导得分。FlowBalance将该得分与验证器提供的群体优势进行校准:正优势轨迹保留引导,负优势轨迹反转引导,无偏好时禁用引导。由此产生的能量函数指数级重加权参考策略,每组轨迹仅需一次对数分区估计即可实现目标分布拟合。该方法通过轨迹平衡实现结果校准的自我引导,无需单独的词元级模仿损失。分析表明其具有组内对比保持性、最小变化逆KL表征、验证器对目标奖励的单调控制,以及对拒绝响应中假阳性自我引导的精确修正。在数学推理任务中,FlowBalance在Qwen3-4B和Qwen3-8B上均优于FlowRL,同时提升训练速度与稳定性,避免直接OPSD的响应长度坍缩问题,并在受控AIME24诊断中展现更高正确策略多样性。

原文摘要 · Abstract (English)

A reasoning model can improve from its own on-policy experience, but this inner loop is fragile: terminal verifiers provide reliable yet sparse supervision, while dense same-model guidance can reinforce false confidence or overconcentrate learning on a narrow solution mode. We introduce FlowBalance, a verifier-grounded self-improvement method that learns a normalized distribution over complete responses. For each on-policy trajectory, a frozen training-time view of the same policy uses privileged context to produce token-level log-probability gains, which are aggregated into a trajectory-level self-guidance score. FlowBalance calibrates this score with the verifier-derived group advantage: guidance is retained on positive-advantage trajectories, reversed on negative-advantage trajectories, and disabled when the rollout group provides no outcome preference. The resulting energy exponentially reweights a reference policy, and profiled trajectory balance fits the normalized target with one log-partition estimate per rollout group. This realizes outcome-calibrated self-guidance via trajectory balance, without a separate token-level imitation loss. Our analysis establishes within-group contrast preservation, a minimum-change reverse-KL characterization, monotonic verifier control of target reward, and an exact correction against false-positive self-guidance on rejected responses. On mathematical reasoning, FlowBalance improves average performance over FlowRL on both Qwen3-4B and Qwen3-8B, while also improving training speed and stability, avoiding direct OPSD's response-length collapse, and exhibiting higher correct-strategy diversity in a controlled AIME24 diagnostic.

自改进推理增强验证器校准

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