English

MaxCutPool: differentiable feature-aware Maxcut for pooling in graph neural networks

Machine Learning 2025-02-03 v3

Abstract

We propose a novel approach to compute the MAXCUT in attributed graphs, i.e., graphs with features associated with nodes and edges. Our approach works well on any kind of graph topology and can find solutions that jointly optimize the MAXCUT along with other objectives. Based on the obtained MAXCUT partition, we implement a hierarchical graph pooling layer for Graph Neural Networks, which is sparse, trainable end-to-end, and particularly suitable for downstream tasks on heterophilic graphs.

Keywords

Cite

@article{arxiv.2409.05100,
  title  = {MaxCutPool: differentiable feature-aware Maxcut for pooling in graph neural networks},
  author = {Carlo Abate and Filippo Maria Bianchi},
  journal= {arXiv preprint arXiv:2409.05100},
  year   = {2025}
}

Comments

Accepted at ICLR 2025

R2 v1 2026-06-28T18:37:44.706Z