Search papers, labs, and topics across Lattice.
The paper introduces Conditionally Additive Local Models (CALMs), a new model class that extends GAMs by enabling multiple univariate shape functions per feature, each active in different regions of the input space defined by logical conditions on interacting features. This approach aims to balance the interpretability of GAMs with the accuracy of GA$^2$Ms by maintaining local additivity while capturing interactions through region-specific effects. A distillation-based training pipeline is proposed to identify homogeneous regions and fit interpretable shape functions, demonstrating superior performance compared to GAMs and comparable accuracy to GA$^2$Ms across various tasks.
Finally, a model that rivals GA$^2$Ms in accuracy without sacrificing the interpretability of GAMs.
Generalized additive models (GAMs) offer interpretability through independent univariate feature effects but underfit when interactions are present in data. GA$^2$Ms add selected pairwise interactions which improves accuracy, but sacrifices interpretability and limits model auditing. We propose \emph{Conditionally Additive Local Models} (CALMs), a new model class, that balances the interpretability of GAMs with the accuracy of GA$^2$Ms. CALMs allow multiple univariate shape functions per feature, each active in different regions of the input space. These regions are defined independently for each feature as simple logical conditions (thresholds) on the features it interacts with. As a result, effects remain locally additive while varying across subregions to capture interactions. We further propose a principled distillation-based training pipeline that identifies homogeneous regions with limited interactions and fits interpretable shape functions via region-aware backfitting. Experiments on diverse classification and regression tasks show that CALMs consistently outperform GAMs and achieve accuracy comparable with GA$^2$Ms. Overall, CALMs offer a compelling trade-off between predictive accuracy and interpretability.
