English

Horseshoe Mixtures-of-Experts (HS-MoE)

Machine Learning 2026-01-15 v1 Machine Learning

Abstract

Horseshoe mixtures-of-experts (HS-MoE) models provide a Bayesian framework for sparse expert selection in mixture-of-experts architectures. We combine the horseshoe prior's adaptive global-local shrinkage with input-dependent gating, yielding data-adaptive sparsity in expert usage. Our primary methodological contribution is a particle learning algorithm for sequential inference, in which the filter is propagated forward in time while tracking only sufficient statistics. We also discuss how HS-MoE relates to modern mixture-of-experts layers in large language models, which are deployed under extreme sparsity constraints (e.g., activating a small number of experts per token out of a large pool).

Keywords

Cite

@article{arxiv.2601.09043,
  title  = {Horseshoe Mixtures-of-Experts (HS-MoE)},
  author = {Nick Polson and Vadim Sokolov},
  journal= {arXiv preprint arXiv:2601.09043},
  year   = {2026}
}