The Contribution of Explainable Machine Learning Algorithms Using ROI-based Brain Surface Morphology Parameters in Distinguishing Early-onset Schizophrenia From Bipolar Disorder.

ABSTRACT

RATIONALE AND OBJECTIVES: To differentiate early-onset schizophrenia (EOS) from early-onset bipolar disorder (EBD) using surface-based morphometry measurements and brain volumes using machine learning (ML) algorithms. METHOD: High-resolution T1-weighted images were obtained to measure cortical thickness (CT), gyrification, gyrification index (GI), sulcal depth (SD), fractal dimension (FD), and brain volumes. After the feature selection step, ML classifiers were applied for each feature set and the combination of them. The SHapley Additive exPlanations (SHAP) technique was implemented to interpret the contribution of each feature. FINDINGS: 144 adolescents (16.2 ± 1.4 years, female=39%) with EOS (n = 81) and EBD (n = 63) were included. The Adaptive Boosting (AdaBoost) algorithm had the highest accuracy (82.75%) in the whole dataset that includes all variables from Destrieux atlas. The best-performing algorithms were K-nearest neighbors (KNN) for FD subset, support vector machine (SVM) for SD subset, and AdaBoost for GI subset. The KNN algorithm had the highest accuracy (accuracy=79.31%) in the whole dataset from the Desikan-Killiany-Tourville atlas. CONCLUSION: This study demonstrates the use of ML in the differential diagnosis of EOS and EBD using surface-based morphometry measurements. Future studies could focus on multicenter data for the validation of these results.