Contrastive Graph Pooling for Explainable Classification of Brain Networks
- URL: http://arxiv.org/abs/2307.11133v3
- Date: Fri, 6 Sep 2024 06:58:02 GMT
- Title: Contrastive Graph Pooling for Explainable Classification of Brain Networks
- Authors: Jiaxing Xu, Qingtian Bian, Xinhang Li, Aihu Zhang, Yiping Ke, Miao Qiao, Wei Zhang, Wei Khang Jeremy Sim, Balázs Gulyás,
- Abstract summary: Functional magnetic resonance imaging (fMRI) is a commonly used technique to measure neural activation.
Recent analysis of fMRI data models the brain as a graph and extracts features by graph neural networks (GNNs)
We propose ContrastPool to better utilize GNN for brain networks, meeting fMRI-specific requirements.
- Score: 9.917580917431293
- License: http://creativecommons.org/licenses/by-nc-sa/4.0/
- Abstract: 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.
Related papers
- Graph Neural Networks for Brain Graph Learning: A Survey [53.74244221027981]
Graph neural networks (GNNs) have demonstrated a significant advantage in mining graph-structured data.
GNNs to learn brain graph representations for brain disorder analysis has recently gained increasing attention.
In this paper, we aim to bridge this gap by reviewing brain graph learning works that utilize GNNs.
arXiv Detail & Related papers (2024-06-01T02:47:39Z) - DBGDGM: Dynamic Brain Graph Deep Generative Model [63.23390833353625]
Graphs are a natural representation of brain activity derived from functional magnetic imaging (fMRI) data.
It is well known that clusters of anatomical brain regions, known as functional connectivity networks (FCNs), encode temporal relationships which can serve as useful biomarkers for understanding brain function and dysfunction.
Previous works, however, ignore the temporal dynamics of the brain and focus on static graphs.
We propose a dynamic brain graph deep generative model (DBGDGM) which simultaneously clusters brain regions into temporally evolving communities and learns dynamic unsupervised node embeddings.
arXiv Detail & Related papers (2023-01-26T20:45:30Z) - Learning Task-Aware Effective Brain Connectivity for fMRI Analysis with
Graph Neural Networks [28.460737693330245]
We propose TBDS, an end-to-end framework based on underlineTask-aware underlineBrain connectivity underlineDAG for fMRI analysis.
The key component of TBDS is the brain network generator which adopts a DAG learning approach to transform the raw time-series into task-aware brain connectivities.
Comprehensive experiments on two fMRI datasets demonstrate the efficacy of TBDS.
arXiv Detail & Related papers (2022-11-01T03:59:54Z) - DynDepNet: Learning Time-Varying Dependency Structures from fMRI Data
via Dynamic Graph Structure Learning [58.94034282469377]
We propose DynDepNet, a novel method for learning the optimal time-varying dependency structure of fMRI data induced by downstream prediction tasks.
Experiments on real-world fMRI datasets, for the task of sex classification, demonstrate that DynDepNet achieves state-of-the-art results.
arXiv Detail & Related papers (2022-09-27T16:32:11Z) - Hierarchical Graph Convolutional Network Built by Multiscale Atlases for
Brain Disorder Diagnosis Using Functional Connectivity [48.75665245214903]
We propose a novel framework to perform multiscale FCN analysis for brain disorder diagnosis.
We first use a set of well-defined multiscale atlases to compute multiscale FCNs.
Then, we utilize biologically meaningful brain hierarchical relationships among the regions in multiscale atlases to perform nodal pooling.
arXiv Detail & Related papers (2022-09-22T04:17:57Z) - FBNETGEN: Task-aware GNN-based fMRI Analysis via Functional Brain
Network Generation [11.434951542977515]
We develop FBNETGEN, a task-aware and interpretable fMRI analysis framework via deep brain network generation.
Along with the process, the key novel component is the graph generator which learns to transform raw time-series features into task-oriented brain networks.
Our learnable graphs also provide unique interpretations by highlighting prediction-related brain regions.
arXiv Detail & Related papers (2022-05-25T03:26:50Z) - Deep Reinforcement Learning Guided Graph Neural Networks for Brain
Network Analysis [61.53545734991802]
We propose a novel brain network representation framework, namely BN-GNN, which searches for the optimal GNN architecture for each brain network.
Our proposed BN-GNN improves the performance of traditional GNNs on different brain network analysis tasks.
arXiv Detail & Related papers (2022-03-18T07:05:27Z) - Aiding Medical Diagnosis Through the Application of Graph Neural
Networks to Functional MRI Scans [0.0]
Graph Neural Networks (GNNs) have been shown to be a powerful tool for generating predictions from biological data.
We present a novel approach to representing resting-state fMRI data as a graph containing nodes and edges without omitting any of the voxels.
We show that GNNs can successfully predict the disease and sex of a person.
arXiv Detail & Related papers (2021-12-01T14:10:52Z) - Pooling Regularized Graph Neural Network for fMRI Biomarker Analysis [29.489129970039873]
A promising approach to identify the salient regions is using Graph Neural Networks (GNNs)
We propose an interpretable GNN framework with a novel salient region selection mechanism to determine neurological brain biomarkers associated with disorders.
We apply the PR-GNN framework on a Biopoint Autism Spectral Disorder (ASD) fMRI dataset.
arXiv Detail & Related papers (2020-07-29T04:19:36Z) - Understanding Graph Isomorphism Network for rs-fMRI Functional
Connectivity Analysis [49.05541693243502]
We develop a framework for analyzing fMRI data using the Graph Isomorphism Network (GIN)
One of the important contributions of this paper is the observation that the GIN is a dual representation of convolutional neural network (CNN) in the graph space.
We exploit CNN-based saliency map techniques for the GNN, which we tailor to the proposed GIN with one-hot encoding.
arXiv Detail & Related papers (2020-01-10T23:40:09Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.