Contrastive Graph Pooling for Explainable Classification of Brain Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xu, Jiaxing, Bian, Qingtian, Li, Xinhang, Zhang, Aihu, Ke, Yiping, Qiao, Miao, Zhang, Wei, Sim, Wei Khang Jeremy, Gulyás, Balázs
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916383986548736
author Xu, Jiaxing
Bian, Qingtian
Li, Xinhang
Zhang, Aihu
Ke, Yiping
Qiao, Miao
Zhang, Wei
Sim, Wei Khang Jeremy
Gulyás, Balázs
author_facet Xu, Jiaxing
Bian, Qingtian
Li, Xinhang
Zhang, Aihu
Ke, Yiping
Qiao, Miao
Zhang, Wei
Sim, Wei Khang Jeremy
Gulyás, Balázs
contents Functional magnetic resonance imaging (fMRI) is a commonly used technique to measure neural activation. Its application has been particularly important in identifying underlying neurodegenerative conditions such as Parkinson's, Alzheimer's, and Autism. Recent analysis of fMRI data models the brain as a graph and extracts features by graph neural networks (GNNs). However, the unique characteristics of fMRI data require a special design of GNN. Tailoring GNN to generate effective and domain-explainable features remains challenging. In this paper, we propose a contrastive dual-attention block and a differentiable graph pooling method called ContrastPool to better utilize GNN for brain networks, meeting fMRI-specific requirements. We apply our method to 5 resting-state fMRI brain network datasets of 3 diseases and demonstrate its superiority over state-of-the-art baselines. Our case study confirms that the patterns extracted by our method match the domain knowledge in neuroscience literature, and disclose direct and interesting insights. Our contributions underscore the potential of ContrastPool for advancing the understanding of brain networks and neurodegenerative conditions. The source code is available at https://github.com/AngusMonroe/ContrastPool.
format Preprint
id arxiv_https___arxiv_org_abs_2307_11133
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Contrastive Graph Pooling for Explainable Classification of Brain Networks
Xu, Jiaxing
Bian, Qingtian
Li, Xinhang
Zhang, Aihu
Ke, Yiping
Qiao, Miao
Zhang, Wei
Sim, Wei Khang Jeremy
Gulyás, Balázs
Neurons and Cognition
Artificial Intelligence
Machine Learning
Functional magnetic resonance imaging (fMRI) is a commonly used technique to measure neural activation. Its application has been particularly important in identifying underlying neurodegenerative conditions such as Parkinson's, Alzheimer's, and Autism. Recent analysis of fMRI data models the brain as a graph and extracts features by graph neural networks (GNNs). However, the unique characteristics of fMRI data require a special design of GNN. Tailoring GNN to generate effective and domain-explainable features remains challenging. In this paper, we propose a contrastive dual-attention block and a differentiable graph pooling method called ContrastPool to better utilize GNN for brain networks, meeting fMRI-specific requirements. We apply our method to 5 resting-state fMRI brain network datasets of 3 diseases and demonstrate its superiority over state-of-the-art baselines. Our case study confirms that the patterns extracted by our method match the domain knowledge in neuroscience literature, and disclose direct and interesting insights. Our contributions underscore the potential of ContrastPool for advancing the understanding of brain networks and neurodegenerative conditions. The source code is available at https://github.com/AngusMonroe/ContrastPool.
title Contrastive Graph Pooling for Explainable Classification of Brain Networks
topic Neurons and Cognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2307.11133