This is the official code base for the paper: Homomorphism Counts for Graph Neural Networks: All About That Basis.
This repository is divided into four sub-projects:
- The subdirectory
brec
is a partial clone of https://github.com/GraphPKU/BREC/tree/966b5ed5c27bf6372ea081f90ef8c8b2a3342ebc by Wang & Zhang (2023). This project can be used to reproduce the BREC experiements from Section 5.4 of the paper. - The subdirectory
hombasis-bench
is a PyTorch re-implementation of some of the models that were benchmarked by Dwivedi et al. (2023) and Hu et al. (2020). This project can be used to reproduce the ZINC and COLLAB experiments from Sections 5.1 and 5.3 of the paper, respectively. - The subdirectory
pact
provides the code for all homomorphism pattern counting that was used in this paper. - The subdirectory
qm9
is a clone of https://github.com/radoslav11/SP-MPNN/tree/main by Abboud et al. (2022). This project can be used to reproduce the QM9 experiments from Section 5.2 of the paper.
For installation and dependency information for each sub-project, please refer to the README files in the respective directories.