English

Express Your Doubts -- Probabilistic World Modeling Should not be Based on Token logprobs

Computation and Language 2026-05-13 v2 Artificial Intelligence

Abstract

Language modeling has shifted in recent years from a distribution over strings to prediction models with textual inputs and outputs for general-purpose tasks. This position paper highlights the often overlooked implications of this shift for the use of large language models (LLMs) as probability estimators, especially for world probabilities. In light of the theoretical distinction between distribution estimation and response prediction, we examine LLM training phases and common use cases for LLM output probabilities. We show that the different settings lead to distinct, potentially conflicting, desired output distributions. This lack of clarity leads to pitfalls when using output probabilities as event probabilities. Our position advocates for second-order prediction -- incorporating probabilities explicitly as part of the output -- as a theoretically sound method, in contrast to using token logprobs. We conclude with suggestions for potential directions to improve the probabilistic soundness of this method.

Keywords

Cite

@article{arxiv.2505.02072,
  title  = {Express Your Doubts -- Probabilistic World Modeling Should not be Based on Token logprobs},
  author = {Eitan Wagner and Omri Abend},
  journal= {arXiv preprint arXiv:2505.02072},
  year   = {2026}
}

Comments

Accepted to ICML 2026 (position track)

R2 v1 2026-06-28T23:20:34.299Z