Lune

ICLR2026Top-tier venue

Eliciting Numerical Predictive Distributions of LLMs Without Auto-Regression

Julianna Piskorz, Kasia Kobalczyk, Mihaela van der Schaar

2026Year
2Citations

Abstract

Large Language Models (LLMs) have recently been successfully applied to regression tasks-such as time series forecasting and tabular prediction-by leveraging their in-context learning abilities. However, their autoregressive decoding process may be ill-suited to continuous-valued outputs, where obtaining predictive distributions over numerical targets requires repeated sampling, leading to high computational cost and inference time. In this work, we investigate whether distributional properties of LLM predictions can be recovered without explicit autoregressive generation. To this end, we study a set of regression probes trained to predict statistical functionals (e.g., mean, median, quantiles) of the LLM's numerical output distribution directly from its internal representations. Our results suggest that LLM embeddings carry informative signals about summary statistics of their predictive distributions, including the numerical uncertainty. This investigation opens up new questions about how LLMs internally encode uncertainty in numerical tasks, and about the feasibility of lightweight alternatives to sampling-based approaches for uncertainty-aware numerical predictions. Code to reproduce our experiments can be found at https://github.com/kasia-kobalczyk/guess_llm.git .

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 912fb8ed-ce3f-4219-aa5c-da2671d7a3a3

Builds on6

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines