A Multimodal MRI-Based Machine Learning Framework for Classifying Cognitive Impairment in Cerebral Small Vessel Disease
Author Block: G. Lin, W. Chen, M. Chen, J. Ji; Lishui/CN
Purpose: This study aims to propose a multimodal magnetic resonance imaging (MRI)-based machine learning framework to effectively classify mild cognitive impairment (MCI) and no cognitive impairment (NCI) in patients with cerebral small vessel disease (CSVD).
Methods or Background: We enrolled 223 patients with CSVD, categorized into NCI (n = 121) and MCI (n = 102) groups based on neurocognitive assessments. Multimodal MRI data, including T1-weighted, resting-state functional MRI, and diffusion tensor images, were collected. Image preprocessing, feature extraction, and feature selection methods were applied to obtain MRI features from the three modalities. The AutoGluon platform was utilized for model development, and traditional machine learning algorithms were applied for comparison. The models were validated using a validation cohort of 97 patients with CSVD, and their performance was assessed via receiver operating characteristic curve (ROC) analysis.
Results or Findings: The AutoGluon model to distinguish MCI from NCI based on multimodal MRI features demonstrated a high area under the ROC curve (AUC), accuracy, sensitivity, specificity, and F1-score in the testing set (0.894, 85.65%, 84.31%, 86.78%, and 84.31%, respectively) and validation cohort (0.846, 79.38%, 81.82%, 77.36%, and 78.26%, respectively). Other models built using traditional machine learning algorithms had AUCs of 0.661–0.732, and their prediction accuracies were significantly lower than that of the AutoGluon model (P < 0.001).
Conclusion: Our study provides a multimodal MRI-based machine learning framework, utilizing the AutoGluon platform, that outperforms traditional algorithms in classifying MCI and NCI in patients with CSVD, offering a promising tool for the early prediction of MCI in CSVD.
Limitations: As a retrospective study, it is susceptible to selection bias, which may limit its generalizability.
Funding for this study: This study is supported by Zhejiang Public Welfare Research Program (LGF20H220002, LGF19H180010), and Zhejiang Provincial Healthcare Program (2024KY562)
Has your study been approved by an ethics committee? Yes
Ethics committee - additional information: This study was approved by the Ethics Committee of the Fifth Affiliated Hospital of Wenzhou Medical University (approval number: 2024-266)