Consistent Variance Estimation for Q-Function Estimators in Finite-Horizon MDP Tree Search

arXiv:2607.14555 2026 Training 1 ideas extracted · analyzed Aug 30, 2026

What the math gives to ML

The paper offers a concrete law-of-total-variance decomposition for finite-horizon tree returns and shows that treating sampled root-to-leaf paths as i.i.d. produces an asymptotically persistent downward bias. Its transferable asset is a node-local recursive estimator that propagates uncertainty through branching transitions, rather than recomputing a misleading sample variance over correlated trajectories. The most promising neural-network transfer is uncertainty-aware Bellman training for world models, value networks, and branching rollout systems: propagate reward, transition, and child-value variance separately, then use the resulting variance to weight targets, prioritize data, or stop rollouts.

Ideas from this paper

Failed on benchmark 2026

Recursive Bellman Variance Targets

Replace the naive sample variance of correlated rollout returns with a recursive variance target attached to every state-action node or latent rollout node. The target separates uncertainty caused by immediate reward noise, stochastic next-state selection, and uncertainty already present in child value estimates, enabling calibrated heteroscedastic Bellman updates and uncertainty-aware rollout allocation.

Useful7/10
Difficulty5/10
Novelty7/10
Paper: Consistent Variance Estimation for Q-Function Estimators in Finite-Horizon MDP Tree Search arXiv:2607.14555