Machine learning application for classification of Alzheimer's disease stages using (18)F-flortaucipir positron emission tomography

BACKGROUND: The progression of Alzheimer’s dementia (AD) can be classified into three stages: cognitive unimpairment (CU), mild cognitive impairment (MCI), and AD. The purpose of this study was to implement a machine learning (ML) framework for AD stage classification using the standard uptake value ratio (SUVR) extracted from (18)F-flortaucipir positron emission tomography (PET) images. We demonstrate the utility of tau SUVR for AD stage classification. We used clinical variables (age, sex, education, mini-mental state examination scores) and SUVR extracted from PET images scanned at baseline. Four types of ML frameworks, such as logistic regression, support vector machine (SVM), extreme gradient boosting, and multilayer perceptron (MLP), were used and explained by Shapley Additive Explanations (SHAP) to classify the AD stage. RESULTS: Of a total of 199 participants, 74, 69, and 56 patients were in the CU, MCI, and AD groups, respectively; their mean age was 71.5 years, and 106 (53.3%) were men. In the classification between CU and AD, the effect of clinical and tau SUVR was high in all classification tasks and all models had a mean area under the receiver operating characteristic curve (AUC) > 0.96. In the classification between MCI and AD, the independent effect of tau SUVR in SVM had an AUC of 0.88 (p < 0.05), which was the highest compared to other models. In the classification between MCI and CU, the AUC of each classification model was higher with tau SUVR variables than with clinical variables independently, which yielded an AUC of 0.75(p < 0.05) in MLP, which was the highest. As an explanation by SHAP for the classification between MCI and CU, and AD and CU, the amygdala and entorhinal cortex greatly affected the classification results. In the classification between MCI and AD, the para-hippocampal and temporal cortex affected model performance. Especially entorhinal cortex and amygdala showed a higher effect on model performance than all clinical variables in the classification between MCI and CU. CONCLUSIONS: The independent effect of tau deposition indicates that it is an effective biomarker in classifying CU and MCI into clinical stages using MLP. It is also very effective in classifying AD stages using SVM with clinical information that can be easily obtained at clinical screening. SUPPLEMENTARY INFORMATION: The online version contains supplementary material available at 10.1186/s12938-023-01107-w.