Deep Joint Learning of Pathological Region Localization and Alzheimer's Disease Diagnosis

Changhyun Park,Heung-Il Suk
DOI: https://doi.org/10.48550/arXiv.2108.04555
2021-08-10
Abstract:The identification of Alzheimer's disease (AD) and its early stages using structural magnetic resonance imaging (MRI) has been attracting the attention of researchers. Various data-driven approaches have been introduced to capture subtle and local morphological changes of the brain accompanied by the disease progression. One of the typical approaches for capturing subtle changes is patch-level feature representation. However, the predetermined regions to extract patches can limit classification performance by interrupting the exploration of potential biomarkers. In addition, the existing patch-level analyses have difficulty explaining their decision-making. To address these problems, we propose the BrainBagNet with a position-based gate (PG-BrainBagNet), a framework for jointly learning pathological region localization and AD diagnosis in an end-to-end manner. In advance, as all scans are aligned to a template in image processing, the position of brain images can be represented through the 3D Cartesian space shared by the overall MRI scans. The proposed method represents the patch-level response from whole-brain MRI scans and discriminative brain-region from position information. Based on the outcomes, the patch-level class evidence is calculated, and then the image-level prediction is inferred by a transparent aggregation. The proposed models were evaluated on the ADNI datasets. In five-fold cross-validation, the classification performance of the proposed method outperformed that of the state-of-the-art methods in both AD diagnosis (AD vs. normal control) and mild cognitive impairment (MCI) conversion prediction (progressive MCI vs. stable MCI) tasks. In addition, changes in the identified discriminant regions and patch-level class evidence according to the patch size used for model training are presented and analyzed.
Artificial Intelligence
What problem does this paper attempt to address?
This paper aims to solve the problem of identifying Alzheimer's disease (AD) and its early stages, especially in terms of identification using structural magnetic resonance imaging (MRI). Specifically, the paper focuses on how to capture the subtle and local morphological changes accompanying the disease progression through MRI to improve the accuracy of AD diagnosis. The existing patch - level feature representation methods have two main problems: First, the preset region - extraction patches may limit the classification performance because they prevent the exploration of potential biomarkers; second, the existing patch - level analysis is difficult to interpret its decision - making process because the final decision is made through the interaction of non - linear patch - level feature representations. To overcome these problems, the authors propose a framework named PG - BrainBagNet, which can jointly learn pathological region localization and AD diagnosis and is implemented in an end - to - end manner. The key innovation points of this method are as follows: 1. **Utilization of position information**: Since all scans are aligned to a template, the position of brain images can be represented through the shared 3D Cartesian space. PG - BrainBagNet combines the position information of the coronal, sagittal and axial planes, thereby representing the volumetric position information of MRI scans more accurately. 2. **Patch - level response representation**: This method represents patch - level responses from whole - brain MRI scans and extracts discriminative brain regions from position information. Based on these results, patch - level class evidence is calculated and image - level predictions are inferred through a transparent aggregation method. 3. **Transparent decision aggregation**: By transparently aggregating patch - level responses, PG - BrainBagNet not only improves the classification performance but also provides patch - level class evidence, enhancing the interpretability of decisions. 4. **Performance verification**: The model was evaluated on the Alzheimer's Disease Neuroimaging Initiative (ADNI) datasets (i.e., ADNI - 1 and ADNI - 2). In five - fold cross - validation, the classification performance of this method in AD diagnosis (AD vs. normal control group) and mild cognitive impairment (MCI) conversion prediction (progressive MCI vs. stable MCI) tasks is better than that of existing methods. In conclusion, this paper proposes a new deep - learning framework that can improve the accuracy of AD diagnosis while enhancing the interpretability of the model, providing strong support for clinical applications.