Comparative Evaluation of Machine Learning Strategies for Analyzing Big Data in Psychiatry

Int J Mol Sci. 2018 Oct 29;19(11):3387. doi: 10.3390/ijms19113387.

Abstract

The requirement of innovative big data analytics has become a critical success factor for research in biological psychiatry. Integrative analyses across distributed data resources are considered essential for untangling the biological complexity of mental illnesses. However, little is known about algorithm properties for such integrative machine learning. Here, we performed a comparative analysis of eight machine learning algorithms for identification of reproducible biological fingerprints across data sources, using five transcriptome-wide expression datasets of schizophrenia patients and controls as a use case. We found that multi-task learning (MTL) with network structure (MTL_NET) showed superior accuracy compared to other MTL formulations as well as single task learning, and tied performance with support vector machines (SVM). Compared to SVM, MTL_NET showed significant benefits regarding the variability of accuracy estimates, as well as its robustness to cross-dataset and sampling variability. These results support the utility of this algorithm as a flexible tool for integrative machine learning in psychiatry.

Keywords: biomarker discovery; machine learning; multi-task learning; psychiatry.

Publication types

  • Comparative Study

MeSH terms

  • Algorithms
  • Big Data*
  • Data Mining / methods*
  • Humans
  • Machine Learning*
  • Schizophrenia / genetics*
  • Transcriptome*