Interpreting and improving deep-learning models with reality checks

Chandan Singh, Wooseok Ha, Bin Yu
8/16/2021
stat.MLcs.LG

Official code: laura-rieger/deep-explanation-penalization (127 stars, MIT, PyTorch, last commit 2021-03-22). 3 more implementations below.

Abstract

Recent deep-learning models have achieved impressive predictive performance by learning complex functions of many variables, often at the cost of interpretability. This chapter covers recent work aiming to interpret models by attributing importance to features and feature groups for a single prediction. Importantly, the proposed attributions assign importance to interactions between features, in addition to features in isolation. These attributions are shown to yield insights across real-world domains, including bio-imaging, cosmology image and natural-language processing. We then show how these attributions can be used to directly improve the generalization of a neural network or to distill it into a simple model. Throughout the chapter, we emphasize the use of reality checks to scrutinize the proposed interpretation techniques.

View on arXivView PDF

Code Implementations(4)

PyTorchMIT

Code for using CDEP from the paper "Interpretations are useful: penalizing explanations to align neural networks with prior knowledge" https://arxiv.org/abs/1909.13584

12714Python, Jupyter NotebookFeb 12, 20195 years agoMIT
aiartificial-intelligencecdepconvolutional-neural-networkdata-science+15 more
PyTorchMIT

Using / reproducing ACD from the paper "Hierarchical interpretations for neural network predictions" 🧠 (ICLR 2019)

12521TeX, PythonMay 18, 20185 years agoMIT
acdaiartificial-intelligenceconvolutional-neural-networksdata-science+15 more
PyTorchMIT

Adaptive, interpretable wavelets across domains (NeurIPS 2021)

8213Python, Jupyter NotebookFeb 12, 20214 years agoMIT
deep-learningdistillationexplainabilityinterpretabilitymachine-learning+6 more
PyTorchMIT

Using / reproducing TRIM from the paper "Transformation Importance with Applications to Cosmology" 🌌 (ICLR Workshop 2020)

71Shell, PythonNov 19, 20195 years agoMIT
aiartificial-intelligenceattributiondata-sciencedeep-learning+15 more

Similar Papers

Cite this paper

@article{singh2021interpreting,
  title  = {Interpreting and improving deep-learning models with reality checks},
  author = {Chandan Singh and Wooseok Ha and Bin Yu},
  year   = {2021},
  eprint = {2108.06847},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2108.06847}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2108.06847.svg)](https://codeofpaper.com/paper/2108.06847)

Discussion