Neural Arithmetic Logic Units
Not on the map. Why?
Abstract (arXiv)
Neural networks can learn to represent and manipulate numerical information, but they seldom generalize well outside of the range of numerical values encountered during training. To encourage more systematic numerical extrapolation, we propose an architecture that represents numerical quantities as linear activations which are manipulated using primitive arithmetic operators, controlled by learned gates. We call this module a neural arithmetic logic unit (NALU), by analogy to the arithmetic logic unit in traditional processors. Experiments show that NALU-enhanced neural networks can learn to track time, perform arithmetic over images of numbers, translate numerical language into real-valued scalars, execute computer code, and count objects in images. In contrast to conventional architectures, we obtain substantially better generalization both inside and outside of the range of numerical values encountered during training, often extrapolating orders of magnitude beyond trained numerical ranges.
Key facts. Cited by 69 papers in the map (data through 7 Oct 2026). First posted on arXiv 1 Aug 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 0; 2020: 0; 2021: 19; 2022: 14; 2023: 14; 2024: 4; 2025: 5; 2026: 12.
69 papers.
Sort:
By area:
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,131
- 2018Scalable agent alignment via reward modeling: a research directioncited by 438
- 2021Combinatorial optimization and reasoning with graph neural networkscited by 267
- 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 136
- 2022The CLRS Algorithmic Reasoning Benchmarkcited by 103
- 2021Representing Numbers in NLP: a Survey and a Visioncited by 83
- 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
- 2023Length Generalization in Arithmetic Transformerscited by 64
- 2021Linear algebra with transformerscited by 62
- 2022Theory of Graph Neural Networks: Representation and Learningcited by 60
- 2022Deep Symbolic Regression for Recurrent Sequencescited by 53
- 2021Quadratic Residual Networks: A New Class of Neural Networks for Solving Forward and Inverse Problems in Physics Involving PDEscited by 42
38 more
- 2022Relational Attention: Generalizing Transformers for Graph-Structured Taskscited by 41
- 2023Learning the greatest common divisor: explaining transformer predictionscited by 33
- 2021Dangers of Bayesian Model Averaging under Covariate Shiftcited by 31
- 2022SALSA: Attacking Lattice Cryptography with Transformerscited by 28
- 2021Symbolic Brittleness in Sequence Models: on Systematic Generalization in Symbolic Mathematicscited by 27
- 2022Deep Learning and Symbolic Regression for Discovering Parametric Equationscited by 22
- 2023Generative learning for nonlinear dynamicscited by 22
- 2023Can neural networks do arithmetic? A survey on the elementary numerical skills of state-of-the-art deep learning modelscited by 18
- 2021MC-LSTM: Mass-Conserving LSTMcited by 17
- 2021Pretrained Language Models are Symbolic Mathematics Solvers too!cited by 17
- 2023Neural Algorithmic Reasoning Without Intermediate Supervisioncited by 16
- 2024Softened Symbol Grounding for Neuro-symbolic Systemscited by 16
- 2021Investigating Numeracy Learning Ability of a Text-to-Text Transfer Modelcited by 13
- 2021Truth-Conditional Captioning of Time Series Datacited by 11
- 2022Induced Natural Language Rationales and Interleaved Markup Tokens Enable Extrapolation in Large Language Modelscited by 10
- 2024Arithmetic Feature Interaction Is Necessary for Deep Tabular Learningcited by 8
- 2022Transformers discover an elementary calculation system exploiting local attention and grid-like problem representationcited by 7
- 2022Automatic Rule Induction for Interpretable Semi-Supervised Learningcited by 6
- 2023Deciphering Raw Data in Neuro-Symbolic Learning with Provable Guaranteescited by 6
- 2023A Comprehensive Evaluation of Tool-Assisted Generation Strategiescited by 6
- 2025KANO: Kolmogorov-Arnold Neural Operatorcited by 6
- 2021Neural Sequence-to-grid Module for Learning Symbolic Rulescited by 5
- 2021Evolutionary Training and Abstraction Yields Algorithmic Generalization of Neural Computerscited by 5
- 2021Learning to Pool in Graph Neural Networks for Extrapolationcited by 4
- 2022Asynchronous Neural Networks for Learning in Graphscited by 4
- 2023Exploring the Approximation Capabilities of Multiplicative Neural Networks for Smooth Functionscited by 4
- 2025Learning to Add, Multiply, and Execute Algorithmic Instructions Exactly with Neural Networkscited by 4
- 2022Logical Tasks for Measuring Extrapolation and Rule Comprehensioncited by 3
- 2024Acceleration of Grokking in Learning Arithmetic Operations via Kolmogorov-Arnold Representationcited by 3
- 2025Teaching Metric Distance to Discrete Autoregressive Language Modelscited by 3
- 2022Physics Informed Symbolic Networkscited by 2
- 2023Towards a Neural Lambda Calculus: Neurosymbolic AI Applied to the Foundations of Functional Programmingcited by 2
- 2023NeuralFastLAS: Fast Logic-Based Learning from Raw Datacited by 2
- 2026Explicit Fuzzy Logic in the Feed-Forward Layer: Self-Forgetting Quantifiers Discover Legible Grammatical-Licensing Detectorscited by 2
- 2021Multi-Class Multi-Instance Count Conditioned Adversarial Image Generationcited by 1
- 2022Open World Learning Graph Convolution for Latency Estimation in Routing Networkscited by 1
- 2023Learning to solve arithmetic problems with a virtual abacuscited by 1
- 2023A Hybrid System for Systematic Generalization in Simple Arithmetic Problemscited by 1
50 most cited of 69 All 69 (JSON)
- 2026The Geometry of Logic: Stratification Induces Semantic Structure and Robust Reasoning26 Sep 2026
- 2026Mask-Based Priors Are More Persistent than Query-Key Initializations1 Aug 2026
- 2026Explicit Fuzzy Logic in the Feed-Forward Layer: Self-Forgetting Quantifiers Discover Legible Grammatical-Licensing Detectors30 Jun 2026
- 2026TriPLU: Bypassing the Gate with Direct Trilinear Product FFNs in Tiny Language Models17 Jun 2026
- 2026BRo-JEPA: Learning Modular Transformations in Latent Space31 May 2026
- 2026Universal Quantum Transformer29 Apr 2026
- 2026THEIA: Learning Complete Kleene Three-Valued Logic in a Pure-Neural Modular Architecture13 Apr 2026
- 2026Reasoning as Energy Minimization over Structured Latent Trajectories30 Mar 2026
- 2026A Small-Scale System for Autoregressive Program Synthesis Enabling Controlled Experimentation9 Feb 2026
- 2026From Fuzzy to Exact: The Halo Architecture for Infinite-Depth Reasoning via Rational Arithmetic26 Jan 2026
- 2026Even GPT-5.2 Can't Count to Five: The Case for Zero-Error Horizons in Trustworthy LLMs22 Jan 2026
- 2026Differentiable Logic Synthesis: Spectral Coefficient Selection via Sinkhorn-Constrained Composition20 Jan 2026
38 more
- 2025KANO: Kolmogorov-Arnold Neural Operator20 Sep 2025
- 2025The Domain Mixed Unit: A New Neural Arithmetic Layer9 Sep 2025
- 2025Prime Convolutional Model: Breaking the Ground for Theoretical Explainability4 Mar 2025
- 2025Teaching Metric Distance to Discrete Autoregressive Language Models4 Mar 2025
- 2025Learning to Add, Multiply, and Execute Algorithmic Instructions Exactly with Neural Networks24 Feb 2025
- 2024The Extrapolation Power of Implicit Models19 Jul 2024
- 2024Acceleration of Grokking in Learning Arithmetic Operations via Kolmogorov-Arnold Representation26 May 2024
- 2024Softened Symbol Grounding for Neuro-symbolic Systems1 Mar 2024
- 2024Arithmetic Feature Interaction Is Necessary for Deep Tabular Learning4 Feb 2024
- 2023Generative learning for nonlinear dynamics7 Nov 2023
- 2023A Comprehensive Evaluation of Tool-Assisted Generation Strategies16 Oct 2023
- 2023NeuralFastLAS: Fast Logic-Based Learning from Raw Data8 Oct 2023
- 2023Ensemble Mask Networks12 Sep 2023
- 2023Learning the greatest common divisor: explaining transformer predictions29 Aug 2023
- 2023Deciphering Raw Data in Neuro-Symbolic Learning with Provable Guarantees21 Aug 2023
- 2023A Hybrid System for Systematic Generalization in Simple Arithmetic Problems29 Jun 2023
- 2023Length Generalization in Arithmetic Transformers27 Jun 2023
- 2023Neural Algorithmic Reasoning Without Intermediate Supervision23 Jun 2023
- 2023Towards a Neural Lambda Calculus: Neurosymbolic AI Applied to the Foundations of Functional Programming18 Apr 2023
- 2023NPS: A Framework for Accurate Program Sampling Using Graph Neural Network18 Apr 2023
- 2023Can neural networks do arithmetic? A survey on the elementary numerical skills of state-of-the-art deep learning models14 Mar 2023
- 2023Learning to solve arithmetic problems with a virtual abacus17 Jan 2023
- 2023Exploring the Approximation Capabilities of Multiplicative Neural Networks for Smooth Functions11 Jan 2023
- 2022Logical Tasks for Measuring Extrapolation and Rule Comprehension14 Nov 2022
- 2022Improving the Robustness of Neural Multiplication Units with Reversible Stochasticity10 Nov 2022
- 2022Relational Attention: Generalizing Transformers for Graph-Structured Tasks11 Oct 2022
- 2022Induced Natural Language Rationales and Interleaved Markup Tokens Enable Extrapolation in Large Language Models24 Aug 2022
- 2022Physics Informed Symbolic Networks11 Jul 2022
- 2022SALSA: Attacking Lattice Cryptography with Transformers11 Jul 2022
- 2022Open World Learning Graph Convolution for Latency Estimation in Routing Networks8 Jul 2022
- 2022Transformers discover an elementary calculation system exploiting local attention and grid-like problem representation6 Jul 2022
- 2022Deep Learning and Symbolic Regression for Discovering Parametric Equations1 Jul 2022
- 2022The CLRS Algorithmic Reasoning Benchmark31 May 2022
- 2022Asynchronous Neural Networks for Learning in Graphs24 May 2022
- 2022Automatic Rule Induction for Interpretable Semi-Supervised Learning18 May 2022
- 2022Theory of Graph Neural Networks: Representation and Learning16 Apr 2022
- 2022Deep Symbolic Regression for Recurrent Sequences12 Jan 2022
- 2021Linear algebra with transformers3 Dec 2021
50 newest of 69 All 69 (JSON)
- 2023Can neural networks do arithmetic? A survey on the elementary numerical skills of state-of-the-art deep learning modelscited by 18
- 2022Transformers discover an elementary calculation system exploiting local attention and grid-like problem representationcited by 7
- 2023Deciphering Raw Data in Neuro-Symbolic Learning with Provable Guaranteescited by 6
- 2022Logical Tasks for Measuring Extrapolation and Rule Comprehensioncited by 3
- 2025Teaching Metric Distance to Discrete Autoregressive Language Modelscited by 3
- 2023A Hybrid System for Systematic Generalization in Simple Arithmetic Problemscited by 1
- 2026From Fuzzy to Exact: The Halo Architecture for Infinite-Depth Reasoning via Rational Arithmeticcited by 1
- 2026Reasoning as Energy Minimization over Structured Latent Trajectoriescited by 0
- 2026The Geometry of Logic: Stratification Induces Semantic Structure and Robust Reasoningcited by 0
- 2026The Geometry of Logic: Stratification Induces Semantic Structure and Robust Reasoning26 Sep 2026
- 2026Reasoning as Energy Minimization over Structured Latent Trajectories30 Mar 2026
- 2026From Fuzzy to Exact: The Halo Architecture for Infinite-Depth Reasoning via Rational Arithmetic26 Jan 2026
- 2025Teaching Metric Distance to Discrete Autoregressive Language Models4 Mar 2025
- 2023Deciphering Raw Data in Neuro-Symbolic Learning with Provable Guarantees21 Aug 2023
- 2023A Hybrid System for Systematic Generalization in Simple Arithmetic Problems29 Jun 2023
- 2023Can neural networks do arithmetic? A survey on the elementary numerical skills of state-of-the-art deep learning models14 Mar 2023
- 2022Logical Tasks for Measuring Extrapolation and Rule Comprehension14 Nov 2022
- 2022Transformers discover an elementary calculation system exploiting local attention and grid-like problem representation6 Jul 2022
- 2026BRo-JEPA: Learning Modular Transformations in Latent Space31 May 2026
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,131
- 2021Combinatorial optimization and reasoning with graph neural networkscited by 267
- 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 136
- 2022The CLRS Algorithmic Reasoning Benchmarkcited by 103
- 2021Representing Numbers in NLP: a Survey and a Visioncited by 83
- 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
- 2023Length Generalization in Arithmetic Transformerscited by 64
- 2021Linear algebra with transformerscited by 62
- 2022Theory of Graph Neural Networks: Representation and Learningcited by 60
- 2022Deep Symbolic Regression for Recurrent Sequencescited by 53
- 2021Quadratic Residual Networks: A New Class of Neural Networks for Solving Forward and Inverse Problems in Physics Involving PDEscited by 42
- 2022Relational Attention: Generalizing Transformers for Graph-Structured Taskscited by 41
12 most cited of 58 All 58 (JSON)
- 2026Mask-Based Priors Are More Persistent than Query-Key Initializations1 Aug 2026
- 2026Explicit Fuzzy Logic in the Feed-Forward Layer: Self-Forgetting Quantifiers Discover Legible Grammatical-Licensing Detectors30 Jun 2026
- 2026TriPLU: Bypassing the Gate with Direct Trilinear Product FFNs in Tiny Language Models17 Jun 2026
- 2026Universal Quantum Transformer29 Apr 2026
- 2026THEIA: Learning Complete Kleene Three-Valued Logic in a Pure-Neural Modular Architecture13 Apr 2026
- 2026A Small-Scale System for Autoregressive Program Synthesis Enabling Controlled Experimentation9 Feb 2026
- 2026Even GPT-5.2 Can't Count to Five: The Case for Zero-Error Horizons in Trustworthy LLMs22 Jan 2026
- 2026Differentiable Logic Synthesis: Spectral Coefficient Selection via Sinkhorn-Constrained Composition20 Jan 2026
- 2025KANO: Kolmogorov-Arnold Neural Operator20 Sep 2025
- 2025The Domain Mixed Unit: A New Neural Arithmetic Layer9 Sep 2025
- 2025Prime Convolutional Model: Breaking the Ground for Theoretical Explainability4 Mar 2025
- 2025Learning to Add, Multiply, and Execute Algorithmic Instructions Exactly with Neural Networks24 Feb 2025
12 newest of 58 All 58 (JSON)