GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → egyptdj → stagin

egyptdj / stagin

Licence: GPL-3.0 license
STAGIN: Spatio-Temporal Attention Graph Isomorphism Network

Programming Languages

python
139335 projects - #7 most used programming language

Labels

attentionneuroimagingfmriconnectomeexplainabilityneuripsgnngraph-neural-networkstaginneurips2021

Projects that are alternatives of or similar to stagin

brainGraph
Graph theory analysis of brain MRI data
Stars: ✭ 136 (+300%)
Mutual labels:  neuroimaging, fmri, connectome
spatio-temporal-brain
A Deep Graph Neural Network Architecture for Modelling Spatio-temporal Dynamics in rs-fMRI Data
Stars: ✭ 22 (-35.29%)
Mutual labels:  fmri, explainability, gnn
GNNs-in-Network-Neuroscience
A review of papers proposing novel GNN methods with application to brain connectivity published in 2017-2020.
Stars: ✭ 92 (+170.59%)
Mutual labels:  gnn, graph-neural-network
PDN
The official PyTorch implementation of "Pathfinder Discovery Networks for Neural Message Passing" (WebConf '21)
Stars: ✭ 44 (+29.41%)
Mutual labels:  gnn, graph-neural-network
gnn-lspe
Source code for GNN-LSPE (Graph Neural Networks with Learnable Structural and Positional Representations), ICLR 2022
Stars: ✭ 165 (+385.29%)
Mutual labels:  attention, gnn
ReFine
Official code of "Towards Multi-Grained Explainability for Graph Neural Networks" (2021 NeurIPS)
Stars: ✭ 40 (+17.65%)
Mutual labels:  graph-neural-network, neurips2021
fmriflows
fmriflows is a consortium of many (dependent) fMRI analysis pipelines, including anatomical and functional pre-processing, univariate 1st and 2nd-level analysis, as well as multivariate pattern analysis.
Stars: ✭ 40 (+17.65%)
Mutual labels:  neuroimaging, fmri
visualqc
VisualQC : assistive tool to ease the quality control workflow of neuroimaging data.
Stars: ✭ 56 (+64.71%)
Mutual labels:  neuroimaging, fmri
mtad-gat-pytorch
PyTorch implementation of MTAD-GAT (Multivariate Time-Series Anomaly Detection via Graph Attention Networks) by Zhao et. al (2020, https://arxiv.org/abs/2009.02040).
Stars: ✭ 85 (+150%)
Mutual labels:  attention, gnn
Revisiting-Contrastive-SSL
Revisiting Contrastive Methods for Unsupervised Learning of Visual Representations. [NeurIPS 2021]
Stars: ✭ 81 (+138.24%)
Mutual labels:  neurips, neurips2021
Awesome-Federated-Learning-on-Graph-and-GNN-papers
Federated learning on graph, especially on graph neural networks (GNNs), knowledge graph, and private GNN.
Stars: ✭ 206 (+505.88%)
Mutual labels:  gnn, graph-neural-network
connectomemapper3
Connectome Mapper 3 is a BIDS App that implements full anatomical, diffusion, resting/state functional MRI, and recently EEG processing pipelines, from raw T1 / DWI / BOLD , and preprocessed EEG data to multi-resolution brain parcellation with corresponding connection matrices.
Stars: ✭ 45 (+32.35%)
Mutual labels:  fmri, connectome
classifier multi label seq2seq attention
multi-label，classifier，text classification，多标签文本分类，文本分类，BERT，ALBERT，multi-label-classification，seq2seq，attention，beam search
Stars: ✭ 26 (-23.53%)
Mutual labels:  attention
gqa-node-properties
Recalling node properties from a knowledge graph
Stars: ✭ 19 (-44.12%)
Mutual labels:  attention
clinica
Software platform for clinical neuroimaging studies
Stars: ✭ 153 (+350%)
Mutual labels:  neuroimaging
gnn-re-ranking
A real-time GNN-based method. Understanding Image Retrieval Re-Ranking: A Graph Neural Network Perspective
Stars: ✭ 64 (+88.24%)
Mutual labels:  gnn
transformer
A PyTorch Implementation of "Attention Is All You Need"
Stars: ✭ 28 (-17.65%)
Mutual labels:  attention
Social-Knowledge-Graph-Papers
A paper list of research about social knowledge graph
Stars: ✭ 27 (-20.59%)
Mutual labels:  graph-neural-network
Linear-Attention-Mechanism
Attention mechanism
Stars: ✭ 27 (-20.59%)
Mutual labels:  attention
Hierarchical-Word-Sense-Disambiguation-using-WordNet-Senses
Word Sense Disambiguation using Word Specific models, All word models and Hierarchical models in Tensorflow
Stars: ✭ 33 (-2.94%)
Mutual labels:  attention
View All Similar Projects ➔

STAGIN

Spatio-Temporal Attention Graph Isomorphism Network

Paper

Learning Dynamic Graph Representation of Brain Connectome with Spatio-Temporal Attention
Byung-Hoon Kim, Jong Chul Ye, Jae-Jin Kim
presented at NeurIPS 2021
arXiv, OpenReview, proceeding

Concept

Schematic illustration of STAGIN

Dataset

The fMRI data used for the experiments of the paper should be downloaded from the Human Connectome Project.

Example structure of the directory tree
data (specified by option --sourcedir)
├─── behavioral
│    ├─── hcp.csv
│    ├─── hcp_taskrest_EMOTION.csv
│    ├─── hcp_taskrest_GAMBLING.csv
│    ├─── ...
│    └─── hcp_taskrest_WM.csv
├─── img
│    ├─── REST
│    │    ├─── 123456.nii.gz
│    │    ├─── 234567.nii.gz
│    │    ├─── ...
│    │    └─── 999999.nii.gz
│    └─── TASK
│         ├─── EMOTION
│         │    ├─── 123456.nii.gz
│         │    ├─── 234567.nii.gz
│         │    ├─── ...
│         │    └─── 999999.nii.gz
│         ├─── GAMBLING
│         │    ├─── ...
│         │    └─── 999999.nii.gz
│         ├─── ...
│         └─── WM
│              ├─── ...
│              └─── 999999.nii.gz
└───roi
     └─── 7_400_coord.csv
Example content of the csv files
data/behavioral/hcp.csv
| Subject | Gender |
|---------|--------|
| 123456  |   F    |
| 234567  |   M    |
| ......  | ...... |
| 999999  |   F    |
data/behavioral/hcp_taskrest_WM.csv
| Task | Rest |
|------|------|
|  0   |  1   |
|  0   |  1   |
| ...  | ...  |
|  1   |  0   |
data/roi/7_400_coord.csv
| ROI Index | Label Name                 | R | A | S |
|-----------|----------------------------|---|---|---|
|         0 | NONE                       | NA| NA| NA|
|         1 | 7Networks_LH_Vis_1         |-32|-42|-20|
|         2 | 7Networks_LH_Vis_2         |-30|-32|-18|
|       ... | .........                  | . | . | . |
|       400 | 7Networks_RH_Default_PCC_9 | 8 |-50| 44|

Commands

Run the main script to perform experiments

python main.py

Command-line options can be listed with -h flag.

python main.py -h

Requirements

  • python 3.8.5
  • numpy == 1.20.2
  • torch == 1.7.0
  • torchvision == 0.8.1
  • einops == 0.3.0
  • sklearn == 0.24.2
  • nilearn == 0.7.1
  • nipy == 0.5.0
  • pingouin == 0.3.11
  • tensorboard == 2.5.0
  • tqdm == 4.60.0

For brainplot:

  • MRIcroGL >= 1.2
  • opencv-python == 4.5.2

Updates

  • 2022-04-29 5c262d8d: Top k-percentile values from the adjacency matrix is now calculated without the need for calling .detach().cpu().numpy() which improves computation speed.

Contact

[email protected]

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].