Improved image classification explainability with high-accuracy heatmaps.

ABSTRACT: Deep learning models have become increasingly used for image-based classification. In critical applications such as medical imaging, it is important to convey the reasoning behind the models' decisions in human-understandable forms. In this work, we propose Pyramid Localization Network (PYLON), a deep learning model that delivers precise location explanation by increasing the resolution of heatmaps produced by class activation map (CAM). PYLON substantially improves the quality of CAM's heatmaps in both general image and medical image domains and excels at pinpointing the locations of small objects. Most importantly, PYLON does not require expert annotation of the object location but instead can be trained using only image-level label. This capability is especially important for domain whe

SUBMITTER: Preechakul K 

PROVIDER: S-EPMC8889368 | biostudies-literature | 2022 Mar

REPOSITORIES: biostudies-literature