One-Token Verification for Reasoning Correctness Estimation

Zhan Zhuang, Xiequn Wang, Zebin Chen, Feiyang Ye, Ying Wei, Kede Ma, Yu Zhang


Abstract
Recent breakthroughs in large language models (LLMs) have led to notable successes in complex reasoning tasks, such as mathematical problem solving. A common strategy for improving performance is parallel thinking, in which multiple reasoning traces are generated and the final prediction is made using aggregation schemes like majority voting or best-of-N decoding. However, two key challenges persist. First, multi-sample decoding incurs substantial inference latency, especially for long-form outputs. Second, effective mechanisms for reliably assessing the correctness of individual reasoning traces are still limited. To address these challenges, we introduce One-Token Verification (OTV), a computational method that estimates reasoning correctness in a single forward pass during generation. OTV is activated by a learnable token and integrated into the LLM via low-rank adaptation to probe internal reasoning signals through the key-value cache, supporting token-level correctness estimation at any stage of generation without disrupting primary reasoning. Experiments on mathematical reasoning show that OTV consistently outperforms existing verifiers while reducing token usage by up to 90% through correctness-guided early termination. This enables the model to favor shorter, more reliable solutions. The code is publicly available at https://github.com/Multimedia-Analytics-Laboratory/OTV
Anthology ID:
2026.emnlp-main.519
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
11906–11930
Language:
URL:
https://aclanthology.org/2026.emnlp-main.519/
DOI:
Bibkey:
Cite (ACL):
Zhan Zhuang, Xiequn Wang, Zebin Chen, Feiyang Ye, Ying Wei, Kede Ma, and Yu Zhang. 2026. One-Token Verification for Reasoning Correctness Estimation. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 11906–11930, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
One-Token Verification for Reasoning Correctness Estimation (Zhuang et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.519.pdf