HomeMachine LearningarXiv:2605.29782

Hista and Numca: Estimate State Value Effectively for LLM Reinforcement Learning

Abstract

Reinforcement learning (RL) refines large language models (LLMs) by directly optimizing model behavior through reward signals. While accurate state value estimation is critical for stable training in classical RL, it remains an underexplored challenge in LLM post-training. In this work, we introduce the State Value Estimation Benchmark (SVEB) to assess state estimation within existing RL frameworks and show that critics in standard approaches like PPO collapse to a coarse group-average baseline. To address this, we propose two techniques: Numca, which leverages numerical spans as gradable milestones for state value estimation, and Hista, a framework that uses LLM's hidden states as representation to weighted average disjoint rollouts and their return. Extensive experiments demonstrate that both methods yield more accurate state value estimates and enhance training performance across different RL algorithms and model sizes without incurring significant computational overhead.

Comments: Accepted at ICML 2026

Cite

@article{arxiv.2605.29782,
  title  = {Hista and Numca: Estimate State Value Effectively for LLM Reinforcement Learning},
  author = {Zizhe Chen and Jiqian Dong and Yizhou Tian and Garry Yang and Yongqiang Chen and Zhitang Chen and James Cheng},
  journal= {arXiv preprint arXiv:2605.29782},
  year   = {2026}
}