A comparative study of deep learning for cortical lesion MRI segmentation with explainability analysis in multiple
Nataliia Molchanova1, Alessandro Cagol2, Mario Ocampo-Pineda3
1Faculty of Biology and Medicine, University of Lausanne (UNIL), Lausanne, Switzerland; Radiology Department, Lausanne University Hospital (CHUV), Lausanne, Switzerland; MedGIFT, Institute of Informatics, School of Management, HES-SO Valais-Wallis University of Applied Sciences and Arts Western Switzerland, Sierre, Switzerland; CIBM Center for Biomedical Imaging, Lausanne, Switzerland.
Abstract:
Cortical lesions (CLs) have emerged as valuable biomarkers in multiple sclerosis (MS), offering high diagnostic specificity and prognostic relevance. However, their routine clinical integration remains limited due to subtle magnetic resonance imaging (MRI) appearance, challenges in expert annotation, and a lack of standardized automated methods. We present a multi-centric comparative study of CL detection and segmentation in MRI. A total of 656 MRI scans, including clinical trial and research data from four institutions, were acquired at 3T and 7T using MP2RAGE and MPRAGE sequences with expert-consensus annotations. We rely on the self-configuring nnU-Net framework, designed for medical imaging segmentation, and propose adaptations tailored to the improved CL detection. We evaluated model generalization through out-of-distribution testing, demonstrating promising lesion detection capabilities with an F1-score of 0.64 and 0.5 in and out of the domain, respectively. We also analyze internal model features and model errors for a better understanding of AI decision-making. Our study examines how data variability, lesion ambiguity, and protocol differences impact model performance, offering future recommendations to address these barriers to clinical adoption. Furthermore, we designed and implemented a medical expert questionnaire for better assessment of clinical value of the model predictions. To reinforce the reproducibility, the implementation and models will be publicly accessible and ready to use at GitHub and Zenodo.
