arXiv Machine Learning By R\'emi Bourgerie, \v{S}ar\=unas Girdzijauskas, Viktoria Fodor

Deep Neural Sheaf Diffusion

Read the original on arXiv Machine Learning →

Deep Neural Sheaf Diffusion (DNSD) is a new graph neural network architecture that replaces the sheaf Laplacian with a sheaf adjacency operator to keep informative signals across many layers. It also uses normalization, odd nonlinearities, and gating, and replaces scalar attention with matrix-valued edge functions. Experiments show DNSD outperforms existing GNNs and Neural Sheaf Diffusion on both synthetic long‑range and real‑world graph benchmarks, achieving up to 30 percentage‑point gains in accuracy.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv AI
Oct 27

Graph-dependent pooling: adaptive granularity in hierarchical graph neural networks

Graph-Dependent Pooling (GDPool) introduces a dynamic approach to hierarchical graph pooling, allowing each graph to determine its own pooling granularity rather than relying on fixed cluster numbers or ratios. The method employs self-attention to select graph-specific anchors and uses soft node assignment to build coarser graphs, addressing optimization conflicts identified through gradient alignment analysis. Experiments on metal‑organic framework and molecular graph datasets demonstrate that GDPool outperforms existing methods in predictive accuracy, modularity, and exhibits stable behavior across various hyperparameters.

By Bohan Yoon, Hyejin Jang
Hugging Face Trending Papers
Aug 3

CoRe-GNN: Multilevel Message passing on Coarsened graphs

Training Graph Neural Networks on large graphs is challenged by the memory cost of storing all node representations across layers. We show that several existing scalable approaches can be written as structured modifications of the GNN propagation matrix, providing a unified perspective that exposes their respective limitations.

arXiv Machine Learning
Sep 29

Fixed Points Without Fixed Diffusion: Implicit Neural Sheaves for Convergent Test-Time Computation

The paper introduces SheafDEQ, a subhomogeneous deep-equilibrium architecture that uses adaptive neural-sheaf propagation to allow richer, edge-dependent transformations in implicit graph neural networks while guaranteeing a unique equilibrium. The authors prove that SheafDEQ’s equilibrium is globally reachable from any positive initialization and remains contractive even with bounded communication staleness. Experiments demonstrate that SheafDEQ outperforms fixed-propagation implicit baselines on tasks such as Sums, MNIST Terrain, Coordinates, and community detection, especially as graph connectivity becomes increasingly heterophilic.

By R\'emi Bourgerie, \v{S}ar\={u}nas Girdzijauskas, Viktoria Fodor