Graph U-Nets
proceedings.mlr.press/v97/gao19a/gao19a.pdfpooling, global pooling layers (Zhao et al.,2015a) perform down-sampling operations on all input units, thereby re-ducing each feature map to a single number. In contrast, k-max pooling layers (Kalchbrenner et al.,2014) select the k-largest units out of each feature map. However, we cannot directly apply these pooling operations to graphs.
graphneural.network - Spektral
graphneural.networkTop-K pooling; Self-Attention Graph (SAG) pooling; Global pooling; Global gated attention pooling; SortPool; Spektral also includes lots of utilities for representing, manipulating, and transforming graphs in your graph deep learning projects. See how to get started with Spektral and have a look at the examples for some templates.
torch-geometric · PyPI
pypi.org › project › torch-geometricOct 26, 2021 · Top-K Pooling from Gao and Ji: Graph U-Nets (ICML 2019), Cangea et al.: Towards Sparse Hierarchical Graph Classifiers (NeurIPS-W 2018) and Knyazev et al.: Understanding Attention and Generalization in Graph Neural Networks (ICLR-W 2019)