/Geometric-attention-neural-network

Learning brain dynamics of evolving manifold functional MRI data using geometric-attention neural network

Primary LanguagePython

Learning Brain Dynamics of Evolving Manifold Functional MRI Data Using Geometric-Attention Neural Network

This repo contains the code of our TMI 2022 paper Learning Brain Dynamics of Evolving Manifold Functional MRI Data Using Geometric-Attention Neural Network.

Usage

  1. create an environment
conda env create -f requirements.yml
  1. activate the environment
conda activate TMI-2022
  1. run train.py
python train.py

Reference

If you find our work useful in your research, please consider citing:

@ARTICLE{9761822,
    author={Dan, Tingting and Huang, Zhuobin and Cai, Hongmin and Laurienti, Paul J. and Wu, Guorong},
    journal={IEEE Transactions on Medical Imaging}, 
    title={Learning Brain Dynamics of Evolving Manifold Functional MRI Data Using Geometric-Attention Neural Network}, 
    year={2022},
    volume={41},
    number={10},
    pages={2752-2763},
    doi={10.1109/TMI.2022.3169640}
}