Multiparametric MRI-based machine learning system of molecular subgroups and prognosis in medulloblastoma
摘要
We aimed to use artificial intelligence to accurately identify molecular subgroups of medulloblastoma (MB), predict clinical outcomes, and incorporate deep learning-based imaging features into the risk stratification.
MethodsThe MRI features were extracted for molecular subgroups by a novel multi-parameter convolutional neural network (CNN) called Bi-ResNet-MB. Then, MR features were used to establish a prognosis model based on XGBoost. Finally, a novel risk stratification system to stratify the patients based on the M2R Score (Machine learning-based Medulloblastoma Risk Score) was proposed.
ResultsA total of 139 MB patients (36 female, average age 7.27 ± 3.62 years) were treated at Beijing Tiantan Hospital. The Bi-ResNet-MB model excelled in molecular subgroup classification, achieving an average AUC of 0.946 (95% CI: 0.899–0.993). For prognostic prediction, our models achieved AUCs of 0.840 (95% CI: 0.792–0.888), 0.949 (95% CI: 0.899–0.999), and 0.960 (95% CI: 0.915–1.000) for OS, and 0.946 (95% CI: 0.905–0.987), 0.932 (95% CI: 0.875–0.989), and 0.964 (95% CI: 0.921–1.000) for PFS at 1, 3, and 5 years. In an independent validation dataset of 108 patients (33 female, average age 7.11 ± 2.92 years), the average AUC of molecular subgroup classification reached 0.894 (95% CI: 0.797–1.000). For PFS prediction at 1, 3, and 5 years, the AUCs were 0.832 (95% CI: 0.724–0.920), 0.875 (95% CI: 0.781–0.967), and 0.907 (95% CI: 0.760–1.000), respectively.
ConclusionsBased on machine learning and MRI data, models for MB molecular subgroups and prognosis prediction and the novel risk stratification system may significantly benefit patients.
Key Points