Enhancing the Transformer with Explicit Relational Encoding for Math Problem Solving
Not on the map. Why?
Abstract (arXiv)
We incorporate Tensor-Product Representations within the Transformer in order to better support the explicit representation of relation structure. Our Tensor-Product Transformer (TP-Transformer) sets a new state of the art on the recently-introduced Mathematics Dataset containing 56 categories of free-form math word-problems. The essential component of the model is a novel attention mechanism, called TP-Attention, which explicitly encodes the relations between each Transformer cell and the other cells from which values have been retrieved by attention. TP-Attention goes beyond linear combination of retrieved values, strengthening representation-building and resolving ambiguities introduced by multiple layers of standard attention. The TP-Transformer's attention maps give better insights into how it is capable of solving the Mathematics Dataset's challenging problems. Pretrained models and code will be made available after publication.
Key facts. Cited by 35 papers in the map (data through 14 Oct 2026). First posted on arXiv 15 Oct 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 10; 2022: 8; 2023: 3; 2024: 8; 2025: 3; 2026: 3.
35 papers.
Sort:
By area:
- 2024Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attentioncited by 200
- 2023Embers of Autoregression: Understanding Large Language Models Through the Problem They are Trained to Solvecited by 178
- 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 139
- 2021How much do language models copy from their training data? Evaluating linguistic novelty in text generation using RAVENcited by 124
- 2022Causal Transformer for Estimating Counterfactual Outcomescited by 123
- 2021The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformerscited by 117
- 2021Going Beyond Linear Transformers with Recurrent Fast Weight Programmerscited by 89
- 2021LIME: Learning Inductive Bias for Primitives of Mathematical Reasoningcited by 46
- 2022Neurocompositional computing: From the Central Paradox of Cognition to a new generation of AI systemscited by 44
- 2022Formal Specifications from Natural Languagecited by 33
- 2021Compositional Processing Emerges in Neural Networks Solving Math Problemscited by 11
- 2021Enriching Transformers with Structured Tensor-Product Representations for Abstractive Summarizationcited by 11
23 more
- 2022Induced Natural Language Rationales and Interleaved Markup Tokens Enable Extrapolation in Large Language Modelscited by 10
- 2021Learning Symbolic Rules for Reasoning in Quasi-Natural Languagecited by 9
- 2024Probabilistic Abduction for Visual Abstract Reasoning via Learning Rules in Vector-symbolic Architecturescited by 9
- 2024Towards Learning Abductive Reasoning using VSA Distributed Representationscited by 9
- 2025Fast weight programming and linear transformers: from machine learning to neurobiologycited by 8
- 2024From Frege to chatGPT: Compositionality in language, cognition, and deep neural networkscited by 7
- 2023LLMs as Potential Brainstorming Partners for Math and Science Problemscited by 6
- 2021Generating Symbolic Reasoning Problems with Transformer GANscited by 5
- 2022Structural Biases for Improving Transformers on Translation into Morphologically Rich Languagescited by 5
- 2023Differentiable Tree Operations Promote Compositional Generalizationcited by 5
- 2022Enhancing Neural Mathematical Reasoning by Abductive Combination with Symbolic Librarycited by 4
- 2025Survey on Question Answering over Visually Rich Documents: Methods, Challenges, and Trendscited by 4
- 2024Arctic-TILT. Business Document Understanding at Sub-Billion Scalecited by 3
- 2025Beyond Sliding Windows: Learning to Manage Memory in Non-Markovian Environmentscited by 3
- 2021A Reinforcement Learning Environment for Mathematical Reasoning via Program Synthesiscited by 2
- 2022Semantic Representations of Mathematical Expressions in a Continuous Vector Spacecited by 2
- 2024Attention-based Iterative Decomposition for Tensor Product Representationcited by 2
- 2024Terminating Differentiable Tree Expertscited by 1
- 2026Tensor Product Representation Probes Reveal Shared Structure Across Linear Directionscited by 1
- 2026The Emergent Symbolic Structure of Artificial Neural Networkscited by 1
- 2022A Short Survey of Systematic Generalizationcited by 0
- 2024Discrete Dictionary-based Decomposition Layer for Structured Representation Learningcited by 0
- 2026TANGO: Treating Tokens as Operatorscited by 0
- 2026The Emergent Symbolic Structure of Artificial Neural Networks30 Aug 2026
- 2026TANGO: Treating Tokens as Operators22 Aug 2026
- 2026Tensor Product Representation Probes Reveal Shared Structure Across Linear Directions11 May 2026
- 2025Beyond Sliding Windows: Learning to Manage Memory in Non-Markovian Environments22 Dec 2025
- 2025Fast weight programming and linear transformers: from machine learning to neurobiology11 Aug 2025
- 2025Survey on Question Answering over Visually Rich Documents: Methods, Challenges, and Trends4 Jan 2025
- 2024Arctic-TILT. Business Document Understanding at Sub-Billion Scale8 Aug 2024
- 2024Terminating Differentiable Tree Experts2 Jul 2024
- 2024Towards Learning Abductive Reasoning using VSA Distributed Representations27 Jun 2024
- 2024Discrete Dictionary-based Decomposition Layer for Structured Representation Learning11 Jun 2024
- 2024Attention-based Iterative Decomposition for Tensor Product Representation3 Jun 2024
- 2024From Frege to chatGPT: Compositionality in language, cognition, and deep neural networks24 May 2024
23 more
- 2024Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention10 Apr 2024
- 2024Probabilistic Abduction for Visual Abstract Reasoning via Learning Rules in Vector-symbolic Architectures29 Jan 2024
- 2023LLMs as Potential Brainstorming Partners for Math and Science Problems10 Oct 2023
- 2023Embers of Autoregression: Understanding Large Language Models Through the Problem They are Trained to Solve24 Sep 2023
- 2023Differentiable Tree Operations Promote Compositional Generalization1 Jun 2023
- 2022A Short Survey of Systematic Generalization22 Nov 2022
- 2022Semantic Representations of Mathematical Expressions in a Continuous Vector Space8 Oct 2022
- 2022Induced Natural Language Rationales and Interleaved Markup Tokens Enable Extrapolation in Large Language Models24 Aug 2022
- 2022Structural Biases for Improving Transformers on Translation into Morphologically Rich Languages11 Aug 2022
- 2022Formal Specifications from Natural Language4 Jun 2022
- 2022Neurocompositional computing: From the Central Paradox of Cognition to a new generation of AI systems2 May 2022
- 2022Causal Transformer for Estimating Counterfactual Outcomes14 Apr 2022
- 2022Enhancing Neural Mathematical Reasoning by Abductive Combination with Symbolic Library28 Mar 2022
- 2021Learning Symbolic Rules for Reasoning in Quasi-Natural Language23 Nov 2021
- 2021How much do language models copy from their training data? Evaluating linguistic novelty in text generation using RAVEN18 Nov 2021
- 2021Generating Symbolic Reasoning Problems with Transformer GANs19 Oct 2021
- 2021The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers26 Aug 2021
- 2021A Reinforcement Learning Environment for Mathematical Reasoning via Program Synthesis15 Jul 2021
- 2021Going Beyond Linear Transformers with Recurrent Fast Weight Programmers11 Jun 2021
- 2021Enriching Transformers with Structured Tensor-Product Representations for Abstractive Summarization2 Jun 2021
- 2021Compositional Processing Emerges in Neural Networks Solving Math Problems19 May 2021
- 2021Investigating the Limitations of Transformers with Simple Arithmetic Tasks25 Feb 2021
- 2021LIME: Learning Inductive Bias for Primitives of Mathematical Reasoning15 Jan 2021
- 2024Probabilistic Abduction for Visual Abstract Reasoning via Learning Rules in Vector-symbolic Architecturescited by 9
- 2024Towards Learning Abductive Reasoning using VSA Distributed Representationscited by 9
- 2022Enhancing Neural Mathematical Reasoning by Abductive Combination with Symbolic Librarycited by 4
- 2024Towards Learning Abductive Reasoning using VSA Distributed Representations27 Jun 2024
- 2024Probabilistic Abduction for Visual Abstract Reasoning via Learning Rules in Vector-symbolic Architectures29 Jan 2024
- 2022Enhancing Neural Mathematical Reasoning by Abductive Combination with Symbolic Library28 Mar 2022
- 2024Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attentioncited by 200
- 2023Embers of Autoregression: Understanding Large Language Models Through the Problem They are Trained to Solvecited by 178
- 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 139
- 2021How much do language models copy from their training data? Evaluating linguistic novelty in text generation using RAVENcited by 124
- 2022Causal Transformer for Estimating Counterfactual Outcomescited by 123
- 2021The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformerscited by 117
- 2021Going Beyond Linear Transformers with Recurrent Fast Weight Programmerscited by 89
- 2021LIME: Learning Inductive Bias for Primitives of Mathematical Reasoningcited by 46
- 2022Neurocompositional computing: From the Central Paradox of Cognition to a new generation of AI systemscited by 44
- 2022Formal Specifications from Natural Languagecited by 33
- 2021Compositional Processing Emerges in Neural Networks Solving Math Problemscited by 11
- 2021Enriching Transformers with Structured Tensor-Product Representations for Abstractive Summarizationcited by 11
12 most cited of 31 All 31 (JSON)
- 2026The Emergent Symbolic Structure of Artificial Neural Networks30 Aug 2026
- 2026TANGO: Treating Tokens as Operators22 Aug 2026
- 2025Beyond Sliding Windows: Learning to Manage Memory in Non-Markovian Environments22 Dec 2025
- 2025Fast weight programming and linear transformers: from machine learning to neurobiology11 Aug 2025
- 2025Survey on Question Answering over Visually Rich Documents: Methods, Challenges, and Trends4 Jan 2025
- 2024Arctic-TILT. Business Document Understanding at Sub-Billion Scale8 Aug 2024
- 2024Terminating Differentiable Tree Experts2 Jul 2024
- 2024Discrete Dictionary-based Decomposition Layer for Structured Representation Learning11 Jun 2024
- 2024Attention-based Iterative Decomposition for Tensor Product Representation3 Jun 2024
- 2024From Frege to chatGPT: Compositionality in language, cognition, and deep neural networks24 May 2024
- 2024Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention10 Apr 2024
- 2023LLMs as Potential Brainstorming Partners for Math and Science Problems10 Oct 2023
12 newest of 31 All 31 (JSON)