Paper ID: 2409.10263

Hierarchical Graph Pooling Based on Minimum Description Length

Jan von Pichowski, Christopher Blöcker, Ingo Scholtes

Graph pooling is an essential part of deep graph representation learning. We introduce MapEqPool, a principled pooling operator that takes the inherent hierarchical structure of real-world graphs into account. MapEqPool builds on the map equation, an information-theoretic objective function for community detection based on the minimum description length principle which naturally implements Occam's razor and balances between model complexity and fit. We demonstrate MapEqPool's competitive performance with an empirical comparison against various baselines across standard graph classification datasets.

Submitted: Sep 16, 2024