Neural Architecture Search for Optimization of Spatial-Temporal Brain Network Decomposition

Qing Li, Wei Zhang, Jinglei Lv, Xia Wu, Tianming Liu

Research output: Chapter in Book/Report/Conference proceedingConference contribution

4 Scopus citations

Abstract

Using neural networks to explore spatial patterns and temporal dynamics of human brain activities has been an important yet challenging problem because it is hard to manually design the most optimal neural networks. There have been several promising deep learning methods that can decompose neuroscientifically meaningful spatial-temporal patterns from 4D fMRI data, e.g., the deep sparse recurrent auto-encoder (DSRAE). However, those previous studies still depend on hand-crafted neural network structures and hyperparameters, which are not optimal in various senses. In this paper, we employ evolutionary algorithms to optimize such DSRAE neural networks by minimizing the expected loss of the generated architectures on data reconstruction via the neural architecture search (NAS) framework, named NAS-DSRAE. The optimized NAS-DSRAE is evaluated by the publicly available human connectome project (HCP) fMRI datasets and our promising results showed that NAS-DSRAE has sufficient generalizability to model the spatial-temporal features and is better than the hand-crafted model. To our best knowledge, the proposed NAS-DSRAE is among the earliest NAS models that can extract connectome-scale meaningful spatial-temporal brain networks from 4D fMRI data.

Original languageEnglish (US)
Title of host publicationMedical Image Computing and Computer Assisted Intervention – MICCAI 2020 - 23rd International Conference, Proceedings
EditorsAnne L. Martel, Purang Abolmaesumi, Danail Stoyanov, Diana Mateus, Maria A. Zuluaga, S. Kevin Zhou, Daniel Racoceanu, Leo Joskowicz
PublisherSpringer Science and Business Media Deutschland GmbH
Pages377-386
Number of pages10
ISBN (Print)9783030597276
DOIs
StatePublished - 2020
Externally publishedYes
Event23rd International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2020 - Lima, Peru
Duration: Oct 4 2020Oct 8 2020

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12267 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference23rd International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2020
Country/TerritoryPeru
CityLima
Period10/4/2010/8/20

Keywords

  • Auto-encoder
  • Neural architecture search
  • Recurrent neural network

ASJC Scopus subject areas

  • Theoretical Computer Science
  • General Computer Science

Fingerprint

Dive into the research topics of 'Neural Architecture Search for Optimization of Spatial-Temporal Brain Network Decomposition'. Together they form a unique fingerprint.

Cite this