Improving the Accuracy of Amortized Model Comparison with Self-Consistency
By: Šimon Kucharský , Aayush Mishra , Daniel Habermann and more
Amortized Bayesian inference (ABI) offers fast, scalable approximations to posterior densities by training neural surrogates on data simulated from the statistical model. However, ABI methods are highly sensitive to model misspecification: when observed data fall outside the training distribution (generative scope of the statistical models), neural surrogates can behave unpredictably. This makes it a challenge in a model comparison setting, where multiple statistical models are considered, of which at least some are misspecified. Recent work on self-consistency (SC) provides a promising remedy to this issue, accessible even for empirical data (without ground-truth labels). In this work, we investigate how SC can improve amortized model comparison conceptualized in four different ways. Across two synthetic and two real-world case studies, we find that approaches for model comparison that estimate marginal likelihoods through approximate parameter posteriors consistently outperform methods that directly approximate model evidence or posterior model probabilities. SC training improves robustness when the likelihood is available, even under severe model misspecification. The benefits of SC for methods without access of analytic likelihoods are more limited and inconsistent. Our results suggest practical guidance for reliable amortized Bayesian model comparison: prefer parameter posterior-based methods and augment them with SC training on empirical datasets to mitigate extrapolation bias under model misspecification.
Similar Papers
Towards Trustworthy Amortized Bayesian Model Comparison
Machine Learning (Stat)
Helps computers pick the best explanation for data.
Uncertainty-Aware Surrogate-based Amortized Bayesian Inference for Computationally Expensive Models
Machine Learning (Stat)
Makes computer guesses better with less work.
Optimal Self-Consistency for Efficient Reasoning with Large Language Models
Machine Learning (CS)
Makes AI smarter with fewer guesses.