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.