M3d-CAM：用于生成医学深度学习三维数据注意力图的PyTorch库
计算机视觉与模式识别
2020-07-02 v1 机器学习
摘要
M3d-CAM是一个易于使用的库，用于生成基于CNN的PyTorch模型的注意力图，从而提升人类对其模型预测的可解释性。注意力图可通过多种方法生成，如导向反向传播（Guided Backpropagation）、Grad-CAM、导向Grad-CAM和Grad-CAM++。这些注意力图可视化了在某一特定层对模型预测影响最大的输入数据区域。此外，M3d-CAM支持分类与分割任务下的2D和3D数据。其一个关键特性是，在大多数情况下仅需一行代码即可为模型生成注意力图，使M3d-CAM基本上是即插即用的。
引用
@article{arxiv.2007.00453,
title = {M3d-CAM: A PyTorch library to generate 3D data attention maps for medical deep learning},
author = {Karol Gotkowski and Camila Gonzalez and Andreas Bucher and Anirban Mukhopadhyay},
journal= {arXiv preprint arXiv:2007.00453},
year = {2020}
}