BOB: Bayesian Optimized Bootstrap for Uncertainty Quantification in Gaussian Mixture Models
Abstract: A natural way to quantify uncertainties in Gaussian mixture models (GMMs) is through Bayesian methods. That said, sampling from the joint posterior distribution of GMMs via standard Markov chain Monte Carlo (MCMC) imposes several computational challenges, which have prevented a broader full Bayesian implementation of these models. A growing body of literature has introduced the Weighted Likelihood Bootstrap and the Weighted Bayesian Bootstrap as alternatives to MCMC sampling. The core idea of these methods is to repeatedly compute maximum a posteriori (MAP) estimates on many randomly weighted posterior densities. These MAP estimates then can be treated as approximate posterior draws. Nonetheless, a central question remains unanswered: How to select the random weights under arbitrary sample sizes. We, therefore, introduce the Bayesian Optimized Bootstrap (BOB), a computational method to automatically select these random weights by minimizing, through Bayesian Optimization, a black-box and noisy version of the reverse Kullback-Leibler (KL) divergence between the Bayesian posterior and an approximate posterior obtained via random weighting. Our proposed method outperforms competing approaches in recovering the Bayesian posterior, it provides a better uncertainty quantification, and it retains key asymptotic properties from existing methods. BOB's performance is demonstrated through extensive simulations, along with real-world data analyses.
- \bibcommenthead
- Comparative analysis of statistical pattern recognition methods in high dimensional settings. Pattern Recognition 27(8): 1065–1077 .
- A new class of stochastic em algorithms. escaping local maxima and handling intractable sampling. Computational Statistics & Data Analysis 159: 107159 .
- Combining mixture components for clustering. Journal of Computational and Graphical Statistics 19(2): 332–353 .
- Bishop, C.M. 2006. Pattern recognition and machine learning. Springer New York, NY.
- Variational inference: A review for statisticians. Journal of the American Statistical Association 112(518): 859–877 .
- Bull, A.D. 2011. Convergence rates of efficient global optimization algorithms. Journal of Machine Learning Research 12(88): 2879–2904 .
- Stan: A probabilistic programming language. Journal of Statistical Software 76(1): 1–32 .
- Computational and inferential difficulties with mixture posterior distributions. Journal of the American Statistical Association 95(451): 957–970 .
- Complete gradient clustering algorithm for features analysis of x-ray images. Information Technologies in Biomedicine 2: 15–24 .
- Maximum likelihood from incomplete data via the em algorithm. Journal of the Royal Statistical Society Series B: Statistical Methodology 39(1): 1–22 .
- Diebolt, J. and C.P. Robert. 1994. Estimation of finite mixture distributions through bayesian sampling. Journal of the Royal Statistical Society Series B: Statistical Methodology 56(2): 363–375 .
- Scalable nonparametric sampling from multimodal posteriors with the posterior bootstrap. Proceedings of the 36th International Conference on Machine Learning 97: 1952–1962 .
- Fraley, C. and A.E. Raftery. 2002. Model-based clustering, discriminant analysis, and density estimation. Journal of the American Statistical Association 97(458): 611–631 .
- Sparse inverse covariance estimation with the graphical lasso. Biostatistics 9(3): 432–441 .
- Geisser, S. 2017. Predictive Inference. Chapman and Hall/CRC.
- Gelman, A. and D.B. Rubin. 1992. Inference from iterative simulation using multiple sequences. Statistical Science 7(4): 457–472 .
- The no-u-turn sampler: Adaptively setting path lengths in hamiltonian monte carlo. Journal of Machine Learning Research 15(47): 1593–1623 .
- Hofmeyr, D.P. 2021. Fast exact evaluation of univariate kernel sums. IEEE Transactions on Pattern Analysis and Machine Intelligence 43(2): 447–458 .
- Hofmeyr, D.P. 2022. Fast kernel smoothing in r with applications to projection pursuit. Journal of Statistical Software 101(3): 1–33 .
- Izenman, A.J. and C.J. Sommer. 1988. Philatelic mixtures and multimodal densities. Journal of the American Statistical Association 83(404): 941–953 .
- Jones, D.R. 2001. A taxonomy of global optimization methods based on response surfaces. Journal of Global Optimization 21: 345–383 .
- Efficient global optimization of expensive black-box functions. Journal of Global Optimization 13: 455–492 .
- Tuning hyperparameters without grad students: Scalable and robust bayesian optimisation with dragonfly. Journal of Machine Learning Research 21(81): 1–27 .
- Optimization by simulated annealing. Science 220(4598): 671–680 .
- Automatic differentiation variational inference. Journal of Machine Learning Research 18(14): 1–45 .
- Deterministic approximate em algorithm; application to the riemann approximation em and the tempered em. Algorithms 15(3): 78 .
- Revisiting bayesian optimization in the light of the coco benchmark. Structural and Multidisciplinary Optimization 64(5): 3063–3087 .
- General bayesian updating and the loss-likelihood bootstrap. Biometrika 106(2): 465–478 .
- Bayesian posterior predictive checks for complex models. Sociological Methods & Research 32(3): 301–335 .
- High-dimensional bayesian optimization using low-dimensional feature spaces. Machine Learning 109: 1925–1943 .
- Automatic differentiation variational inference with mixtures. International Conference on Artificial Intelligence and Statistics 130: 3250–3258 .
- Weighted bayesian bootstrap for scalable posterior distributions. Canadian Journal of Statistics 49(2): 421–437 .
- Newton, M.A. and A.E. Raftery. 1994. Approximate bayesian inference with the weighted likelihood bootstrap. Journal of the Royal Statistical Society Series B: Statistical Methodology 56(1): 3–48 .
- Ng, T.L. and M.A. Newton. 2022. Random weighting in lasso regression. Electronic Journal of Statistics 16(1): 3430–3481 .
- Scalable bayesian nonparametric clustering and classification. Journal of Computational and Graphical Statistics 29(1): 53–65 .
- Bayesian bootstrap spike-and-slab lasso. Journal of the American Statistical Association 118(543): 2013–2028 .
- Deep bootstrap for bayesian inference. Philosophical Transactions of the Royal Society AÂ 381(2247): 20220154 .
- Nobile, A. and A.T. Fearnside. 2007. Bayesian finite mixtures with an unknown number of components: The allocation sampler. Statistics and Computing 17: 147–162 .
- Stochastic volatility with leverage: Fast and efficient likelihood inference. Journal of Econometrics 140(2): 425–449 .
- Pompe, E. 2021. Introducing prior information in weighted likelihood bootstrap with applications to model misspecification. arXiv preprint arXiv:2103.14445 .
- Nonparametric copula density estimation methodologies. Mathematics 12(3): 398 .
- Gaussian Processes for Machine Learning. The MIT Press.
- Richardson, S. and P.J. Green. 1997. On bayesian analysis of mixtures with an unknown number of components (with discussion). Journal of the Royal Statistical Society Series B: Statistical Methodology 59(4): 731–792 .
- Dicekriging, diceoptim: Two r packages for the analysis of computer experiments by kriging-based metamodeling and optimization. Journal of Statistical Software 51(1): 1–55 .
- Sambridge, M. 2014. A Parallel Tempering algorithm for probabilistic sampling and multimodal optimization. Geophysical Journal International 196(1): 357–374 .
- Schilling, R.L. 2017. Measures, Integrals and Martingales. Cambridge University Press.
- Sklar, M. 1959. Fonctions de répartition à N dimensions et leurs marges. Annales de l’ISUP 8(3): 229–231 .
- Practical bayesian optimization of machine learning algorithms. Advances in Neural Information Processing Systems 25: 2951–2959 .
- Stephens, M. 2000a. Bayesian analysis of mixture models with an unknown number of components—an alternative to reversible jump methods. The Annals of Statistics 28(1): 40 – 74 .
- Stephens, M. 2000b. Dealing with label switching in mixture models. Journal of the Royal Statistical Society Series B: Statistical Methodology 62(4): 795–809 .
- Fast, scalable approximations to posterior distributions in extended latent gaussian models. Journal of Computational and Graphical Statistics 32(1): 84–98 .
- Overfitting bayesian mixture models with an unknown number of components. PloS one 10(7): e0131739 .
- Bayesian Cluster Analysis: Point Estimation and Credible Balls (with Discussion). Bayesian Analysis 13(2): 559 – 626 .
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.