Research on Interpretable Machine Learning Models for Diagnosis and Staging of Mild Cognitive Impairment

Chongyang He¹Yanyan Zhou^1,2Yi Chen^1*Yang Jing³

• ¹Department of Radiology, Chongqing Red Cross Hospital, Jiangbei District People’s Hospital, Chongqing, China., Chong Qing, China
• ²Affiliated Hospital of Zunyi Medical University, Zunyi, China
• ³Huiying Medical Technology Beijing Co Ltd, Beijing, China

Background: Mild cognitive impairment (MCI) is a critical prodromal stage of Alzheimer's disease (AD), further categorized into early MCI (EMCI) and late MCI (LMCI). Early and accurate diagnosis is essential for effective prevention and intervention of AD. This study aims to develop an accessible and interpretable machine learning model to facilitate early diagnosis and subtype staging of MCI. Methods: A total of 268 participants were recruited from the ADNI, including cognitively normal individuals (CN, n=132), EMCI (n=95), and LMCI (n=41). Participants were randomly divided into training (80%) and testing (20%) cohorts. Multimodal data encompassing whole-brain T1-WI MRI radiomics, clinical neuropsychological scales and plasma protein biomarkers were collected. Logistic regression (LR) and random forest (RF) algorithms were employed to construct six unimodal models based on above three categories of features, as well as a combined model combining all features. Diagnostic performance for the three-class classification task (CN, EMCI, LMCI) was evaluated using receiver operating characteristic (ROC) curve. Furthermore, Shapley Additive Explanations (SHAP) were applied to quantify the contribution of individual features within the integrated model. Results: The combined model significantly outperformed unimodal models across all metrics, achieving macro_AUC=0.92, micro_AUC=0.91, and ACC=0.81 in the training set, and macro_AUC=0.87, micro_AUC=0.87, and ACC=0.76 in the testing set. The LR-based radiomics model ranked second. Models based solely on clinical neuropsychological scales or plasma protein biomarkers demonstrated comparatively lower classification performance. SHAP analysis highlighted neuropsychological scales (ADAS-Cog, MoCA) and radiomic features from critical brain regions (hippocampus, middle temporal gyrus, entorhinal cortex) as pivotal contributors to model efficacy. Conclusion: The integration of whole-brain structural MRI (sMRI) radiomics, neuropsychological scales, and plasma protein biomarkers significantly improves the precision of diagnosing and staging mild cognitive impairment (MCI). Radiomic characteristics derived from critical cerebral regions yield valuable pathological information that facilitates clinical interpretation. This methodology presents a promising strategy for the early identification and individualized management of MCI.