Search papers, labs, and topics across Lattice.
This paper conducts a thorough theoretical analysis of Bayesian mixture-of-experts models that utilize a softmax gating mechanism, addressing their posterior distribution behavior in the contexts of density estimation, parameter estimation, and model selection. The authors establish posterior contraction rates and convergence guarantees, revealing how the number of experts can be fixed or learned adaptively. These findings not only enhance the understanding of Bayesian mixture-of-experts but also offer practical insights for model design, particularly in managing the complex identifiability issues inherent to these models.
Bayesian mixture-of-experts models can achieve robust density and parameter estimation with adaptive expert selection, fundamentally reshaping our approach to complex probabilistic modeling.
Mixture-of-experts models provide a flexible framework for learning complex probabilistic input-output relationships by combining multiple expert models through an input-dependent gating mechanism. These models have become increasingly prominent in modern machine learning, yet their theoretical properties in the Bayesian framework remain largely unexplored. In this paper, we study Bayesian mixture-of-experts models, focusing on the ubiquitous softmax-based gating mechanism. Specifically, we investigate the asymptotic behavior of the posterior distribution for three fundamental statistical tasks: density estimation, parameter estimation, and model selection. First, we establish posterior contraction rates for density estimation, both in the regimes with a fixed, known number of experts and with a random learnable number of experts. We then analyze parameter estimation and derive convergence guarantees based on tailored Voronoi-type losses, which account for the complex identifiability structure of mixture-of-experts models. Finally, we propose and analyze two complementary strategies for selecting the number of experts. Taken together, these results provide one of the first systematic theoretical analyses of Bayesian mixture-of-experts models with softmax gating, and yield several theory-grounded insights for practical model design.
