arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.10294eess.IVcs.CV

Improving 3D convolutional neural network comprehensibility via interactive visualization of relevance maps: Evaluation in Alzheimer's disease

German Center for Neurodegenerative Diseases (DZNE)(德国神经退行性疾病中心);University of Rostock(罗斯托克大学);Charité – Universitätsmedizin Berlin(夏里特医学院);University Hospital Bonn(波恩大学医院);Ludwig Maximilian University(慕尼黑大学);Rostock University Medical Center(罗斯托克大学医学中心);Georg-August-University(哥廷根大学);Otto-von-Guericke University(奥托·冯·古里克大学);Bernstein Center for Computational Neuroscience(伯恩斯坦计算神经科学中心);Hertie Institute for Clinical Brain Research(赫蒂临床脑研究所);University of Tuebingen(图宾根大学);University of Cologne(科隆大学);Darmstadt University of Technology(达姆施塔特工业大学);Imperial College London(伦敦帝国理工学院);Technical University Munich(慕尼黑工业大学);University Medical Center Goettingen(哥廷根大学医学中心);Leibniz Institute for Neurobiology(莱布尼茨神经生物学研究所);University of Toronto(多伦多大学);University of Aveiro(阿威罗大学)
Martin Dyrba, Moritz Hanzig, Slawek Altenstein, Sebastian Bader, Tommaso Ballarini, Frederic Brosseron, Katharina Buerger, Daniel Cantré, Peter Dechent, Laura D… 展开作者

Martin Dyrba, Moritz Hanzig, Slawek Altenstein, Sebastian Bader, Tommaso Ballarini, Frederic Brosseron, Katharina Buerger, Daniel Cantré, Peter Dechent, Laura Dobisch, Emrah Düzel, Michael Ewers, Klaus Fliessbach, Wenzel Glanz, John-Dylan Haynes, Michael T. Heneka, Daniel Janowitz, Deniz B. Keles, Ingo Kilimann, Christoph Laske, Franziska Maier, Coraline D. Metzger, Matthias H. Munk, Robert Perneczky, Oliver Peters, Lukas Preis, Josef Priller, Boris Rauchmann, Nina Roy, Klaus Scheffler, Anja Schneider, Björn H. Schott, Annika Spottke, Eike J. Spruth, Marc-André Weber, Birgit Ertl-Wagner, Michael Wagner, Jens Wiltfang, Frank Jessen, Stefan J. Teipel

英文摘要:

Background: Although convolutional neural networks (CNN) achieve high diagnostic accuracy for detecting Alzheimer's disease (AD) dementia based on magnetic resonance imaging (MRI) scans, they are not yet applied in clinical routine. One important reason for this is a lack of model comprehensibility. Recently developed visualization methods for deriving CNN relevance maps may help to fill this gap. We investigated whether models with higher accuracy also rely more on discriminative brain regions predefined by prior knowledge. Methods: We trained a CNN for the detection of AD in N=663 T1-weighted MRI scans of patients with dementia and amnestic mild cognitive impairment (MCI) and verified the accuracy of the models via cross-validation and in three independent samples including N=1655 cases. We evaluated the association of relevance scores and hippocampus volume to validate the clinical utility of this approach. To improve model comprehensibility, we implemented an interactive visualization of 3D CNN relevance maps. Results: Across three independent datasets, group separation showed high accuracy for AD dementia vs. controls (AUC$\geq$0.92) and moderate accuracy for MCI vs. controls (AUC$\approx$0.75). Relevance maps indicated that hippocampal atrophy was considered as the most informative factor for AD detection, with additional contributions from atrophy in other cortical and subcortical regions. Relevance scores within the hippocampus were highly correlated with hippocampal volumes (Pearson's r$\approx$-0.86, p<0.001). Conclusion: The relevance maps highlighted atrophy in regions that we had hypothesized a priori. This strengthens the comprehensibility of the CNN models, which were trained in a purely data-driven manner based on the scans and diagnosis labels.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑