English

Understanding Anatomy Classification Through Attentive Response Maps

Computer Vision and Pattern Recognition 2018-02-08 v3

Abstract

One of the main challenges for broad adoption of deep learning based models such as convolutional neural networks (CNN), is the lack of understanding of their decisions. In many applications, a simpler, less capable model that can be easily understood is favorable to a black-box model that has superior performance. In this paper, we present an approach for designing CNNs based on visualization of the internal activations of the model. We visualize the model's response through attentive response maps obtained using a fractional stride convolution technique and compare the results with known imaging landmarks from the medical literature. We show that sufficiently deep and capable models can be successfully trained to use the same medical landmarks a human expert would use. Our approach allows for communicating the model decision process well, but also offers insight towards detecting biases.

Keywords

Cite

@article{arxiv.1611.06284,
  title  = {Understanding Anatomy Classification Through Attentive Response Maps},
  author = {Devinder Kumar and Vlado Menkovski and Graham W. Taylor and Alexander Wong},
  journal= {arXiv preprint arXiv:1611.06284},
  year   = {2018}
}

Comments

Accepted at ISBI, 2018

R2 v1 2026-06-22T16:57:41.057Z