AI-Driven 3D MRI Image Analysis Enabling Precision in Multiple Sclerosis Diagnostics
Author Block: F. Orzan1, L. Dioșan2, Z. Bálint1; 1Cluj Napoca/RO, 2Cluj-Napoca/RO
Purpose: We aimed to build an AI-based decision support system for automatic segmentation and characterization of MS lesions from 3D MRI, with the long-term goal of personalized treatment. This study focuses on preprocessing and segmentation, providing the basis for future characterization and clinical integration.
Methods or Background: Our study evaluated a 3D U-Net segmentation model with four encoding-decoding levels, integrating attention gates and skip connections to improve lesion localization. The architecture employs BatchNorm3d, ReLU activations, and Dropout for regularization. Evaluation was conducted on 40 cases from the MICCAI 2021 dataset. Preprocessing included intensity normalization, isotropic resampling, N4 bias field correction, skull stripping, and rigid registration. Training used Adam optimization with BCE loss over 50 epochs, with early stopping, 5-fold cross-validation, and an 80/20 train-validation split.
Results or Findings: This study focuses exclusively on image preprocessing, model development, and lesion segmentation. Textural analysis, lesion classification, and the development of a user interface for fine-tuning will be addressed in future work.
The model showed a steady decrease in both training and validation loss, with stabilization after epoch 40. The close alignment between losses suggests minimal overfitting and good generalization - an essential property for clinical applications. The Dice score increased consistently, reaching 0.78 on training and 0.72 on validation data, confirming the model’s ability to accurately segment MS lesions across diverse samples.
Conclusion: Our U-Net model with attention gates and skip connections shows promising performance for automated MS detection on 3D MRI. Future work will add textural analysis, lesion classification, user interface development, and validation on multi-vendor datasets for robustness and scalability.
Limitations: A key limitation of MS detection algorithms, including our study, is the scarcity of large imaging datasets, which limits performance relative to human experts.
Funding for this study: Not applicable
Has your study been approved by an ethics committee? Not applicable
Ethics committee - additional information: