STEP: Soft Trajectory Estimation of Prediction Correctness via Per-Step Probability Scoring

Chenyang Yu, Zifan Zeng, Qunli Zhang, Feng Liu


Abstract
Estimating the correctness of large language model (LLM) predictions without access to ground truth is critical for reliable deployment in high-stakes settings. Existing approaches, such as entropy and perplexity, operate solely on final output statistics, discarding the rich information encoded in intermediate reasoning steps. To address this limitation, we propose STEP (Soft Trajectory Estimation of Prediction correctness), a training-free framework that scores each reasoning step via lightweight forward passes to obtain a soft probability distribution qi over candidate answers, from which we derive a set of soft trajectory features capturing per-step confidence and its progression toward the final answer. Extensive experiments on various benchmarks demonstrate consistent and substantial improvements, particularly in AUROC, over output-based baselines and hard-indicator trajectory methods. Notably, Temporally-Weighted Consistency, which assigns higher importance to later reasoning steps where answer distributions have stabilized, contributes the largest gains, highlighting the importance of temporal structure in reasoning trajectories. These results further validate the advantage of preserving continuous probability signals over binary discretization.
Anthology ID:
2026.uncertainlp-main.5
Volume:
The Proceedings of the Third Workshop on Uncertainty-Aware NLP
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Wilker Aziz, Jonathan Berant, Bryan Eikema, Evgenia Ilia, Marie-Catherine de Marneffe, Barbara Plank, Artem Shelmanov, Swabha Swayamdipta, Jörg Tiedemann, Artem Vazhentsev, Raúl Vázquez, Chrysoula Zerva
Venues:
UncertaiNLP | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
61–72
Language:
URL:
https://aclanthology.org/2026.uncertainlp-main.5/
DOI:
Bibkey:
Cite (ACL):
Chenyang Yu, Zifan Zeng, Qunli Zhang, and Feng Liu. 2026. STEP: Soft Trajectory Estimation of Prediction Correctness via Per-Step Probability Scoring. In The Proceedings of the Third Workshop on Uncertainty-Aware NLP, pages 61–72, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
STEP: Soft Trajectory Estimation of Prediction Correctness via Per-Step Probability Scoring (Yu et al., UncertaiNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.uncertainlp-main.5.pdf