arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2203.10093cs.LGcs.AIcs.NEq-bio.NC
收藏

用于脑网络分析的深度强化学习引导的图神经网络

Deep reinforcement learning guided graph neural networks for brain network analysis

Institute of Information Engineering, Chinese Academy of Sciences(中国科学院信息工程研究所);University of Chinese Academy of Sciences(中国科学院大学);Macquarie University(麦考瑞大学);Beihang University(北京航空航天大学);National Acoustic Laboratories(国家声学实验室);CAEIT(中国电子科学研究院)

Xusheng Zhao, Jia Wu, Hao Peng, Amin Beheshti, Jessica J. M. Monaghan, David McAlpine, Heivet Hernandez-Perez, Mark Dras, Qiong Dai, Yangyang Li, Philip S. Yu, Lifang He

摘要:

现代神经成像技术,如扩散张量成像(DTI)和功能性磁共振成像(fMRI),使我们能够将人脑建模为大脑网络或连接体。捕捉大脑网络的结构信息和层次模式对于理解大脑功能和疾病状态至关重要。近年来,图神经网络(GNNs)具有良好的网络表示学习能力,促使人们提出了许多基于GNNs的脑网络分析方法。具体而言,这些方法应用特征聚合和全局池化来将脑网络实例转换为用于下游脑网络分析任务的有意义的低维表示。然而,现有的基于GNN的方法往往忽略了不同主体的大脑网络可能需要各种聚合迭代,并使用具有固定层数的GNN来学习所有大脑网络。因此,如何充分释放GNN的潜力,促进大脑网络分析仍然是一个不平凡的问题。为了解决这个问题,我们提出了一种新的大脑网络表示框架,即BN-GNN,它为每个大脑网络搜索最佳的GNN架构。具体来说,BN-GNN采用深度强化学习(DRL)来训练元策略,以自动确定给定大脑网络所需的最佳特征聚合数量(反映在GNN层的数量中)。在八个真实脑网络数据集上进行的大量实验表明,我们提出的BN-GNN提高了传统GNN在不同脑网络分析任务上的性能。

英文摘要:

Modern neuroimaging techniques, such as diffusion tensor imaging (DTI) and functional magnetic resonance imaging (fMRI), enable us to model the human brain as a brain network or connectome. Capturing brain networks' structural information and hierarchical patterns is essential for understanding brain functions and disease states. Recently, the promising network representation learning capability of graph neural networks (GNNs) has prompted many GNN-based methods for brain network analysis to be proposed. Specifically, these methods apply feature aggregation and global pooling to convert brain network instances into meaningful low-dimensional representations used for downstream brain network analysis tasks. However, existing GNN-based methods often neglect that brain networks of different subjects may require various aggregation iterations and use GNN with a fixed number of layers to learn all brain networks. Therefore, how to fully release the potential of GNNs to promote brain network analysis is still non-trivial. To solve this problem, we propose a novel brain network representation framework, namely BN-GNN, which searches for the optimal GNN architecture for each brain network. Concretely, BN-GNN employs deep reinforcement learning (DRL) to train a meta-policy to automatically determine the optimal number of feature aggregations (reflected in the number of GNN layers) required for a given brain network. Extensive experiments on eight real-world brain network datasets demonstrate that our proposed BN-GNN improves the performance of traditional GNNs on different brain network analysis tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑