TU Darmstadt / ULB / TUprints

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

Dyrba, Martin ; Hanzig, Moritz ; Altenstein, Slawek ; Bader, Sebastian ; Ballarini, Tommaso ; Brosseron, Frederic ; Buerger, Katharina ; Cantré, Daniel ; Dechent, Peter ; Dobisch, Laura ; Düzel, Emrah ; Ewers, Michael ; Fliessbach, Klaus ; Glanz, Wenzel ; Haynes, John-Dylan ; Heneka, Michael T. ; Janowitz, Daniel ; Keles, Deniz B. ; Kilimann, Ingo ; Laske, Christoph ; Maier, Franziska ; Metzger, Coraline D. ; Munk, Matthias H. ; Perneczky, Robert ; Peters, Oliver ; Preis, Lukas ; Priller, Josef ; Rauchmann, Boris ; Roy, Nina ; Scheffler, Klaus ; Schneider, Anja ; Schott, Björn H. ; Spottke, Annika ; Spruth, Eike J. ; Weber, Marc-André ; Ertl-Wagner, Birgit ; Wagner, Michael ; Wiltfang, Jens ; Jessen, Frank ; Teipel, Stefan J. (2024)
Improving 3D convolutional neural network comprehensibility via interactive visualization of relevance maps: evaluation in Alzheimer’s disease.
In: Alzheimer's Research & Therapy, 2021, 13 (1)
doi: 10.26083/tuprints-00023419
Article, Secondary publication, Publisher's Version

[img] Text
s13195-021-00924-2.pdf
Copyright Information: CC BY 4.0 International - Creative Commons, Attribution.

Download (2MB)
Item Type: Article
Type of entry: Secondary publication
Title: Improving 3D convolutional neural network comprehensibility via interactive visualization of relevance maps: evaluation in Alzheimer’s disease
Language: English
Date: 24 September 2024
Place of Publication: Darmstadt
Year of primary publication: 23 November 2021
Place of primary publication: London
Publisher: BioMed Central
Journal or Publication Title: Alzheimer's Research & Therapy
Volume of the journal: 13
Issue Number: 1
Collation: 18 Seiten
DOI: 10.26083/tuprints-00023419
Corresponding Links:
Origin: Secondary publication DeepGreen
Abstract:

Background: Although convolutional neural networks (CNNs) 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 as they allow the visualization of key input image features that drive the decision of the model. 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 in total 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, thereby allowing intuitive model inspection.

Results: Across the three independent datasets, group separation showed high accuracy for AD dementia versus controls (AUC ≥ 0.91) and moderate accuracy for amnestic MCI versus controls (AUC ≈ 0.74). Relevance maps indicated that hippocampal atrophy was considered 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 ≈ −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. The high hippocampus relevance scores as well as the high performance achieved in independent samples support the validity of the CNN models in the detection of AD-related MRI abnormalities. The presented data-driven and hypothesis-free CNN modeling approach might provide a useful tool to automatically derive discriminative features for complex diagnostic tasks where clear clinical criteria are still missing, for instance for the differential diagnosis between various types of dementia.

Uncontrolled Keywords: Alzheimer’s disease, Deep learning, Convolutional neural network, MRI, Layer-wise relevance propagation
Identification Number: Artikel-ID: 191
Status: Publisher's Version
URN: urn:nbn:de:tuda-tuprints-234199
Additional Information:

Part of Springer Nature

Classification DDC: 100 Philosophy and psychology > 150 Psychology
500 Science and mathematics > 570 Life sciences, biology
600 Technology, medicine, applied sciences > 610 Medicine and health
Divisions: 10 Department of Biology > Systems Neurophysiology
Date Deposited: 24 Sep 2024 11:30
Last Modified: 26 Sep 2024 07:46
SWORD Depositor: Deep Green
URI: https://tuprints.ulb.tu-darmstadt.de/id/eprint/23419
PPN: 521700035
Export:
Actions (login required)
View Item View Item