Multi-Task Weakly-Supervised Attention Network for Dementia Status Estimation With Structural MRI

Chunfeng Lian,Mingxia Liu,Li Wang,Dinggang Shen
DOI: https://doi.org/10.1109/tnnls.2021.3055772
IF: 14.255
2021-01-01
IEEE Transactions on Neural Networks and Learning Systems
Abstract:Accurate prediction of clinical scores (of neuropsychological tests) based on noninvasive structural magnetic resonance imaging (MRI) helps understand the pathological stage of dementia (e.g., Alzheimer's disease (AD)) and forecast its progression. Existing machine/deep learning approaches typically preselect dementia-sensitive brain locations for MRI feature extraction and model construction, potentially leading to undesired heterogeneity between different stages and degraded prediction performance. Besides, these methods usually rely on prior anatomical knowledge (e.g., brain atlas) and time-consuming nonlinear registration for the preselection of brain locations, thereby ignoring individual-specific structural changes during dementia progression because all subjects share the same preselected brain regions. In this article, we propose a multi-task weakly-supervised attention network (MWAN) for the joint regression of multiple clinical scores from baseline MRI scans. Three sequential components are included in MWAN: 1) a backbone fully convolutional network for extracting MRI features; 2) a weakly supervised dementia attention block for automatically identifying subject-specific discriminative brain locations; and 3) an attention-aware multitask regression block for jointly predicting multiple clinical scores. The proposed MWAN is an end-to-end and fully trainable deep learning model in which dementia-aware holistic feature learning and multitask regression model construction are integrated into a unified framework. Our MWAN method was evaluated on two public AD data sets for estimating clinical scores of mini-mental state examination (MMSE), clinical dementia rating sum of boxes (CDRSB), and AD assessment scale cognitive subscale (ADAS-Cog). Quantitative experimental results demonstrate that our method produces superior regression performance compared with state-of-the-art methods. Importantly, qualitative results indic-te that the dementia-sensitive brain locations automatically identified by our MWAN method well retain individual specificities and are biologically meaningful.
computer science, artificial intelligence, theory & methods,engineering, electrical & electronic, hardware & architecture
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to non - invasively and accurately predict the clinical scores of neuropsychological tests based on structural magnetic resonance imaging (MRI) in the clinical assessment of dementias such as Alzheimer's disease (AD). Existing methods usually require pre - selecting brain regions sensitive to dementia for MRI feature extraction and model construction, which may lead to inconsistencies between different stages and rely on prior anatomical knowledge (such as brain atlases) and time - consuming nonlinear registration, thus ignoring the specific structural changes of individuals during the progression of dementia. In addition, these methods face challenges when dealing with continuous variable regression tasks, that is, predicting clinical scores. To this end, the paper proposes a Multi - Task Weakly - Supervised Attention Network (MWAN), aiming to automatically identify discriminative brain regions related to an individual's specific dementia state directly from whole - brain MRI scans and, on this basis, jointly predict multiple clinical scores. MWAN consists of three sequential components: 1) a backbone fully convolutional network (FCN) for extracting MRI features; 2) a weakly - supervised dementia attention block for automatically identifying individual - specific discriminative brain regions; 3) an attention - aware multitask regression block for jointly predicting multiple clinical scores. MWAN is an end - to - end and fully trainable deep - learning model, in which the overall dementia - aware feature learning and multi - task regression model construction are integrated into a unified framework. In this way, the paper aims to improve the accuracy of clinical score prediction while retaining individual specificity and ensuring biological significance.