可微分神经架构搜索用于最优空间/时间大脑功能网络分解。

Differentiable neural architecture search for optimal spatial/temporal brain function network decomposition.

机构信息

School of Artificial Intelligence, Beijing Normal University, Beijing, China; Engineering Research Center of Intelligent Technology and Educational Application, Ministry of Education, Beijing, China.

School of Artificial Intelligence, Beijing Normal University, Beijing, China; Engineering Research Center of Intelligent Technology and Educational Application, Ministry of Education, Beijing, China.

出版信息

Med Image Anal. 2021 Apr;69:101974. doi: 10.1016/j.media.2021.101974. Epub 2021 Jan 20.

Abstract

It has been a key topic to decompose the brain's spatial/temporal function networks from 4D functional magnetic resonance imaging (fMRI) data. With the advantages of robust and meaningful brain pattern extraction, deep neural networks have been shown to be more powerful and flexible in fMRI data modeling than other traditional methods. However, the challenge of designing neural network architecture for high-dimensional and complex fMRI data has also been realized recently. In this paper, we propose a new spatial/temporal differentiable neural architecture search algorithm (ST-DARTS) for optimal brain network decomposition. The core idea of ST-DARTS is to optimize the inner cell structure of the vanilla recurrent neural network (RNN) in order to effectively decompose spatial/temporal brain function networks from fMRI data. Based on the evaluations on all seven fMRI tasks in human connectome project (HCP) dataset, the ST-DARTS model is shown to perform promisingly, both spatially (i.e., it can recognize the most stimuli-correlated spatial brain network activation that is very similar to the benchmark) and temporally (i.e., its temporal activity is highly positively correlated with the task-design). To further improve the efficiency of ST-DARTS model, we introduce a flexible early-stopping mechanism, named as ST-DARTS+, which further improves experimental results significantly. To our best knowledge, the proposed ST-DARTS and ST-DARTS+ models are among the early efforts in optimally decomposing spatial/temporal brain function networks from fMRI data with neural architecture search strategy and they demonstrate great promise.

摘要

从 4D 功能磁共振成像 (fMRI) 数据中分解大脑的空间/时间功能网络一直是一个关键课题。由于具有强大且有意义的大脑模式提取优势,深度神经网络在 fMRI 数据建模方面比其他传统方法更强大、更灵活。然而,最近也意识到了为高维复杂 fMRI 数据设计神经网络架构的挑战。在本文中,我们提出了一种新的时空可微分神经架构搜索算法(ST-DARTS),用于优化大脑网络分解。ST-DARTS 的核心思想是优化香草递归神经网络(RNN)的内部单元结构,以便从 fMRI 数据中有效地分解空间/时间大脑功能网络。基于人类连接组计划(HCP)数据集上的所有七个 fMRI 任务的评估,ST-DARTS 模型表现出色,无论是在空间上(即,它可以识别与基准非常相似的最受刺激相关的空间大脑网络激活)还是在时间上(即,其时间活动与任务设计高度正相关)。为了进一步提高 ST-DARTS 模型的效率,我们引入了一种灵活的提前停止机制,称为 ST-DARTS+,它进一步显著提高了实验结果。据我们所知,所提出的 ST-DARTS 和 ST-DARTS+模型是最早使用神经架构搜索策略从 fMRI 数据中最佳分解空间/时间大脑功能网络的努力之一,它们展示了巨大的潜力。

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索