Mixtures of Transparent Local Models
Niffa Cheick Oumar Diaby, Thierry Duchesne, Mario Marchand
TL;DR
This work introduces Mixtures of Transparent Local Models (MoTLM), a framework for interpretable prediction that partitions the input space around points of interest and learns simple local linear predictors within each locality, with an external predictor for regions outside all localities. Grounded in the PAC-Bayesian theory, MoTLM optimizes a loss that blends local predictions with locality-aware gating and provides rigorous risk bounds via KL-divergence between posterior and prior distributions. The authors derive closed-form loss and KL terms for binary classification and linear regression with known PoIs, and extend to unknown PoIs using non-central chi-square and related functions, employing Quasi-Monte Carlo to approximate complex terms. Empirically, MoTLM achieves competitive accuracy on synthetic and real datasets while maintaining interpretability, outperforming or matching linear baselines and approaching kernel-based opaque models in some settings. The approach offers a principled, interpretable alternative to post-hoc explanations and suggests promising directions for extending to other transparent models and distance metrics.
Abstract
The predominance of machine learning models in many spheres of human activity has led to a growing demand for their transparency. The transparency of models makes it possible to discern some factors, such as security or non-discrimination. In this paper, we propose a mixture of transparent local models as an alternative solution for designing interpretable (or transparent) models. Our approach is designed for the situations where a simple and transparent function is suitable for modeling the label of instances in some localities/regions of the input space, but may change abruptly as we move from one locality to another. Consequently, the proposed algorithm is to learn both the transparent labeling function and the locality of the input space where the labeling function achieves a small risk in its assigned locality. By using a new multi-predictor (and multi-locality) loss function, we established rigorous PAC-Bayesian risk bounds for the case of binary linear classification problem and that of linear regression. In both cases, synthetic data sets were used to illustrate how the learning algorithms work. The results obtained from real data sets highlight the competitiveness of our approach compared to other existing methods as well as certain opaque models. Keywords: PAC-Bayes, risk bounds, local models, transparent models, mixtures of local transparent models.
