https://github.com/alexduvalinho/hoscpool
Higher-order clustering and pooling for GNNs
https://github.com/alexduvalinho/hoscpool
Last synced: about 2 months ago
JSON representation
Higher-order clustering and pooling for GNNs
- Host: GitHub
- URL: https://github.com/alexduvalinho/hoscpool
- Owner: AlexDuvalinho
- License: mit
- Created: 2022-08-19T09:01:23.000Z (almost 3 years ago)
- Default Branch: main
- Last Pushed: 2023-02-21T14:47:24.000Z (over 2 years ago)
- Last Synced: 2025-03-26T01:01:48.244Z (2 months ago)
- Language: Python
- Size: 7.81 KB
- Stars: 8
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# HoscPool
Higher-order clustering and pooling for GNNs
Paper accepted at the 31st ACM International Conference on Information and Knowledge Management (CIKM, 2022).
https://arxiv.org/abs/2209.03473
In this repo, we provide:
(1) the code for the pooling operator itself in [hoscpool.py](/hoscpool.py). It can be used similarly to MinCutPool or DiffPool and will soon be available in Pytorch Geometric.
(2) a short [example](/example.py) of how to use hoscpool for an arbitrary graph classification task.
### Set up
If needed, install the required packages: torch and torch-geometric.
### Citation
Please cite the original paper if you are using HoscPool in your work.
```
@inproceedings{duval2022higher,
title={Higher-order clustering and pooling for graph neural networks},
author={Duval, Alexandre and Malliaros, Fragkiskos},
booktitle={Proceedings of the 31st ACM International Conference on Information \& Knowledge Management},
pages={426--435},
year={2022}
}
```