NeurIPS2024

Weisfeiler and Leman Go Loopy: A New Hierarchy for Graph Representational Learning

Raffaele Paolino, Sohir Maskey, Pascal Welke, Gitta Kutyniok

摘要

We introduce rr-loopy Weisfeiler-Leman (rr-\ell{}WL), a novel hierarchy of graph isomorphism tests and a corresponding GNN framework, rr-\ell{}MPNN, that can count cycles up to length r+2r + 2. Most notably, we show that rr-\ell{}WL can count homomorphisms of cactus graphs. This strictly extends classical 1-WL, which can only count homomorphisms of trees and, in fact, is incomparable to kk-WL for any fixed kk. We empirically validate the expressive and counting power of the proposed rr-\ell{}MPNN on several synthetic datasets and present state-of-the-art predictive performance on various real-world datasets. The code is available at https://github.com/RPaolino/loopy