Unitary Convolutions for Message-passing and Positional Encodings on Directed Graphs
Lukas Fesser, Bobak Kiani, Melanie Weber
Abstract
In many real-world networks, relationships are inherently directional, yet most graph neural networks (GNNs) assume undirected edges, and naïve adaptations of undirected GNNs to directed graphs amplify oversmoothing and gradient pathologies that cap model depth. Unitary graph convolutions (UniConv) provably prevent representational collapse and oversmoothing, but do not incorporate edge directionality or edge features. In this paper, we introduce a directed unitary GNN with edge features (Dune), which retains these guarantees while overcoming Uni-Conv's limitations by incorporating edge directionality and edge features. Dune keeps gradient norms bounded at any number of layers, allowing it to benefit from neural network depth, unlike existing directed GNNs. The same unitary operator can be embedded in hybrid architectures with graph transformers, where its wavelike propagation supplies positional information and reduces the importance of random-walk or Laplacian-based encodings. We prove that Dune avoids exponential oversmoothing that plagues existing directed GNNs, and empirically show that it achieves state-of-the-art performance on 12 directed-graph benchmarks while remaining trainable beyond 100 layers, improving performance by up to 18 percentage points over baselines. Our results establish unitary convolutions as a scalable, geometry-aware foundation for deep learning on directed graphs. Code is available here.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on24
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- Recipe for a General, Powerful, Scalable Graph TransformerLadislav Rampásek, Michael Galkin, Vijay Prakash Dwivedi, Anh Tuan Luu et al.NeurIPS 2022 · 1,216 citations
- Rethinking Graph Transformers with Spectral AttentionDevin Kreuzer, Dominique Beaini, William L. Hamilton, Vincent Létourneau et al.NeurIPS 2021 · 854 citations
- Large Scale Learning on Non-Homophilous Graphs: New Benchmarks and Strong Simple MethodsDerek Lim, Felix Hohne, Xiuyu Li, Sijia Linda Huang et al.NeurIPS 2021 · 534 citations
- Graph Neural Networks with Learnable Structural and Positional RepresentationsVijay Prakash Dwivedi, Anh Tuan Luu, Thomas Laurent, Yoshua Bengio et al.ICLR 2022 · 464 citations
Related papers
- Unitary Convolutions for Learning on Graphs and GroupsBobak T. Kiani, Lukas Fesser, Melanie WeberNeurIPS 2024 · 13 citations
- LightDiC: A Simple yet Effective Approach for Large-scale Digraph Representation LearningXunkai Li, Meihao Liao, Zhengyu Wu, Daohan Su et al.VLDB 2024 · 13 citations
- Orthogonal Graph Neural NetworksKai Guo, Kaixiong Zhou, Xia Hu, Yu Li et al.AAAI 2022 · 41 citations
- Dirichlet Energy Constrained Learning for Deep Graph Neural NetworksKaixiong Zhou, Xiao Huang, Daochen Zha, Rui Chen et al.NeurIPS 2021 · 171 citations
- Graph Neural Networks for Edge Signals: Orientation Equivariance and InvarianceDominik Fuchsgruber, Tim Postuvan, Stephan Günnemann, Simon GeislerICLR 2025
