Artificial intelligence to predict clinical disability in patients with multiple sclerosis using FLAIR MRI

Abstract

Purpose: The purpose of this study was to create an algorithm that combines multiple machine-learning techniques to predict the expanded disability status scale (EDSS) score of patients with multiple sclerosis at two years solely based on age, sex and fluid attenuated inversion recovery (FLAIR) MRI data.

Materials and methods: Our algorithm combined several complementary predictors: a pure deep learning predictor based on a convolutional neural network (CNN) that learns from the images, as well as classical machine-learning predictors based on random forest regressors and manifold learning trained using the location of lesion load with respect to white matter tracts. The aggregation of the predictors was done through a weighted average taking into account prediction errors for different EDSS ranges. The training dataset consisted of 971 multiple sclerosis patients from the "Observatoire français de la sclérose en plaques" (OFSEP) cohort with initial FLAIR MRI and corresponding EDSS score at two years. A test dataset (475 subjects) was provided without an EDSS score. Ten percent of the training dataset was used for validation.

Results: Our algorithm predicted EDSS score in patients with multiple sclerosis and achieved a MSE=2.2 with the validation dataset and a MSE=3 (mean EDSS error=1.7) with the test dataset.

Conclusion: Our method predicts two-year clinical disability in patients with multiple sclerosis with a mean EDSS score error of 1.7, using FLAIR sequence and basic patient demographics. This supports the use of our model to predict EDSS score progression. These promising results should be further validated on an external validation cohort.

Keywords: Artificial intelligence; Disability prediction; Machine learning; Magnetic resonance imaging (MRI); Multiple sclerosis.