Local-Global Geometric Insights for Graph Neural Networks via Entropic Curvature
AuthorsRachid Caich, Yassine Abbahaddou
Resources
This work uses global geometric curvature to explain and improve how information flows through graph neural networks.
Key results
Improves the backbone in 16 of 20 backbone–dataset pairs at p < 0.05.
ENT achieves the best result on 4 of 5 node-classification benchmarks.
Test accuracy in percent for SAGE with E-Gate on Wisconsin.
Approximate improvement reached by MCR on stochastic block models after 10 added edges.
What the paper found
In “Local–Global Geometric Insights for Graph Neural Networks via Entropic Curvature,” Rachid Caich and Yassine Abbahaddou, from the Centre de recherches mathématiques at the University of Montréal and LIX at École Polytechnique, replace edge-local Ollivier–Ricci and Forman curvature with a global transport-based notion grounded in Lott–Sturm–Villani entropy convexity along W1-Wasserstein geodesics. Their tractable two-hop proxy, weak entropic curvature κw, provably lower-bounds global curvature and yields Poincaré-type control of oversmoothing, transport–entropy generalization bounds, and an expansion paradox: large sparse graphs cannot simultaneously have strong spectral expansion and positive curvature, unifying oversquashing and oversmoothing as opposite geometric regimes. The theory produces three practical tools: E-Gate, which modulates message aggregation by node curvature; ENT, a three-dimensional curvature encoding; and Midpoint-Completion Rewiring, or MCR, which adds alternative two-hop routes to bottlenecks. E-Gate improves the backbone in 16 of 20 tested backbone–dataset pairs at p < 0.05, while ENT leads competing encodings on 4 of 5 benchmarks. On Wisconsin, SAGE with E-Gate reaches 79.22 percent accuracy versus 75.10 percent for baseline SAGE. Across six random-graph families, MCR consistently beats Ollivier- and Forman-based rewiring, reaching a spectral-gap improvement of approximately 0.6 on stochastic block models after 10 added edges. Experiments use PyTorch Geometric and NVIDIA GPUs, including the RTX 3090.
Original abstract
Curvature notions on graphs, particularly Ollivier-Ricci and Forman, have emerged as powerful tools for addressing fundamental issues in Graph Neural Networks (GNNs) such as oversmoothing and oversquashing, but rely almost exclusively on local edge-level comparisons and therefore fail to certify how information actually propagates over long distances. We introduce Entropic Curvature, a global, transport-based curvature obtained by extending the Lott-Sturm-Villani framework to graphs through the displacement convexity of entropy along Wasserstein geodesics. We define a tractable Weak Entropic Curvature proxy that lower-bounds the global entropic curvature, and from it derive (i) a Poincare-type inequality controlling oversmoothing, (ii) a transport-entropy generalization bound, and (iii) an expansion paradox proving that sparsity, strong spectral expansion, and positive entropic curvature cannot coexist in large graphs, unifying oversmoothing and oversquashing as opposite ends of a single curvature spectrum. We translate the theory into three practical mechanisms, the E-Gate aggregator, the ENT structural encoding, and Midpoint-Completion Rewiring (MCR), and benchmark them against SDRF, FoSR, BORF, LCP, and Graph Ricci Flow on six node-classification benchmarks, and graph-classification.
Read the original paperMore in Graph Learning
Browse all 32 papers →CodeGraph: Open-Taxonomy Knowledge Graph for Source Code with Wikidata Grounding
Federico Pennino, Andrea Gurioli, Stefano Zacchiroli, Maurizio Gabbrielli, Paolo Ferragina
CodeGraph turns 167 million source files into a Wikidata-grounded knowledge graph of algorithms, paradigms, patterns, and software domains.
GraphWrit3R: End-to-End 3D Scene Graph Writing
Luka Milivojevic, Nikola Popovic, Sayan Deb Sarkar, Sebastian Koch, Iro Armeni, Luc Van Gool, Danda Pani Paudel
GraphWrit3R turns 3D spatial data into open-vocabulary scene graphs using multimodal encoders and an LLM, without requiring ground-truth object annotations at inference.
Statistical Inference for Causal Discovery under Selection and Latent Variables via Single-Target Interventions
Xiaotian Hou, Kwangmoon Park, Hongzhe Li
This work shows how a small, carefully designed set of single-variable interventions can recover causal structure even when hidden confounders and selection bias complicate the data.