ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
5.0k
Citations
534
Influential Citations
PLoS ONE
Venue
2015
Year
Understanding and interpreting classification decisions of automated image classification systems is of high value in many applications, as it allows to verify the reasoning of the system and provides additional information to the human expert. Although machine learning methods are solving very successfully a plethora of tasks, they have in most cases the disadvantage of acting as a black box, not providing any information about what made them arrive at a particular decision. This work proposes a general solution to the problem of understanding classification decisions by pixel-wise decomposition of nonlinear classifiers. We introduce a methodology that allows to visualize the contributions of single pixels to predictions for kernel-based classifiers over Bag of Words features and for multilayered neural networks. These pixel contributions can be visualized as heatmaps and are provided to a human expert who can intuitively not only verify the validity of the classification decision, but also focus further analysis on regions of potential interest. We evaluate our method for classifiers trained on PASCAL VOC 2009 images, synthetic image data containing geometric shapes, the MNIST handwritten digits data set and for the pre-trained ImageNet model available as part of the Caffe open source package.
This paper addresses a critical gap in machine learning: the lack of interpretability in nonlinear classifiers, particularly deep neural networks. As AI systems increasingly impact high-stakes domains like medical imaging and autonomous driving, understanding why a model makes a specific decision is as important as the decision itself. Prior to this work, most explanation methods were limited to linear models or simple feature importance measures. By introducing Layer-Wise Relevance Propagation (LRP), the authors provided a principled, theoretically grounded approach to decompose any nonlinear classifier's output into pixel-level contributions.
The significance is amplified by the paper's timing (2015), when deep learning was rapidly gaining traction but interpretability tools were scarce. LRP offered a practical solution that could be applied to state-of-the-art models like the Caffe ImageNet network, making it immediately useful to practitioners. The method's generality—covering both kernel methods and neural networks—ensured broad applicability across the field.
The paper evaluates LRP on four datasets: synthetic geometric shapes (where ground truth is known), MNIST handwritten digits, PASCAL VOC 2009 object detection, and the ImageNet ILSVRC-2012 challenge using the Caffe reference model. On synthetic data, LRP heatmaps correctly highlight the shape's interior and edges. For MNIST, the method focuses on the digit's stroke regions. On PASCAL VOC, LRP identifies object parts (e.g., wheels of a car) that drive classification. For ImageNet, the heatmaps align with human-annotated object bounding boxes, demonstrating practical utility. No quantitative metrics like accuracy or IoU are reported; evaluation is qualitative via visual inspection.
LRP has become a cornerstone of explainable AI, inspiring numerous follow-up works such as DeepLIFT, Integrated Gradients, and Grad-CAM. Its conservation principle influenced the theoretical understanding of attribution methods. The paper's clear exposition and open-source implementation accelerated adoption in fields like medical imaging, where interpretability is mandatory. Today, LRP remains a standard baseline for evaluating new explanation techniques and is integrated into popular frameworks like iNNvestigate and Captum.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba