Back

Explainable artificial intelligence for neuroimaging-based dementia diagnosis and prognosis

Martin, S. A.; Zhao, A.; Qu, J.; Imms, P.; Irimia, A.; Barkhof, F.; Cole, J. H.; Alzheimer's Disease Neuroimaging Initiative,

2025-01-14 radiology and imaging
10.1101/2025.01.13.25320382 medRxiv
Show abstract

INTRODUCTION: Artificial intelligence and neuroimaging enable accurate dementia prediction, but black box models can be difficult to trust. Explainable artificial intelligence (XAI) describes techniques to understand model behaviour and the influence of features, however deciding which method is most appropriate is non-trivial. Vision transformers (ViT) have also gained popularity, providing a self-explainable, alternative to traditional convolutional neural networks (CNN). METHODS: We used T1-weighted MRI to train models on two tasks: Alzheimers disease (AD) classification (diagnosis) and predicting conversion from mild-cognitive impairment (MCI) to AD (prognosis). We compared ten XAI methods across CNN and ViT architectures. RESULTS: Models achieved balanced accuracies of 81% and 67% for diagnosis and prognosis. XAI outputs highlighted brain regions relevant to AD and contained useful information for MCI prognosis. DISCUSSION: XAI can be used to verify that models are utilising relevant features and to generate valuable measures for further analysis.

Matching journals

The top 8 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.