中文

通过相关性图的交互式可视化提升三维卷积神经网络可理解性:在阿尔茨海默病中的评估

图像与视频处理 2021-11-08 v5 计算机视觉与模式识别

摘要

背景:尽管卷积神经网络(CNN)在基于磁共振成像(MRI)扫描检测阿尔茨海默病(AD)痴呆方面达到了很高的诊断准确率,但其尚未应用于临床常规。一个重要原因是缺乏模型可理解性。近期开发的用于推导 CNN 相关性图的可视化方法可能有助于填补这一空白。我们研究了准确率更高的模型是否也更多依赖于由先验知识预定义的判别性脑区。方法:我们在 N=663 例痴呆及遗忘性轻度认知障碍(MCI)患者的 T1 加权 MRI 扫描上训练了用于 AD 检测的 CNN,并通过交叉验证和包含 N=1655 例的三个独立样本验证了模型的准确性。我们评估了相关性评分与海马体积之间的关联,以验证该方法的临床效用。为提升模型可理解性,我们实现了三维 CNN 相关性图的交互式可视化。结果:在三个独立数据集中,AD 痴呆与对照组之间的分组区分显示出高准确率(AUC\geq0.92),MCI 与对照组之间为中等准确率(AUC\approx0.75)。相关性图表明,海马萎缩被视为 AD 检测中最具信息量的因素,其他皮质与皮质下区域的萎缩亦有贡献。海马内的相关性评分与海马体积高度相关(Pearson 相关系数 r\approx-0.86,p<0.001)。结论:相关性图突出了我们事先假设的萎缩区域。这增强了以纯数据驱动方式基于扫描与诊断标签训练的 CNN 模型的可理解性。

关键词

引用

@article{arxiv.2012.10294,
  title  = {Improving 3D convolutional neural network comprehensibility via interactive visualization of relevance maps: Evaluation in Alzheimer's disease},
  author = {Martin Dyrba and Moritz Hanzig and Slawek Altenstein and Sebastian Bader and Tommaso Ballarini and Frederic Brosseron and Katharina Buerger and Daniel Cantré and Peter Dechent and Laura Dobisch and Emrah Düzel and Michael Ewers and Klaus Fliessbach and Wenzel Glanz and John-Dylan Haynes and Michael T. Heneka and Daniel Janowitz and Deniz B. Keles and Ingo Kilimann and Christoph Laske and Franziska Maier and Coraline D. Metzger and Matthias H. Munk and Robert Perneczky and Oliver Peters and Lukas Preis and Josef Priller and Boris Rauchmann and Nina Roy and Klaus Scheffler and Anja Schneider and Björn H. Schott and Annika Spottke and Eike J. Spruth and Marc-André Weber and Birgit Ertl-Wagner and Michael Wagner and Jens Wiltfang and Frank Jessen and Stefan J. Teipel},
  journal= {arXiv preprint arXiv:2012.10294},
  year   = {2021}
}

备注

24 pages, 9 figures/tables, supplementary material, source code available on GitHub