English

Exploring Expert Specialization through Unsupervised Training in Sparse Mixture of Experts

Machine Learning 2025-09-15 v1 Artificial Intelligence

Abstract

Understanding the internal organization of neural networks remains a fundamental challenge in deep learning interpretability. We address this challenge by exploring a novel Sparse Mixture of Experts Variational Autoencoder (SMoE-VAE) architecture. We test our model on the QuickDraw dataset, comparing unsupervised expert routing against a supervised baseline guided by ground-truth labels. Surprisingly, we find that unsupervised routing consistently achieves superior reconstruction performance. The experts learn to identify meaningful sub-categorical structures that often transcend human-defined class boundaries. Through t-SNE visualizations and reconstruction analysis, we investigate how MoE models uncover fundamental data structures that are more aligned with the model's objective than predefined labels. Furthermore, our study on the impact of dataset size provides insights into the trade-offs between data quantity and expert specialization, offering guidance for designing efficient MoE architectures.

Keywords

Cite

@article{arxiv.2509.10025,
  title  = {Exploring Expert Specialization through Unsupervised Training in Sparse Mixture of Experts},
  author = {Strahinja Nikolic and Ilker Oguz and Demetri Psaltis},
  journal= {arXiv preprint arXiv:2509.10025},
  year   = {2025}
}

Comments

14 pages, 7 figures

R2 v1 2026-07-01T05:33:05.621Z