Regression-based machine-learning approaches to predict task activation using resting-state fMRI

Alexander D. Cohen, Ziyi Chen, Oiwi Parker Jones, Chen Niu, Yang Wang

Research output: Contribution to journalArticle

1 Scopus citations

Abstract

Resting-state fMRI has shown the ability to predict task activation on an individual basis by using a general linear model (GLM) to map resting-state network features to activation z-scores. The question remains whether the relatively simplistic GLM is the best approach to accomplish this prediction. In this study, several regression-based machine-learning approaches were compared, including GLMs, feed-forward neural networks, and random forest bootstrap aggregation (bagging). Resting-state and task data from 350 Human Connectome Project subjects were analyzed. First, the effect of the number of training subjects on the prediction accuracy was evaluated. In addition, the prediction accuracy and Dice coefficient were compared across models. Prediction accuracy increased with the training number up to 200 subjects; however, an elbow in the prediction curve occurred around 30–40 training subjects. All models performed well with correlation matrices, which displayed correlation between actual and predicted task activation for all subjects, exhibiting a strong diagonal trend for all tasks. Overall, the neural network and random forest bagging techniques outperformed the GLM. These approaches, however, require additional computing power and processing time. These results show that, while the GLM performs well, resting-state fMRI prediction of task activation could benefit from more complex machine learning approaches.

Original languageEnglish (US)
JournalHuman Brain Mapping
DOIs
StateAccepted/In press - Jan 1 2019
Externally publishedYes

    Fingerprint

Keywords

  • fMRI
  • machine learning
  • neural networks
  • random-forest bootstrap aggregation
  • resting state

ASJC Scopus subject areas

  • Anatomy
  • Radiological and Ultrasound Technology
  • Radiology Nuclear Medicine and imaging
  • Neurology
  • Clinical Neurology

Cite this