When Interpretability Meets Generalization: Delta-GAM for Robust Extrapolation in Out-of-Distribution Settings
Linxiao Yang, Wenwei Wang, Qiming Chen, Zhipeng Zeng, Liang Sun
Abstract
Out-of-Distribution (OOD) extrapolation, where test data feature values extend beyond the training range, poses significant challenges in machine learning. While existing solutions often sacrifice interpretability, resulting in limited applicability in high-stakes applications where interpretability is a a critical requirement. In this paper, we propose Delta-GAM, an interpretable Generalized Additive Model (GAM) that achieves robust extrapolation in OOD scenarios. Our method jointly learns (1) feature-target relationships and (2) functional adaptations for extrapolating beyond the training distribution by reformulating GAM fitting as a second-order interaction problem between features and their distributional offsets. We theoretically show that smooth GAM shape functions induce an approximately low-rank structure in these interactions, enabling efficient decomposition via a specialized neural network. Experiments on synthetic and real-world data demonstrate Delta-GAM's superior performance in OOD extrapolation tasks while preserving model interpretability, bridging a key gap in trustworthy machine learning.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get 2150e4e1-29ac-4115-afd0-120fcb9032fcRelated papers
- GRAND-SLAMIN' Interpretable Additive Modeling with Structural ConstraintsShibal Ibrahim, Gabriel Afriat, Kayhan Behdin, Rahul MazumderNeurIPS 2023 · 15 citations
- Neural Basis Models for InterpretabilityFilip Radenovic, Abhimanyu Dubey, Dhruv MahajanNeurIPS 2022 · 82 citations
- pureGAM: Learning an Inherently Pure Additive ModelXingzhi Sun, Ziyu Wang, Rui Ding, Shi Han et al.KDD 2022 · 4 citations
- NODE-GAM: Neural Generalized Additive Model for Interpretable Deep LearningChun-Hao Chang, Rich Caruana, Anna GoldenbergICLR 2022 · 114 citations
- Scalable Interpretability via PolynomialsAbhimanyu Dubey, Filip Radenovic, Dhruv MahajanNeurIPS 2022 · 42 citations
