Diagnosing and Improving Probabilistic Reasoning in Large Language Models

Huaman Sun, Dingcheng Wang, Jason Hartline, Jessica Hullman
9/29/2026
cs.AI

No confident code implementation found for this paper. 5 possible matches can be shown below.

Abstract

Large language models (LLMs) are increasingly proposed as decision assistants who must reason probabilistically from available evidence under explicit decision costs. We propose a decision-theoretic framework that decomposes LLMs' decision loss into two components: forming accurate beliefs from provided evidence and translating those beliefs into actions that optimize a provided utility function. Using a synthetic benchmark with known ground truth, we apply the decomposition to characterize probabilistic reasoning in frontier and open-sourced models. We further evaluate whether RL interventions targeting beliefs, decisions, or both improve these components across three domains, whether improvements transfer across components and elicitation formats, and whether decision performance can improve without improvement in belief formation. We find that targeting one component of probabilistic reasoning redistributes decision loss, improving the target without necessarily transferring to others, and that jointly targeting belief formation and decision-making improves both but hinges on matched formats between training and evaluation.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 5 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{sun2026diagnosing,
  title  = {Diagnosing and Improving Probabilistic Reasoning in Large Language Models},
  author = {Huaman Sun and Dingcheng Wang and Jason Hartline and Jessica Hullman},
  year   = {2026},
  eprint = {2609.38005},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2609.38005v1}
}

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.38005.svg)](https://codeofpaper.com/paper/2609.38005)

Discussion