Development and validation of an automatic classification algorithm for the diagnosis of Alzheimer's disease using a high-performance interpretable deep learning network

Abstract

Objectives: To develop and validate an automatic classification algorithm for diagnosing Alzheimer's disease (AD) or mild cognitive impairment (MCI).

Methods and materials: This study evaluated a high-performance interpretable network algorithm (TabNet) and compared its performance with that of XGBoost, a widely used classifier. Brain segmentation was performed using a commercially approved software. TabNet and XGBoost were trained on the volumes or radiomics features of 102 segmented regions for classifying subjects into AD, MCI, or cognitively normal (CN) groups. The diagnostic performances of the two algorithms were compared using areas under the curves (AUCs). Additionally, 20 deep learning-based AD signature areas were investigated.

Results: Between December 2014 and March 2017, 161 AD, 153 MCI, and 306 CN cases were enrolled. Another 120 AD, 90 MCI, and 141 CN cases were included for the internal validation. Public datasets were used for external validation. TabNet with volume features had an AUC of 0.951 (95% confidence interval [CI], 0.947-0.955) for AD vs CN, which was similar to that of XGBoost (0.953 [95% CI, 0.951-0.955], p = 0.41). External validation revealed the similar performances of two classifiers using volume features (0.871 vs. 0.871, p = 0.86). Likewise, two algorithms showed similar performances with one another in classifying MCI. The addition of radiomics data did not improve the performance of TabNet. TabNet and XGBoost focused on the same 13/20 regions of interest, including the hippocampus, inferior lateral ventricle, and entorhinal cortex.

Conclusions: TabNet shows high performance in AD classification and detailed interpretation of the selected regions.

Clinical relevance statement: Using a high-performance interpretable deep learning network, the automatic classification algorithm assisted in accurate Alzheimer's disease detection using 3D T1-weighted brain MRI and detailed interpretation of the selected regions.

Key points: • MR volumetry data revealed that TabNet had a high diagnostic performance in differentiating Alzheimer's disease (AD) from cognitive normal cases, which was comparable with that of XGBoost. • The addition of radiomics data to the volume data did not improve the diagnostic performance of TabNet. • Both TabNet and XGBoost selected the clinically meaningful regions of interest in AD, including the hippocampus, inferior lateral ventricle, and entorhinal cortex.

Keywords: Alzheimer disease; Deep learning; Machine learning.