Interpretation of Prediction Models Using the Input Gradient
About
State of the art machine learning algorithms are highly optimized to provide the optimal prediction possible, naturally resulting in complex models. While these models often outperform simpler more interpretable models by order of magnitudes, in terms of understanding the way the model functions, we are often facing a "black box". In this paper we suggest a simple method to interpret the behavior of any predictive model, both for regression and classification. Given a particular model, the information required to interpret it can be obtained by studying the partial derivatives of the model with respect to the input. We exemplify this insight by interpreting convolutional and multi-layer neural networks in the field of natural language processing.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Referring Segmentation | RefCOCO (val) | -- | 84 | |
| Image Segmentation | COCO | mIoU21 | 39 | |
| Visual Attribution | Thinking-Model Attribution Dataset Science | LDS73 | 24 | |
| Visual Attribution | Thinking-Model Attribution Dataset General | LDS0.71 | 24 | |
| Visual Attribution | Thinking-Model Attribution Dataset Document | LDS0.7 | 24 | |
| Visual Attribution | Thinking-Model Attribution Dataset Math | LDS63 | 24 | |
| Visual Attribution | Thinking-Model Attribution Dataset Code | LDS62 | 24 | |
| Visual Attribution | Thinking-Model Attribution Dataset Overall | Avg. LDS64 | 24 | |
| Text Classification Explanation | Reuters (test) | Delta Acc (Top-1)8.8 | 21 | |
| Text Classification Explanation | IMDB (test) | Delta Acc (Top-1)0.036 | 21 |