Probability is Not Enough: Exploring and Counting Divergent Tokens for Reasoning Uncertainty Quantification in LLMs

Feiyang Li, Shengjing Liu, Qi Zhan, Sijie Cheng, Weiqing Wang +5 more
9/29/2026
cs.AI

No official code found. Best implementation: Liflysheep/DTC-Open (0 stars, last commit 2026-09-30).

Abstract

As the chain-of-thought reasoning capabilities of large language models improve, evaluating and calibrating their reasoning confidence is becoming increasingly important for quantifying the uncertainty of their answers. Current methods for estimating the confidence of large language models are generally based on probabilities of selected key tokens, but the underlying mechanism remains unclear. Our pilot study finds that replacing selected token probabilities with coarse substitutes can also improve calibration, motivating us to further explore effective signals of model confidence. We introduce Divergent Token Confidence (DTC), a framework that estimates confidence by counting tokens at which two models strongly disagree during decoding. DTC identifies these divergent tokens using the Jensen-Shannon divergence between next-token distributions evaluated along the same reasoning trajectory. We find that their count is almost negatively associated with answer accuracy, thereby serving as a simple yet effective signal for uncertainty quantification. DTC supports both white-box and black-box evaluation using auxiliary models, without explicit training and affecting the generation process. Experiments across multiple model families and six mathematical benchmarks demonstrate improved calibration over probability-based and verbalized baselines. Under white-box evaluation, the count-only estimator achieves an average expected calibration error of 13.0%, compared with 32.7%-42.4% for standard full-sequence confidence methods. In black-box settings, it also improves calibration over the original verbalized scores. For example, mean expected calibration error falls from 32.1%-40.2% to 13.7%-16.3% on DeepSeek-V3.2. These findings provide new insights for improving reasoning uncertainty quantification in large language models. The code is released at https://github.com/szu-tera/DTC.git.

View on arXivView PDF

Code Implementations(1)

Official code for Divergent Token Confidence (DTC). Paper: arXiv:2609.38070 — count divergent tokens between two models to calibrate LLM chain-of-thought uncertainty.

00Sep 29, 20263 weeks ago

Similar Papers

Cite this paper

@article{li2026probability,
  title  = {Probability is Not Enough: Exploring and Counting Divergent Tokens for Reasoning Uncertainty Quantification in LLMs},
  author = {Feiyang Li and Shengjing Liu and Qi Zhan and Sijie Cheng and Weiqing Wang and Hongwen Chen and Yuxuan Yang and Wen Wang and Yile Wang and Hui Huang},
  year   = {2026},
  eprint = {2609.38070},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2609.38070v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2609.38070.svg)](https://codeofpaper.com/paper/2609.38070)

Discussion