Evaluation of machine learning algorithms performance for the prediction of early multiple sclerosis from resting-state FMRI connectivity data

Brain Imaging Behav. 2019 Aug;13(4):1103-1114. doi: 10.1007/s11682-018-9926-9.

Abstract

Machine Learning application on clinical data in order to support diagnosis and prognostic evaluation arouses growing interest in scientific community. However, choice of right algorithm to use was fundamental to perform reliable and robust classification. Our study aimed to explore if different kinds of Machine Learning technique could be effective to support early diagnosis of Multiple Sclerosis and which of them presented best performance in distinguishing Multiple Sclerosis patients from control subjects. We selected following algorithms: Random Forest, Support Vector Machine, Naïve-Bayes, K-nearest-neighbor and Artificial Neural Network. We applied the Independent Component Analysis to resting-state functional-MRI sequence to identify brain networks. We found 15 networks, from which we extracted the mean signals used into classification. We performed feature selection tasks in all algorithms to obtain the most important variables. We showed that best discriminant network between controls and early Multiple Sclerosis, was the sensori-motor I, according to early manifestation of motor/sensorial deficits in Multiple Sclerosis. Moreover, in classification performance, Random Forest and Support Vector Machine showed same 5-fold cross-validation accuracies (85.7%) using only this network, resulting to be best approaches. We believe that these findings could represent encouraging step toward the translation to clinical diagnosis and prognosis.

Keywords: Artificial neural network; K-nearest-neighbor; Naïve Bayes; Random Forest; Resting state fMRI; Support vector machine.

MeSH terms

  • Adult
  • Algorithms
  • Bayes Theorem
  • Brain
  • Cognition
  • Connectome / methods*
  • Female
  • Forecasting / methods*
  • Humans
  • Machine Learning
  • Magnetic Resonance Imaging / methods
  • Male
  • Middle Aged
  • Multiple Sclerosis / diagnostic imaging*
  • Rest
  • Support Vector Machine