NODE-GAM: Neural Generalized Additive Model for Interpretable Deep Learning

06/03/2021
by   Chun-Hao Chang, et al.
0

Deployment of machine learning models in real high-risk settings (e.g. healthcare) often depends not only on model's accuracy but also on its fairness, robustness and interpretability. Generalized Additive Models (GAMs) have a long history of use in these high-risk domains, but lack desirable features of deep learning such as differentiability and scalability. In this work, we propose a neural GAM (NODE-GAM) and neural GA^2M (NODE-GA^2M) that scale well to large datasets, while remaining interpretable and accurate. We show that our proposed models have comparable accuracy to other non-interpretable models, and outperform other GAMs on large datasets. We also show that our models are more accurate in self-supervised learning setting when access to labeled data is limited.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset