Multi-branch Multi-task 3D-CNN for Alzheimer's Disease Detection.

Junhu Li,Beiji Zou,Ziwen Xu,Qing Liu
DOI: https://doi.org/10.1007/978-3-030-88010-1_52
2021-01-01
Abstract:Mini-Mental State Examination (MMSE) and Magnetic Resonance Imaging (MRI) have been widely used for the diagnosis of Alzheimer's disease (AD) in clinical. The former provides a brief way to assess the severity of cognitive impairment while the latter provides a non-invasive way to enable doctors look inside the brain tissues e.g. cerebrospinal fluid (CSF), gray matter (GM), and white matter (WM) and rule out the possible causes of AD. Different brain tissues response for different cognitive functions, thus the abnormal of brain tissues and MMSE scores are usually highly related. To make full use of different brain tissues and the relationship between state of brain tissues and MMSE scores, this paper proposes a multi-branch multi-task learning framework for AD detection. In detail, three parallel 3D-CNN branches are used to learn deep features from individual brain tissues of CSF, GM and WM segmented from 3D MRI brain scans. Then the tissue-specific features are concatenated and used for AD prediction and MMSE score regression. To boost the feature learning, auxiliary predictions are attached on the top of tissue-specific features and auxiliary supervision are performed. We demonstrate our method on ADNI datasets which includes 415 subjects and conduct 10 repeated experiments to avoid the bias caused by data division. Experimental results show that our method without auxiliary predictions achieves the average accuracy of 89.63% and method with auxiliary predictions achieves average of accuracy of 91%. Both of them outperform the state-of-the-art AD detection methods.
What problem does this paper attempt to address?