cs.LGAug 23, 2026

The Variance of Thought: Policy Variance, Critical Forks, and Local Credit Assignment

Authors: Yingru Li

Abstract

Long-horizon language-model tasks --- multi-step reasoning and tool-using agents alike --- are limited by credit assignment. We analyze it through the policy variance σπ2(s)=Var⁡a∼π[Qπ(s,a)]σ_π^2(s)=\operatorname{Var}_{a\simπ}[Q_π(s,a)], which in a deterministic MDP is the sole source of return variance and is injected in discrete pulses at states we call critical forks. Three results follow. (i) Policy variance is a discovery budget: observing an action of advantage cc requires Ω(c2/σπ2(s))Ω(c^2/σ_π^2(s)) draws, a bound that is exact on the canonical two-point fork. (ii) Policy variance is bounded by the policy's Gini dispersion, σπ2(s)≤1−∥π(⋅∣s)∥22σ_π^2(s)\le 1-\|π(\cdot|s)\|_2^2, a rollout-free necessary condition for criticality computable from logits alone. (iii) The remaining horizon sets the estimation cost: at a fork whose downstream success probability is PP, the Monte Carlo advantage estimate has signal-to-noise ratio of order P\sqrt{P}, so its sample cost scales as 1/P1/P --- a cost that branched sampling shares. Bootstrapping removes it by converting a product of survival probabilities into a sum, provided the value representation is multiplicatively accurate, which argues for log-value parameterization.

Explore similar work

CardsList
  1. Multiscale Policy Optimization for Long-Horizon LLM Agents

    Oct 26, 2026Tao Ren, Hui Yang, Jia Deng +16Policy OptimizationRL for LLM Agents

  2. Credit Assignment with Resets in Language Model Reasoning

    May 25, 2026Ankur Samanta, Akshayaa Magesh, Ayush Jain +7RL for Language Model ReasoningCredit Assignment in RL

  3. Reducing Credit Assignment Variance via Counterfactual Reasoning Paths

    Apr 20, 2026Fei Ding, Yongkang Zhang, Youwei Wang +1RL for Language Model ReasoningCredit Assignment in RL