Skip to main navigation Skip to search Skip to main content

Toward a unified framework for interpreting machine-learning models in neuroimaging

  • Lada Kohoutová
  • , Juyeon Heo
  • , Sungmin Cha
  • , Sungwoo Lee
  • , Taesup Moon
  • , Tor D. Wager
  • , Choong Wan Woo
  • Institute for Basic Science
  • Sungkyunkwan University
  • Dartmouth College
  • University of Colorado Boulder

Research output: Contribution to journalArticlepeer-review

Abstract

Machine learning is a powerful tool for creating computational models relating brain function to behavior, and its use is becoming widespread in neuroscience. However, these models are complex and often hard to interpret, making it difficult to evaluate their neuroscientific validity and contribution to understanding the brain. For neuroimaging-based machine-learning models to be interpretable, they should (i) be comprehensible to humans, (ii) provide useful information about what mental or behavioral constructs are represented in particular brain pathways or regions, and (iii) demonstrate that they are based on relevant neurobiological signal, not artifacts or confounds. In this protocol, we introduce a unified framework that consists of model-, feature- and biology-level assessments to provide complementary results that support the understanding of how and why a model works. Although the framework can be applied to different types of models and data, this protocol provides practical tools and examples of selected analysis methods for a functional MRI dataset and multivariate pattern-based predictive models. A user of the protocol should be familiar with basic programming in MATLAB or Python. This protocol will help build more interpretable neuroimaging-based machine-learning models, contributing to the cumulative understanding of brain mechanisms and brain health. Although the analyses provided here constitute a limited set of tests and take a few hours to days to complete, depending on the size of data and available computational resources, we envision the process of annotating and interpreting models as an open-ended process, involving collaborative efforts across multiple studies and laboratories.

Original languageEnglish
Pages (from-to)1399-1435
Number of pages37
JournalNature Protocols
Volume15
Issue number4
DOIs
StatePublished - 1 Apr 2020

Fingerprint

Dive into the research topics of 'Toward a unified framework for interpreting machine-learning models in neuroimaging'. Together they form a unique fingerprint.

Cite this