Partition, Prompt, Aggregate: Statistical Self-Consistency in Language Models
Summary
This paper evaluates whether large language models satisfy the law of total probability by recursively partitioning populations using binary conditioning trees. The authors compare direct population-level estimates with those reconstructed from subpopulation conditionals, revealing a "macro fallacy" where aggregated fine-grained estimates outperform direct population-level estimates.
Mathematical/empirical assessment
The authors rigorously formalize statistical self-consistency. They define the aggregate alignment error $\text{AErr}(\ell)$ and introduce split and order consistency metrics bounded by a small tolerance $\varepsilon$. Empirically, they validate these concepts using ACS income data, demonstrating that explicit decomposition reduces residual within-subgroup variance while exposing cross-group differences effectively.
Strengths
The contribution is highly plausible and theoretically grounded. Framing in-context learning through the lens of basic probability axioms provides a novel, reference-free evaluation paradigm for LLMs. The mathematical formulation of split consistency, particularly the multi-step aggregation proposition, is elegant and clearly presented.
Concerns
While the framework is strong, the assumption of context-independent pairwise order discrepancy limits the generalizability of the order consistency score, as prompt context might subtly influence these order effects. Furthermore, the empirical validation relies heavily on specific sociodemographic splits like age and employment; it remains unclear if the macro fallacy persists for more abstract or continuous attributes. Nevertheless, these are manageable limitations in an otherwise highly compelling study.
Final decision
Strong accept