by , , ,
Abstract:
In-context learning is commonly interpreted as a form of conditional inference, in which the prompt specifies a context and the model's output is treated as an estimate of the corresponding conditional distribution. If this interpretation holds, then LLM estimates should satisfy basic probabilistic identities. In particular, the law of total probability asserts that prior-weighted conditional distributions aggregate into population-level marginals over any valid partition of the population. In this work, we investigate to what extent LLM estimates adhere to this self-consistency principle. We use binary trees as an evaluation scaffold to recursively partition a population into increasingly fine-grained subpopulations. We then prompt LLMs with verbalized subpopulation descriptions in context, aggregate the resulting estimates back into population-level estimates, and compare them across partitions of varying granularity. Applying this protocol across problem domains and state-of-the-art frontier models, we show widespread violations of basic consistency properties. An in-depth study of persona prompting further reveals a pattern we call the macro fallacy: estimates reconstructed from more fine-grained subpopulation responses are often better aligned with human reference data than direct population-level estimates. The effect persists across variations in tree structure, split verbalization, and estimation task. Together, these findings establish statistical self-consistency as an unsaturated evaluation criterion complementary to contemporary benchmarks and motivate an asymmetric training objective that anchors aggregate predictions in subpopulation estimates that better align with ground truth.
Reference:
Partition, Prompt, Aggregate: Statistical Self-Consistency in Language Models P. Wolf, T. K. Buening, A. Krause, C. Mendler-DünnerIn Advances in Neural Information Processing Systems (NeurIPS), 2026
Bibtex Entry:
@inproceedings{wolf_2026_self_consistency,
	author = {Patrik Wolf and Thomas Kleine Buening and Andreas Krause and Celestine Mendler-D{\"u}nner},
	booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
	month = {December},
	pdf = {https://arxiv.org/abs/2607.15277},
	title = {Partition, Prompt, Aggregate: Statistical Self-Consistency in Language Models},
	year = {2026}}