M3d-CAM: A PyTorch library to generate 3D data attention maps for medical deep learning
arXiv:2007.00453
Abstract
M3d-CAM is an easy to use library for generating attention maps of CNN-based PyTorch models improving the interpretability of model predictions for humans. The attention maps can be generated with multiple methods like Guided Backpropagation, Grad-CAM, Guided Grad-CAM and Grad-CAM++. These attention maps visualize the regions in the input data that influenced the model prediction the most at a certain layer. Furthermore, M3d-CAM supports 2D and 3D data for the task of classification as well as for segmentation. A key feature is also that in most cases only a single line of code is required for generating attention maps for a model making M3d-CAM basically plug and play.
References in corpus (6)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networks
- Striving for Simplicity: The All Convolutional Net
- COVID-19 Image Data Collection
- A large annotated medical image dataset for the development and evaluation of segmentation algorithms
- nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation