Machine learning trained with quantitative susceptibility mapping to detect mild cognitive impairment in Parkinson's disease

Background: Cognitive decline is commonly observed in Parkinson's disease (PD). Identifying PD with mild cognitive impairment (PD-MCI) is crucial for early initiation of therapeutic interventions and preventing cognitive decline. Objective: We aimed to develop a machine learning model trained with magnetic susceptibility values based on the multi-atlas label-fusion method to classify PD without dementia into PD-MCI and normal cognition (PD-CN). Methods: This multicenter observational cohort study retrospectively reviewed 61 PD-MCI and 59 PD-CN cases for the internal validation cohort and 22 PD-MCI and 21 PD-CN cases for the external validation cohort. The multi-atlas method parcellated the quantitative susceptibility mapping (QSM) images into 20 regions of interest and extracted QSM-based magnetic susceptibility values. Random forest, extreme gradient boosting, and light gradient boosting were selected as machine learning algorithms. Results: All classifiers demonstrated substantial performances in the classification task, particularly the random forest model. The accuracy, sensitivity, specificity, and area under the receiver operating characteristic curve for this model were 79.1%, 77.3%, 81.0%, and 0.78, respectively. The QSM values in the caudate nucleus, which were important features, were inversely correlated with the Montreal Cognitive Assessment scores (right caudate nucleus: r = -0.573, 95% CI: -0.801 to -0.298, p = 0.003; left caudate nucleus: r = -0.659, 95% CI: -0.894 to -0.392, p < 0.001). Conclusions: Machine learning models trained with QSM values successfully classified PD without dementia into PD-MCI and PD-CN groups, suggesting the potential of QSM values as an auxiliary biomarker for early evaluation of cognitive decline in patients with PD.