Systematic Generalization with Edge Transformers
Not on the map. Why?
Abstract (arXiv)
Recent research suggests that systematic generalization in natural language understanding remains a challenge for state-of-the-art neural models such as Transformers and Graph Neural Networks. To tackle this challenge, we propose Edge Transformer, a new model that combines inspiration from Transformers and rule-based symbolic AI. The first key idea in Edge Transformers is to associate vector states with every edge, that is, with every pair of input nodes -- as opposed to just every node, as it is done in the Transformer model. The second major innovation is a triangular attention mechanism that updates edge representations in a way that is inspired by unification from logic programming. We evaluate Edge Transformer on compositional generalization benchmarks in relational reasoning, semantic parsing, and dependency parsing. In all three settings, the Edge Transformer outperforms Relation-aware, Universal and classical Transformer baselines.
Key facts. Cited by 38 papers in the map (data through 14 Oct 2026). Builds on 35 of its 44 references that are in the map. First posted on arXiv 1 Dec 2021.
Builds on
35 of 44 references.
References · 35
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2020End-to-End Object Detection with Transformerscited by 7,414
- 2016Layer Normalizationcited by 5,669
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
- 2020Rethinking Attention with Performerscited by 1,601
- 2018Self-Attention with Relative Position Representationscited by 1,120
- 2020Stanza: A Python Natural Language Processing Toolkit for Many Human Languagescited by 723
23 more
- 2018Universal Transformerscited by 647
- 2019The Neuro-Symbolic Concept Learner: Interpreting Scenes, Words, and Sentences From Natural Supervisioncited by 372
- 2016Deep Biaffine Attention for Neural Dependency Parsingcited by 334
- 2019Measuring Compositional Generalization: A Comprehensive Method on Realistic Datacited by 284
- 2019Compositionality decomposed: how do neural networks generalise?cited by 265
- 2020COGS: A Compositional Generalization Challenge Based on Semantic Interpretationcited by 263
- 2017Learning Explanatory Rules from Noisy Datacited by 211
- 2019CLUTRR: A Diagnostic Benchmark for Inductive Reasoning from Textcited by 162
- 2017End-to-End Differentiable Provingcited by 161
- 2019Good-Enough Compositional Data Augmentationcited by 158
- 2021The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformerscited by 117
- 2019Compositional generalization through meta sequence-to-sequence learningcited by 99
- 2020Compositional Generalization in Semantic Parsing: Pre-training vs. Specialized Architecturescited by 95
- 2020Learning Compositional Rules via Neural Program Synthesiscited by 74
- 2021Making Transformers Solve Compositional Taskscited by 72
- 2020Learning to Recombine and Resample Data for Compositional Generalizationcited by 65
- 2020Learning Reasoning Strategies in End-to-End Differentiable Provingcited by 60
- 2020Measuring Systematic Generalization in Neural Proof Generation with Transformerscited by 43
- 2020Hierarchical Poset Decoding for Compositional Generalization in Languagecited by 26
- 2017Universal Semantic Parsingcited by 21
- 2020*-CFQ: Analyzing the Scalability of Machine Learning on a Compositional Taskcited by 16
- 2019Utilizing Edge Features in Graph Neural Networks via Variational Information Maximizationcited by 10
- 2021Compositional Generalization in Dependency Parsingcited by 4
Built on by
Citing papers per year: 2022: 5; 2023: 14; 2024: 7; 2025: 6; 2026: 6.
38 papers.
Sort:
By area:
- 2023Large Language Models can Learn Rulescited by 69
- 2023Coupling Large Language Models with Logic Programming for Robust and General Reasoning from Textcited by 59
- 2024MoEUT: Mixture-of-Experts Universal Transformerscited by 59
- 2022Relational Attention: Generalizing Transformers for Graph-Structured Taskscited by 41
- 2023Smooth, exact rotational symmetrization for deep learning on point cloudscited by 39
- 2023Sparse Universal Transformercited by 32
- 2024Graph Transformers: A Surveycited by 32
- 2022Uncontrolled Lexical Exposure Leads to Overestimation of Compositional Generalization in Pretrained Modelscited by 26
- 2023Neural-Logic Human-Object Interaction Detectioncited by 25
- 2023ReCOGS: How Incidental Details of a Logical Form Overshadow an Evaluation of Semantic Interpretationcited by 24
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstractioncited by 20
- 2025Strassen Attention, Split VC Dimension and Compositionality in Transformerscited by 12
26 more
- 2024Towards Principled Graph Transformerscited by 11
- 2025How simple can you go? An off-the-shelf transformer approach to molecular dynamicscited by 10
- 2025Fast and Simplex: 2-Simplicial Attention in Tritoncited by 8
- 2025Analyzing the Inner Workings of Transformers in Compositional Generalizationcited by 6
- 2022Transformer Module Networks for Systematic Generalization in Visual Question Answeringcited by 5
- 2023Learning Language Representations with Logical Inductive Biascited by 5
- 2024Towards Understanding the Relationship between In-context Learning and Compositional Generalizationcited by 5
- 2024Systematic Relational Reasoning With Epistemic Graph Neural Networkscited by 4
- 2024Automatic Screening for Children with Speech Disorder using Automatic Speech Recognition: Opportunities and Challengescited by 3
- 2023Laziness Is a Virtue When It Comes to Compositionality in Neural Semantic Parsingcited by 2
- 2023D3: Data Diversity Design for Systematic Generalization in Visual Question Answeringcited by 2
- 2023MetaReVision: Meta-Learning with Retrieval for Visually Grounded Compositional Concept Acquisitioncited by 2
- 2024SHAKTI: A 2.5 Billion Parameter Small Language Model Optimized for Edge AI and Low-Resource Environmentscited by 2
- 2025When No Paths Lead to Rome: Benchmarking Systematic Neural Relational Reasoningcited by 2
- 2022Embeddings as Epistemic States: Limitations on the Use of Pooling Operators for Accumulating Knowledgecited by 1
- 2023HICO-DET-SG and V-COCO-SG: New Data Splits for Evaluating the Systematic Generalization Performance of Human-Object Interaction Detection Modelscited by 1
- 2023A Data Source for Reasoning Embodied Agentscited by 1
- 2026FloydNet: A Learning Paradigm for Global Relational Reasoningcited by 1
- 2026Graph Machine: Exploring Edge Mechanisms as an Inductive Biascited by 1
- 2022A Short Survey of Systematic Generalizationcited by 0
- 2023Friend Ranking in Online Games via Pre-training Edge Transformerscited by 0
- 2025Scaling Graph-Based Dependency Parsing with Arc Vectorization and Attention-Based Refinementcited by 0
- 2026Efficient Lookahead Encoding and Abstracted Width for Learning General Policies in Classical Planningcited by 0
- 2026Project Auto-World: Towards Automated Benchmarking of Neural Relational Reasonerscited by 0
- 2026Higher-Order Fourier Neural Operator: Explicit Mode Mixer for Nonlinear PDEscited by 0
- 2026Linearized 2-Simplicial Attentioncited by 0
- 2026Linearized 2-Simplicial Attention10 Aug 2026
- 2026Graph Machine: Exploring Edge Mechanisms as an Inductive Bias7 Aug 2026
- 2026Higher-Order Fourier Neural Operator: Explicit Mode Mixer for Nonlinear PDEs26 Jun 2026
- 2026Project Auto-World: Towards Automated Benchmarking of Neural Relational Reasoners23 Jun 2026
- 2026Efficient Lookahead Encoding and Abstracted Width for Learning General Policies in Classical Planning18 May 2026
- 2026FloydNet: A Learning Paradigm for Global Relational Reasoning27 Jan 2026
- 2025When No Paths Lead to Rome: Benchmarking Systematic Neural Relational Reasoning27 Oct 2025
- 2025Fast and Simplex: 2-Simplicial Attention in Triton3 Jul 2025
- 2025How simple can you go? An off-the-shelf transformer approach to molecular dynamics3 Mar 2025
- 2025Analyzing the Inner Workings of Transformers in Compositional Generalization21 Feb 2025
- 2025Strassen Attention, Split VC Dimension and Compositionality in Transformers31 Jan 2025
- 2025Scaling Graph-Based Dependency Parsing with Arc Vectorization and Attention-Based Refinement16 Jan 2025
26 more
- 2024SHAKTI: A 2.5 Billion Parameter Small Language Model Optimized for Edge AI and Low-Resource Environments15 Oct 2024
- 2024Automatic Screening for Children with Speech Disorder using Automatic Speech Recognition: Opportunities and Challenges7 Oct 2024
- 2024Systematic Relational Reasoning With Epistemic Graph Neural Networks24 Jul 2024
- 2024Graph Transformers: A Survey13 Jul 2024
- 2024MoEUT: Mixture-of-Experts Universal Transformers25 May 2024
- 2024Towards Understanding the Relationship between In-context Learning and Compositional Generalization18 Mar 2024
- 2024Towards Principled Graph Transformers18 Jan 2024
- 2023Neural-Logic Human-Object Interaction Detection16 Nov 2023
- 2023MetaReVision: Meta-Learning with Retrieval for Visually Grounded Compositional Concept Acquisition2 Nov 2023
- 2023Sparse Universal Transformer11 Oct 2023
- 2023Large Language Models can Learn Rules10 Oct 2023
- 2023D3: Data Diversity Design for Systematic Generalization in Visual Question Answering15 Sep 2023
- 2023A Data Source for Reasoning Embodied Agents14 Sep 2023
- 2023Coupling Large Language Models with Logic Programming for Robust and General Reasoning from Text15 Jul 2023
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstraction4 Jun 2023
- 2023Smooth, exact rotational symmetrization for deep learning on point clouds30 May 2023
- 2023HICO-DET-SG and V-COCO-SG: New Data Splits for Evaluating the Systematic Generalization Performance of Human-Object Interaction Detection Models17 May 2023
- 2023Laziness Is a Virtue When It Comes to Compositionality in Neural Semantic Parsing7 May 2023
- 2023ReCOGS: How Incidental Details of a Logical Form Overshadow an Evaluation of Semantic Interpretation24 Mar 2023
- 2023Friend Ranking in Online Games via Pre-training Edge Transformers20 Feb 2023
- 2023Learning Language Representations with Logical Inductive Bias19 Feb 2023
- 2022Uncontrolled Lexical Exposure Leads to Overestimation of Compositional Generalization in Pretrained Models21 Dec 2022
- 2022A Short Survey of Systematic Generalization22 Nov 2022
- 2022Embeddings as Epistemic States: Limitations on the Use of Pooling Operators for Accumulating Knowledge11 Oct 2022
- 2022Relational Attention: Generalizing Transformers for Graph-Structured Tasks11 Oct 2022
- 2022Transformer Module Networks for Systematic Generalization in Visual Question Answering27 Jan 2022
- 2023Coupling Large Language Models with Logic Programming for Robust and General Reasoning from Textcited by 59
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstractioncited by 20
- 2023A Data Source for Reasoning Embodied Agentscited by 1
- 2026Project Auto-World: Towards Automated Benchmarking of Neural Relational Reasonerscited by 0
- 2026Project Auto-World: Towards Automated Benchmarking of Neural Relational Reasoners23 Jun 2026
- 2023A Data Source for Reasoning Embodied Agents14 Sep 2023
- 2023Coupling Large Language Models with Logic Programming for Robust and General Reasoning from Text15 Jul 2023
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstraction4 Jun 2023
- 2024MoEUT: Mixture-of-Experts Universal Transformerscited by 59
- 2023Sparse Universal Transformercited by 32
- 2024MoEUT: Mixture-of-Experts Universal Transformers25 May 2024
- 2023Sparse Universal Transformer11 Oct 2023
- 2023Large Language Models can Learn Rulescited by 69
- 2022Relational Attention: Generalizing Transformers for Graph-Structured Taskscited by 41
- 2023Smooth, exact rotational symmetrization for deep learning on point cloudscited by 39
- 2024Graph Transformers: A Surveycited by 32
- 2022Uncontrolled Lexical Exposure Leads to Overestimation of Compositional Generalization in Pretrained Modelscited by 26
- 2023Neural-Logic Human-Object Interaction Detectioncited by 25
- 2023ReCOGS: How Incidental Details of a Logical Form Overshadow an Evaluation of Semantic Interpretationcited by 24
- 2025Strassen Attention, Split VC Dimension and Compositionality in Transformerscited by 12
- 2024Towards Principled Graph Transformerscited by 11
- 2025How simple can you go? An off-the-shelf transformer approach to molecular dynamicscited by 10
- 2025Fast and Simplex: 2-Simplicial Attention in Tritoncited by 8
- 2025Analyzing the Inner Workings of Transformers in Compositional Generalizationcited by 6
12 most cited of 32 All 32 (JSON)
- 2026Linearized 2-Simplicial Attention10 Aug 2026
- 2026Graph Machine: Exploring Edge Mechanisms as an Inductive Bias7 Aug 2026
- 2026Higher-Order Fourier Neural Operator: Explicit Mode Mixer for Nonlinear PDEs26 Jun 2026
- 2026Efficient Lookahead Encoding and Abstracted Width for Learning General Policies in Classical Planning18 May 2026
- 2026FloydNet: A Learning Paradigm for Global Relational Reasoning27 Jan 2026
- 2025When No Paths Lead to Rome: Benchmarking Systematic Neural Relational Reasoning27 Oct 2025
- 2025Fast and Simplex: 2-Simplicial Attention in Triton3 Jul 2025
- 2025How simple can you go? An off-the-shelf transformer approach to molecular dynamics3 Mar 2025
- 2025Analyzing the Inner Workings of Transformers in Compositional Generalization21 Feb 2025
- 2025Strassen Attention, Split VC Dimension and Compositionality in Transformers31 Jan 2025
- 2025Scaling Graph-Based Dependency Parsing with Arc Vectorization and Attention-Based Refinement16 Jan 2025
- 2024SHAKTI: A 2.5 Billion Parameter Small Language Model Optimized for Edge AI and Low-Resource Environments15 Oct 2024
12 newest of 32 All 32 (JSON)