cd /news/machine-learning/fork-where-the-model-changes-its-min… · home topics machine-learning article
[ARTICLE · art-127430] src=arxiv.org ↗ pub= topic=machine-learning verified=true sentiment=↑ positive

Fork Where the Model Changes Its Mind: Belief-Shift Branching for Tree-Structured Reinforcement Learning

A new arXiv paper (2609.11061v1) proposes belief-shift branching, a fork-placement method for tree-structured reinforcement learning with verifiable rewards that forks a chain just before the step where the model's consecutive answer beliefs diverge most. In validation against Monte-Carlo value curves, the belief-shift signal ranked first in each of the eight model×benchmark panels, ahead of entropy, structural, and LLM-judge baselines, with the probe costing about 1% of step compute on mathematics and under 5% on code. In RL across three model families and two domains, belief-shift forking led every mathematics aggregate, gaining +2.6 aggregate and +2.9 on AIME 2026 for OLMo-3-7B over the strongest baseline, and swept every OLMo code column, by +6.5 on LiveCodeBench-medium.

by read1 min views1 publishedSep 12, 2026

arXiv:2609.11061v1 Announce Type: new Abstract: Tree-structured rollouts give critic-free reinforcement learning with verifiable rewards (RLVR) step-level credit: fork a chain at an intermediate point, and sibling outcome differences estimate step value. Each fork adds sampling cost, so realistic budgets typically allow only a few forks per chain. A fork placed where the outcome is already largely settled yields siblings that mostly agree and provide almost no credit signal; hence, for a given tree size, where forks are placed largely determines how much step-level RL can gain. Most existing mainstream methods place forks by structure, such as fixed lengths, midpoints, and delimiters, or by next-token entropy. We formalize fork placement as locating the \emph{pivots} of the chain's value curve, where the expected outcome turns. We propose \emph{belief-shift branching}: read the model's answer belief at candidate boundaries and fork just before the step where consecutive beliefs diverge most. Three instantiations, none needing step-level supervision, span access levels: a black-box probe, a logit-lens depth profile, and a learned activation direction, which is fit offline and therefore used only in the validation before RL training. The signal only \emph{places} forks, and the probe costs about $1%$ of step compute on mathematics and under $5%$ on code when it runs inside the rollout engine. In that validation, against Monte-Carlo value curves, a belief-shift signal ranks first in each of the eight model$\times$benchmark panels, ahead of entropy, structural, and LLM-judge baselines. In RL across three model families and two domains, belief-shift forking leads every mathematics aggregate, on OLMo-3-7B by $+2.6$ aggregate and $+2.9$ on AIME 2026 over the strongest baseline, and sweeps every OLMo code column, by $+6.5$ on LiveCodeBench-medium.

── more in #machine-learning 4 stories · sorted by recency
── more on @arxiv 3 stories trending now
sponsored brought to you by zahid.host 4,200+ EU-deployed projects
reading about agents? ship yours in a single git push.

Run your AI side-project on zahid.host

EU-based hosting, git-push deploys, automatic HTTPS, no cold starts. Free tier with a custom domain — perfect for shipping the agent you just read about.

$git push zahid main
Live at https://your-agent.zahid.host
Get free account → Pricing
from €0/mo · no card required
LIVE [news/fork-where-the-model…] indexed:0 read:1min 2026-09-12 ·