/Contrasformer

[CIKM 2024] Contrasformer: A Brain Network Contrastive Transformer for Neurodegenerative Condition Identification

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Contrasformer

This is the official PyTorch implementation of ContrastPool from the paper "Contrasformer: A Brain Network Contrastive Transformer for Neurodegenerative Condition Identification" published in 33rd ACM International Conference on Information and Knowledge Management (CIKM'24).

Model

Data

All Preprocessed data used in this paper are published in this paper. Data splits and configurations are stored in ./data/ and ./configs/. If you want to process your own data, please check the dataloader script ./data/BrainNet.py.

Usage

Please check experiment.sh on how to run the project.

Contact

If you have any questions, please feel free to reach out at jiaxing003@e.ntu.edu.sg.