This is the official PyTorch implementation of ContrastPool from paper "Union Subgraph Neural Networks" accepted by the 38th Annual AAAI Conference on Artificial Intelligence (AAAI‘24).
Link: AAAI PDF.
Please check main.py
on how to run the project.
If you find this code useful, please consider citing our paper:
@inproceedings{xu2024union,
title={Union Subgraph Neural Networks},
author={Xu, Jiaxing and Zhang, Aihu and Bian, Qingtian and Dwivedi, Vijay Prakash and Ke, Yiping},
booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
volume={38},
number={14},
pages={16173--16183},
year={2024}
}
If you have any questions, please feel free to reach out at jiaxing003@e.ntu.edu.sg
.