Transformers over Directed Acyclic Graphs
Not on the map. Why?
Abstract (arXiv)
Transformer models have recently gained popularity in graph representation learning as they have the potential to learn complex relationships beyond the ones captured by regular graph neural networks. The main research question is how to inject the structural bias of graphs into the transformer architecture, and several proposals have been made for undirected molecular graphs and, recently, also for larger network graphs. In this paper, we study transformers over directed acyclic graphs (DAGs) and propose architecture adaptations tailored to DAGs: (1) An attention mechanism that is considerably more efficient than the regular quadratic complexity of transformers and at the same time faithfully captures the DAG structure, and (2) a positional encoding of the DAG's partial order, complementing the former. We rigorously evaluate our approach over various types of tasks, ranging from classifying source code graphs to nodes in citation networks, and show that it is effective in two important aspects: in making graph transformers generally outperform graph neural networks tailored to DAGs and in improving SOTA graph transformer performance in terms of both quality and efficiency.
Key facts. Cited by 23 papers in the map (data through 7 Oct 2026). Builds on 50 of its 72 references that are in the map. First posted on arXiv 24 Oct 2022.
Builds on
50 of 72 references.
References · 50
- 2017Attention Is All You Needcited by 59,932
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2018How Powerful are Graph Neural Networks?cited by 4,355
- 2017Neural Message Passing for Quantum Chemistrycited by 3,460
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,906
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2018Self-Attention with Relative Position Representationscited by 1,112
- 2018Predict then Propagate: Graph Neural Networks meet Personalized PageRankcited by 987
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 867
38 more
- 2018Efficient Neural Architecture Search via Parameter Sharingcited by 831
- 2022Recipe for a General, Powerful, Scalable Graph Transformercited by 702
- 2020A Generalization of Transformer Networks to Graphscited by 653
- 2020On the Bottleneck of Graph Neural Networks and its Practical Implicationscited by 624
- 2020Principal Neighbourhood Aggregation for Graph Netscited by 477
- 2021Understanding over-squashing and bottlenecks on graphs via curvaturecited by 471
- 2021Rethinking Graph Transformers with Spectral Attentioncited by 468
- 2019MixHop: Higher-Order Graph Convolutional Architectures via Sparsified Neighborhood Mixingcited by 442
- 2021Graph Neural Networks with Learnable Structural and Positional Representationscited by 326
- 2015Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networkscited by 305
- 2022Long Range Graph Benchmarkcited by 258
- 2018Modeling polypharmacy side effects with graph convolutional networkscited by 237
- 2019Learning Discrete Structures for Graph Neural Networkscited by 228
- 2017A Survey of Machine Learning for Big Code and Naturalnesscited by 220
- 2020Iterative Deep Graph Learning for Graph Neural Networks: Better and Robust Node Embeddingscited by 212
- 2022Representing Long-Range Context for Graph Neural Networks with Global Attentioncited by 193
- 2022Structure-Aware Transformer for Graph Representation Learningcited by 191
- 2018code2seq: Generating Sequences from Structured Representations of Codecited by 177
- 2021GraphiT: Encoding Graph Structure in Transformerscited by 146
- 2018Graph HyperNetworks for Neural Architecture Searchcited by 131
- 2022Transformer for Graphs: An Overview from Architecture Perspectivecited by 107
- 2021MagNet: A Neural Network for Directed Graphscited by 104
- 2019D-VAE: A Variational Autoencoder for Directed Acyclic Graphscited by 98
- 2021Directed Acyclic Graph Network for Conversational Emotion Recognitioncited by 97
- 2021Parameter Prediction for Unseen Deep Architecturescited by 89
- 2020Code Prediction by Feeding Trees to Transformerscited by 70
- 2021Directed Acyclic Graph Neural Networkscited by 70
- 2018MotifNet: a motif-based Graph Convolutional Network for directed graphscited by 63
- 2021Language-Agnostic Representation Learning of Source Code from Structure and Contextcited by 63
- 2019TreeGen: A Tree-Based Transformer Architecture for Code Generationcited by 58
- 2020A Large-Scale Database for Graph Representation Learningcited by 54
- 2019Spectral-based Graph Convolutional Network for Directed Graphscited by 38
- 2020Answering Complex Queries in Knowledge Graphs with Bidirectional Sequence Encoderscited by 28
- 2022On the Learning of Non-Autoregressive Transformerscited by 27
- 2019Improving Graph Neural Network Representations of Logical Formulae with Subgraph Poolingcited by 20
- 2022PACE: A Parallelizable Computation Encoder for Directed Acyclic Graphscited by 20
- 2022Neural Topological Ordering for Computation Graphscited by 12
- 2015DAG-Recurrent Neural Networks For Scene Labelingcited by 10
Built on by
Citing papers per year: 2023: 1; 2024: 9; 2025: 10; 2026: 3.
23 papers.
Sort:
- 2024Classic GNNs are Strong Baselines: Reassessing GNNs for Node Classificationcited by 83
- 2025Can Classic GNNs Be Strong Baselines for Graph-level Tasks? Simple Architectures Meet Excellencecited by 24
- 2023Enhancing Graph Transformers with Hierarchical Distance Structural Encodingcited by 16
- 2025DeepGate4: Efficient and Effective Representation Learning for Circuit Design at Scalecited by 12
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learningcited by 10
- 2024FlowerFormer: Empowering Neural Architecture Encoding using a Flow-aware Graph Transformercited by 9
- 2025Biologically Plausible Brain Graph Transformercited by 9
- 2024What Are Good Positional Encodings for Directed Graphs?cited by 7
- 2024Convolutional Learning on Directed Acyclic Graphscited by 6
- 2024Commute Graph Neural Networkscited by 5
- 2024DeepSeq2: Enhanced Sequential Circuit Learning with Disentangled Representationscited by 5
- 2025DualEquiNet: A Dual-Space Hierarchical Equivariant Network for Large Biomoleculescited by 5
11 more
- 2025NN-Former: Rethinking Graph Structure in Neural Architecture Representationcited by 2
- 2024Scale-aware Message Passing For Graph Node Classificationcited by 1
- 2025Simple Path Structural Encoding for Graph Transformerscited by 1
- 2025Flow-Attentional Graph Neural Networkscited by 1
- 2025Directed Acyclic Graph Convolutional Networkscited by 1
- 2024CaTs and DAGs: Integrating Directed Acyclic Graphs with Transformers for Causally Constrained Predictionscited by 0
- 2025Language Embedding Meets Dynamic Graph: A New Exploration for Neural Architecture Representation Learningcited by 0
- 2025State Space Models over Directed Graphscited by 0
- 2026Can Graph Learning Learn Circuits?cited by 0
- 2026Node-wise Feature Encoding for Neural Performance Predictioncited by 0
- 2026Learning Sparse Decision Trees via Transformer Variational Auto-Encoderscited by 0
- 2026Learning Sparse Decision Trees via Transformer Variational Auto-Encoders1 Sep 2026
- 2026Node-wise Feature Encoding for Neural Performance Prediction28 Aug 2026
- 2026Can Graph Learning Learn Circuits?9 Aug 2026
- 2025State Space Models over Directed Graphs17 Sep 2025
- 2025NN-Former: Rethinking Graph Structure in Neural Architecture Representation1 Jul 2025
- 2025Directed Acyclic Graph Convolutional Networks13 Jun 2025
- 2025DualEquiNet: A Dual-Space Hierarchical Equivariant Network for Large Biomolecules10 Jun 2025
- 2025Language Embedding Meets Dynamic Graph: A New Exploration for Neural Architecture Representation Learning9 Jun 2025
- 2025Flow-Attentional Graph Neural Networks6 Jun 2025
- 2025Simple Path Structural Encoding for Graph Transformers13 Feb 2025
- 2025Can Classic GNNs Be Strong Baselines for Graph-level Tasks? Simple Architectures Meet Excellence13 Feb 2025
- 2025Biologically Plausible Brain Graph Transformer13 Feb 2025
11 more
- 2025DeepGate4: Efficient and Effective Representation Learning for Circuit Design at Scale2 Feb 2025
- 2024Scale-aware Message Passing For Graph Node Classification28 Nov 2024
- 2024DeepSeq2: Enhanced Sequential Circuit Learning with Disentangled Representations1 Nov 2024
- 2024CaTs and DAGs: Integrating Directed Acyclic Graphs with Transformers for Causally Constrained Predictions18 Oct 2024
- 2024What Are Good Positional Encodings for Directed Graphs?30 Jul 2024
- 2024Commute Graph Neural Networks30 Jun 2024
- 2024Classic GNNs are Strong Baselines: Reassessing GNNs for Node Classification13 Jun 2024
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learning26 May 2024
- 2024Convolutional Learning on Directed Acyclic Graphs5 May 2024
- 2024FlowerFormer: Empowering Neural Architecture Encoding using a Flow-aware Graph Transformer19 Mar 2024
- 2023Enhancing Graph Transformers with Hierarchical Distance Structural Encoding22 Aug 2023
- 2024Classic GNNs are Strong Baselines: Reassessing GNNs for Node Classificationcited by 83
- 2025Can Classic GNNs Be Strong Baselines for Graph-level Tasks? Simple Architectures Meet Excellencecited by 24
- 2023Enhancing Graph Transformers with Hierarchical Distance Structural Encodingcited by 16
- 2025DeepGate4: Efficient and Effective Representation Learning for Circuit Design at Scalecited by 12
- 2024Node Identifiers: Compact, Discrete Representations for Efficient Graph Learningcited by 10
- 2024FlowerFormer: Empowering Neural Architecture Encoding using a Flow-aware Graph Transformercited by 9
- 2025Biologically Plausible Brain Graph Transformercited by 9
- 2024What Are Good Positional Encodings for Directed Graphs?cited by 7
- 2024Convolutional Learning on Directed Acyclic Graphscited by 6
- 2024Commute Graph Neural Networkscited by 5
- 2024DeepSeq2: Enhanced Sequential Circuit Learning with Disentangled Representationscited by 5
- 2025DualEquiNet: A Dual-Space Hierarchical Equivariant Network for Large Biomoleculescited by 5
12 most cited of 23 All 23 (JSON)
- 2026Learning Sparse Decision Trees via Transformer Variational Auto-Encoders1 Sep 2026
- 2026Node-wise Feature Encoding for Neural Performance Prediction28 Aug 2026
- 2026Can Graph Learning Learn Circuits?9 Aug 2026
- 2025State Space Models over Directed Graphs17 Sep 2025
- 2025NN-Former: Rethinking Graph Structure in Neural Architecture Representation1 Jul 2025
- 2025Directed Acyclic Graph Convolutional Networks13 Jun 2025
- 2025DualEquiNet: A Dual-Space Hierarchical Equivariant Network for Large Biomolecules10 Jun 2025
- 2025Language Embedding Meets Dynamic Graph: A New Exploration for Neural Architecture Representation Learning9 Jun 2025
- 2025Flow-Attentional Graph Neural Networks6 Jun 2025
- 2025Simple Path Structural Encoding for Graph Transformers13 Feb 2025
- 2025Can Classic GNNs Be Strong Baselines for Graph-level Tasks? Simple Architectures Meet Excellence13 Feb 2025
- 2025Biologically Plausible Brain Graph Transformer13 Feb 2025
12 newest of 23 All 23 (JSON)