Explainable Mixture Models through Differentiable Rule Learning
Matthias Wilms, Sascha Xu, Jilles Vreeken
Abstract
Mixture models excel at decomposing complex, multi-modal distributions into simpler probabilistic components, but provide no insight into the conditions under which these components arise. We introduce explainable mixture models (XMM), a framework that pairs each mixture component with a human-interpretable rule over descriptive features. This enables mixtures that are not only statistically expressive but also transparently grounded in the underlying data. We formalize the problem and examine conditions under which an XMM exactly captures a target distribution. We then propose a scalable, differentiable learning procedure for discovering sets of rules. Experiments on synthetic and real-world datasets demonstrate that our method discovers interesting sub-populations in both univariate and multivariate settings, offering interpretable insights into the structure of complex distributions.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on1
Related papers
- Unsupervised Causal Binary Concepts Discovery with VAE for Black-Box Model ExplanationThien Q. Tran, Kazuto Fukuchi, Youhei Akimoto, Jun SakumaAAAI 2022 · 11 citations
- What's in the Box? Exploring the Inner Life of Neural Networks with Robust RulesJonas Fischer, Anna Oláh, Jilles VreekenICML 2021 · 11 citations
- DisDiff: Unsupervised Disentanglement of Diffusion Probabilistic ModelsTao Yang, Yuwang Wang, Yan Lu, Nanning ZhengNeurIPS 2023 · 41 citations
- Explaining Latent Representations with a Corpus of ExamplesJonathan Crabbé, Zhaozhi Qian, Fergus Imrie, Mihaela van der SchaarNeurIPS 2021 · 48 citations
- Intrinsic User-Centric Interpretability through Global Mixture of ExpertsVinitra Swamy, Syrielle Montariol, Julian Blackwell, Jibril Frej et al.ICLR 2025
