Ai! Jack!v2026.04.01-f12
← 홈HF Daily Papers

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

원문 보기
중국

AI 분석

•기여: 모델의 자기지도 신호를 검증기 결과로 보정하여 잘못된 확신의 강화와 풀이 다양성 축소를 줄이는 on-policy 자기개선 기법 FlowBalance 제안
•방법: 학습 중 고정한 동일 정책의 뷰가 특권 문맥으로 계산한 토큰 로그확률 증가량을 응답 전체 점수로 집계. 검증기 기반 그룹 advantage가 양수면 지도를 유지하고 음수면 반전하며 결과 선호가 없으면 비활성화한 뒤, 그룹별 로그 정규화 상수를 사용하는 trajectory balance로 목표 분포 학습
•결과: Qwen3-4B·8B 수학 추론에서 FlowRL보다 높은 평균 성능과 학습 속도·안정성 개선 보고. 직접적인 OPSD의 응답 길이 붕괴를 피하고 통제된 AIME24 진단에서 더 다양한 정답 풀이 전략 관찰
•한계: 초록에 성능·속도 개선 수치가 없고 검증기 및 특권 문맥이 필요하며, 수학 외 과제로의 일반화는 제시되지 않음

검증 정보

출처
huggingface.co
피드
HF Daily Papers
수집/분석 기준
2026. 9. 8. 오전 9:00
원문 확인
원문 열기

본문

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.

FlowBalance: 검증기에 기반한 On-Policy 추론 경험의 자기개선 — Ai! Jack! — Ai! Jack!