1 Introduction
The ability to represent and manipulate numerical quantities is apparent in the behavior of many species, from insects to mammals to humans, suggesting that basic quantitative reasoning is a general component of intelligence Dehaene (2011); Gallistel (2017).
While neural networks can successfully represent and manipulate numerical quantities given an appropriate learning signal, the behavior that they learn does not generally exhibit systematic generalization Fodor and Pylyshyn (1988); Marcus (2003). Specifically, one frequently observes failures when quantities that lie outside the numerical range used during training are encountered at test time, even when the target function is simple (e.g., it depends only on aggregating counts or linear extrapolation). This failure pattern indicates that the learned behavior is better characterized by memorization than by systematic abstraction. Whether input distribution shifts that trigger extrapolation failures are of practical concern depends on the environments where the trained models will operate. However, considerable evidence exists showing that animals as simple as bees demonstrate systematic numerical extrapolation gallistel:2000; Gallistel (2017), suggesting that systematicity in reasoning about numerical quantities is ecologically advantageous.
In this paper, we develop a new module that can be used in conjunction with standard neural network architectures (e.g., LSTMs or convnets) but which is biased to learn systematic numerical computation. Our strategy is to represent numerical quantities as individual neurons without a nonlinearity. To these singlevalue neurons, we apply operators that are capable of representing simple functions (e.g.,
, ,, etc.). These operators are controlled by parameters which determine the inputs and operations used to create each output. However, despite this combinatorial character, they are differentiable, making it possible to learn them with backpropagation
Rumelhart et al. (1986).We experiment across a variety of task domains (synthetic, image, text, and code), learning signals (supervised and reinforcement learning), and structures (feedforward and recurrent). We find that our proposed model can learn functions over representations that capture the underlying numerical nature of the data and generalize to numbers that are several orders of magnitude larger than those observed during training. We also observe that our module exhibits a superior numeracy bias relative to linear layers, even when no extrapolation is required. In one case our model exceeds a stateoftheart image counting network by an error margin of 54%. Notably, the only modification we made over the previous stateoftheart was the replacement of its last linear layer with our model.
1.1 Numerical Extrapolation Failures in Neural Networks
To illustrate the failure of systematicity in standard networks, we show the behavior of various MLPs trained to learn the scalar identity function, which is the most straightforward systematic relationship possible. The notion that neural networks struggle to learn identity relations is not new He et al. (2016). We show this because, even though many of the architectures evaluated below could theoretically represent the identity function, they typically fail to acquire it.
In Figure 1, we show the nature of this failure (experimental details and more detailed results in Appendix A
). We train an autoencoder to take a scalar value as input (e.g., the number 3), encode the value within its hidden layers (distributed representations), then reconstruct the input value as a linear combination of the last hidden layer (3 again). Each autoencoder we train is identical in its parameterization (3 hidden layers of size 8), tuning (10,000 iterations, learning rate of 0.01, squared error loss), and initialization, differing only on the choice of nonlinearity on hidden layers. For each point in Figure
1, we train 100 models to encode numbers between 5 and 5 and average their ability to encode numbers between 20 and 20.We see that even over a basic task using a simple architecture, all nonlinear functions fail to learn to represent numbers outside of the range seen during training. The severity of this failure directly corresponds to the degree of nonlinearity within the chosen activation function. Some activations learn to be highly linear (such as PReLU) which reduces error somewhat, but sharply nonlinear functions such as sigmoid and tanh fail consistently. Thus, despite the fact that neural networks are capable of representing functions that extrapolate, in practice we find that they fail to learn to do so.
2 The Neural Accumulator & Neural Arithmetic Logic Unit
Here we propose two models that are able to learn to represent and manipulate numbers in a systematic way. The first supports the ability to accumulate quantities additively, a desirable inductive bias for linear extrapolation. This model forms the basis for a second model, which supports multiplicative extrapolation. This model also illustrates how an inductive bias for arbitrary arithmetic functions can be effectively incorporated into an endtoend model.
Our first model is the neural accumulator
(NAC), which is a special case of a linear (affine) layer whose transformation matrix
consists just of1’s, 0’s, and 1’s; that is, its outputs are additions or subtractions (rather than arbitrary rescalings) of rows in the input vector. This prevents the layer from changing the scale of the representations of the numbers when mapping the input to the output, meaning that they are consistent throughout the model, no matter how many operations are chained together. We improve the inductive bias of a simple linear layer by encouraging 0’s, 1’s, and
1’s within in the following way.Since a hard constraint enforcing that every element of be one of would make learning hard, we propose a continuous and differentiable parameterization of in terms of unconstrained parameters: . This form is convenient for learning with gradient descent and produces matrices whose elements are guaranteed to be in and biased to be close to 1, 0, or 1.^{1}^{1}1The stable points correspond to the saturation points of either or .
The model contains no bias vector, and no squashing nonlinearity is applied to the output.
The Neural Accumulator (NAC) is a linear transformation of its inputs. The transformation matrix is the elementwise product of
and . The Neural Arithmetic Logic Unit (NALU) uses two NACs with tied weights to enable addition/subtraction (smaller purple cell) and multiplication/division (larger purple cell), controlled by a gate (orange cell).While addition and subtraction enable many useful systematic generalizations, a similarly robust ability to learn more complex mathematical functions, such as multiplication, may be be desirable. Figure 2 describes such a cell, the neural arithmetic logic unit (NALU), which learns a weighted sum between two subcells, one capable of addition and subtraction and the other capable of multiplication, division, and power functions such as . Importantly, the NALU demonstrates how the NAC can be extended with gatecontrolled suboperations, facilitating endtoend learning of new classes of numerical functions. As with the NAC, there is the same bias against learning to rescale during the mapping from input to output.
The NALU consists of two NAC cells (the purple cells) interpolated by a learned sigmoidal gate
(the orange cell), such that if the add/subtract subcell’s output value is applied with a weight of 1 (on), the multiply/divide subcell’s is 0 (off) and vice versa. The first NAC (the smaller purple subcell) computes the accumulation vector , which stores results of the NALU’s addition/subtraction operations; it is computed identically to the original NAC, (i.e., ). The second NAC (the larger purple subcell) operates in log space and is therefore capable of learning to multiply and divide, storing its results in :where prevents . Altogether, this cell can learn arithmetic functions consisting of multiplication, addition, subtraction, division, and power functions in a way that extrapolates to numbers outside of the range observed during training.
3 Related Work
Numerical reasoning is central to many problems in intelligence and by extension is an important topic in deep learning
Dehaene (2011). A widely studied task is counting objects in images Arteta et al. (2014); Chan et al. (2008); NIPS2010_4043; Seguí et al. (2015); Xie et al. (2018); Zhang et al. (2015). These models generally take one of two approaches: 1) using a deep neural network to segment individual instances of a particular object and explicitly counting them in a postprocessing step or 2) learning endtoend to predict object counts via a regression loss. Our work is more closely related to the second strategy.Other work more explicitly attempts to model numerical representations and arithmetic functions within the context of learning to execute small snippets of code Zaremba and Sutskever (2014); Reed and de Freitas (2016). Learning to count within a bounded range has also been included in various questionanswer tasks, notably the BaBI tasks Weston et al. (2015a), and many models successfully learn to do so Antol et al. (2015); Kumar et al. (2016); Graves et al. (2016). However, to our knowledge, no tasks of this kind explicitly require counting beyond the range observed during training.
One can also view our work as advocating a new context for linear activations within deep neural networks. This is related to recent architectural innovations such as ResNets He et al. (2016), Highway Networks Srivastava et al. (2015), and DenseNet Huang et al. (2016), which also advocate for linear connections to reduce exploding/vanishing gradients and promote better learning bias. Such connections improved performance, albeit with additional computational overhead due to the increased depth of the resulting architectures.
Our work is also in line with a broader theme in machine learning which seeks to identify, in the form of behaviorgoverning equations, the underlying structure of systems that extrapolate well to unseen parts of the space
Brunton et al. (2016). This is a strong trend in recent neural network literature concerning the systematic representation of concepts within recurrent memory, allowing for functions over these concepts to extrapolate to sequences longer than observed during training. The question of whether and how recurrent networks generalize to sequences longer than they encountered in training has been of enduring interest, especially since wellformed sentences in human languages are apparently unbounded in length, but are learned from a limited sample Gers and Schmidhuber (2001); Lake and Baroni (2018); Weiss et al. (2018). Recent work has also focused on augmenting LSTMs with systematic external memory modules, allowing them to generalize operations such as sorting Weston et al. (2015b); Graves et al. (2014); Grefenstette et al. (2015), again with special interest in generalization to sequences longer than observed during training through systematic abstraction.Finally, the cognitive and neural bases of numerical reasoning in humans and animals have been intensively studied; for a popular overview see Dehaene (2011). Our models are reminiscient of theories which posit that magnitudes are represented as continuous quantities manipulated by accumulation operations leibovich:2016, and, in particular, our singleneuron representation of number recalls Gelman and Gallistel’s posited “numerons”—individual neurons that represent numbers Gelman and Gallistel (1978). However, across many species, continuous quantities appear to be represented using an approximate representation where acuity decreases with magnitude Piazza et al. (2004), quite different from our model’s constant precision.
4 Experiments
The experiments in this paper test numeric reasoning and extrapolation in a variety of settings. We study the explicit learning of simple arithmetic functions directly from numerical input, and indirectly from image data. We consider temporal domains: the translation of text to integer values, and the evaluation of computer programs containing conditional logic and arithmetic. These supervised tasks are supplemented with a reinforcement learning task which implicitly involves counting to keep track of time. We conclude with the previouslystudied MNIST parity task where we obtain stateoftheart prediction accuracy and provide an ablation study to understand which components of NALU provide the most benefit.
4.1 Simple Function Learning Tasks
Static Task (test)  Recurrent Task (test)  

Relu6  None  NAC  NALU  LSTM  ReLU  NAC  NALU  
Interpolation 
0.2  0.0  0.0  0.0  0.0  0.0  0.0  0.0  
0.0  0.0  0.0  0.0  0.0  0.0  0.0  0.0  
3.2  20.9  21.4  0.0  0.0  0.0  1.5  0.0  
4.2  35.0  37.1  5.3  0.0  0.0  1.2  0.0  
0.7  4.3  22.4  0.0  0.0  0.0  2.3  0.0  
0.5  2.2  3.6  0.0  0.0  0.0  2.1  0.0  
Extrapolation 
42.6  0.0  0.0  0.0  96.1  85.5  0.0  0.0  
29.0  0.0  0.0  0.0  97.0  70.9  0.0  0.0  
10.1  29.5  33.3  0.0  98.2  97.9  88.4  0.0  
37.2  52.3  61.3  0.7  95.6  863.5  >999  999  
47.0  25.1  53.3  0.0  98.0  98.0  123.7  0.0  
10.3  20.0  16.4  0.0  95.8  34.1  >999  0.0 
In these initial synthetic experiments, we demonstrate the ability of NACs and NALUs to learn to select relevant inputs and apply different arithmetic functions to them, which are the key functions they are designed to be able to solve (below we will use these as components in more complex architectures). We have two task variants: one where the inputs are presented all at once as a single vector (the static tasks) and a second where inputs are presented sequentially over time (the recurrent tasks). Inputs are randomly generated, and for the target, two values ( and ) are computed as a sum over regular parts of the input. An operation (e.g., ) is then computed providing the training (or evaluation) target. The model is trained endtoend by minimizing the squared loss, and evaluation looks at performance of the model on heldout values from within the training range (interpolation) or on values from outside of the training range (extrapolation). Experimental details and more detailed results are in Appendix B.
On the static task, for baseline models we compare the NAC and NALU to MLPs with a variety of standard nonlinearities as well as a linear model. We report the baseline with the best median heldout performance, which is the Relu6 activation Krizhevsky and Hinton (2010). Results using additional nonlinearities are also in Appendix B. For the recurrent task, we report the performance of an LSTM and the best performing RNN variant from among several common architectures, an RNN with ReLU activations (additional recurrent baselines also in Appendix B).
Table 1 summarizes results and shows that while several standard architectures succeed at these tasks in the interpolation case, none of them succeed at extrapolation. However, in both interpolation and extrapolation, the NAC succeeds at modeling addition and subtraction, whereas the more flexible NALU succeeds at multiplicative operations as well (except for division in the recurrent task^{2}^{2}2Division is much more challenging to extrapolate. While our models limit numbers using nonlinearities, our models are still able to represent numbers that are very, very small. Division allows such small numbers to be in the denominator, greatly amplifying even small drifts in extrapolation ability.).
4.2 MNIST Counting and Arithmetic Tasks
In the previous synthetic task, both inputs and outputs were provided in a generalizationready representation (as floating point numbers), and only the internal operations and representations had to be learned in a way that generalized. In this experiment, we discover whether backpropagation can learn the representation of nonnumeric inputs to NACs/NALUs.
In these tasks, a recurrent network is fed a series of 10 randomly chosen MNIST digits and at the end of the series it must output a numerical value about the series it observed.^{3}^{3}3The input to the recurrent networks is the output the convnet in https://github.com/pytorch/examples/tree/master/mnist.
In the MNIST Digit Counting task, the model must learn to count how many images of each type it has seen (a 10way regression), and in the MNIST Digit Addition task, it must learn to compute the sum of the digits it observed (a linear regression). Each training series is formed using images from the MNIST digit training set, and each testing series from the MNIST test set. Evaluation occurs over heldout sequences of length 10 (interpolation), and two extrapolation lengths: 100 and 1000. Although no direct supervision of the convnet is provided, we estimate how well it has learned to distinguish digits by passing in a test sequences of length 1 (also from the MNIST test dataset) and estimating the accuracy based on the count/sum. Parameters are initialized randomly and trained by backpropagating the mean squared error against the target count vector or the sum.
MNIST Digit Counting Test  MNIST Digit Addition Test  
Classification  Mean Absolute Error  Classification  Mean Absolute Error  
Seq Len  1  10  100  1000  1  10  100  1000 
LSTM  98.29%  0.79  18.2  198.5  0.0%  14.8  800.8  8811.6 
GRU  99.02%  0.73  18.0  198.3  0.0%  1.75  771.4  8775.2 
RNNtanh  38.91%  1.49  18.4  198.7  0.0%  2.98  20.4  200.7 
RNNReLU  9.80%  0.66  39.8  88.18%  19.1  182.1  1171.0  
NAC  99.23%  0.12  0.76  3.32  97.6%  1.42  7.88  57.3 
NALU  97.6%  0.17  0.93  4.18  77.7%  5.11  26.8  248.9 
Table 2 shows the results for both tasks. As we saw before, standard architectures succeed on heldout sequences in the interpolation length, but they completely fail at extrapolation. Notably, the RNNtanh and RNNReLU models also fail to learn to interpolate to shorter sequences than seen during training. However, the NAC and NALU both extrapolate and interpolate well.
4.3 Language to Number Translation Tasks
Neural networks have also been quite successful in working with natural language inputs, and LSTMbased models are stateoftheart in many tasks Graves et al. (2013); Sutskever et al. (2014); Jozefowicz et al. (2016). However, much like other numerical input, it is not clear whether representations of number words are learned in a systematic way. To test this, we created a new translation task which translates a text number expression (e.g., five hundred and fifteen) into a scalar representation (515).
We trained and tested using numbers from 0 to 1000. The training set consists of the numbers 0–19 in addition to a random sample from the rest of the interval, adjusted to make sure that each unique token is present at least once in the training set. There are 169 examples in the training set, 200 in validation, and 631 in the test test. All networks trained on this dataset start with a token embedding layer, followed by encoding through an LSTM, and then a linear layer, NAC, or NALU.
Model  Train MAE  Validation MAE  Test MAE 

LSTM  0.003  29.9  29.4 
LSTM + NAC  80.0  114.1  114.3 
LSTM + NALU  0.12  0.39  0.41 
We observed that both baseline LSTM variants overfit severely to the 169 training set numbers and generalize poorly. The LSTM + NAC performs poorly on both training and test sets. The LSTM + NALU achieves the best generalization performance by a wide margin, suggesting that the multiplier is important for this task.
We show in Figure 3 the intermediate states of the NALU on randomly selected test examples. Without supervision, the model learns to track sensible estimates of the unknown number up to the current token. This allows the network to predict given tokens it has never seen before in isolation, such as e.g. eighty, since it saw eighty one, eighty four and eighty seven during training.^{4}^{4}4Note slight accumulation for the word and owing to the spurious correlation between the use of the word and a subsequent increase in the target value. The and token can be exploited to form addition expressions (see last example), even though these were not seen in training.
4.4 Program Evaluation
Evaluating a program requires the control of several logical and arithmetic operations and internal bookkeeping of intermediate values. We consider the two program evaluation tasks defined in Zaremba and Sutskever (2014). The first consists of simply adding two large integers, and the latter involves evaluating programs containing several operations (if statements, , ). We focus on extrapolation: can the network learn a solution that generalizes to larger value ranges? We investigate this by training with twodigit input integers pulled uniformly from and evaluating on random integers with three and four digits.
Following the setup of Zaremba and Sutskever (2014) we report the the percentage of matching digits between the rounded prediction from the model and target integer, however we handle numeric input differently. Instead of passing the integers characterbycharacter, we pass the full integer value at a single time step and regress the output with an RMSE loss. Our model setup consists of a NALU that is “configured by” an LSTM, that is, its parameters , , and are learned functions of the LSTM output at each timestep. Thus, the LSTM learns to control the NALU, dependent upon operations seen.
We compare to three popular RNNs (UGRNN, LSTM and DNC) and observe in both addition (Supplementary Figure 6) and program evaluation (Figure 4) that all models are able to solve the task at a fixed input domain, however only the NALU is able to extrapolate to larger numbers. In this case we see extrapolation is stable even when the domain is increased by two orders of magnitude.
4.5 Learning to Track Time in a GridWorld Environment
In all experiments thus far, our models have been trained to make numeric predictions. However, as discussed in the introduction, systematic numeric computation appears to underlie a diverse range of (natural) intelligent behaviors. In this task, we test whether a NAC can be used “internally” by an RLtrained agent to develop more systematic generalization to quantitative changes in its environment. We developed a simple gridworld environment task in which an agent is given a time (specified as a real value) and receives a reward if is arrives at a particular location at (and not before) that time. As illustrated in Figure 5, each episode in this task begins () with the agent and a single target red square randomly positioned in a gridworld. At each timestep, the agent receives as input a pixel representation of the state of the (entire) world, and must select a single discrete action from . At the start of the episode, the agent also receives a numeric (integer) instruction , which communicates the exact time the agent must arrive at its destination (the red square).
To achieve the maximum episode reward , the agent must select actions and move around so as to first step onto the red square precisely when . Training episodes end either when the agent reaches the red square or after timing out (). We first trained a conventional A3C agent Mnih et al. (2016) with a recurrent (LSTM) core memory, modified so that the instruction was communicated to the agent via an additional input unit concatenated to the output of the agent’s convnet visual module before being passed to the agent’s LSTM core memory. We also trained a second variant of the same architecture where the instruction was passed both directly to the LSTM memory and passed through an NAC and back into the LSTM. Both agents were trained on episodes where (eight being the lowest value of such that reaching the target destination when is always possible). Both agents quickly learned to master the training episodes. However, as shown in Figure 5, the agent with the NAC performed well on the task for , whereas performance of the standard A3C agent deteriorated for .
It is instructive to also consider why both agents eventually fail. As would be predicted by consideration of extrapolation error observed in previous models, for stimuli greater than 12 the baseline agent behaves as if the stimulus were still 12, arriving at the destination at (too early) and thus receiving incrementally less reward with larger stimuli. In contrast, for stimuli greater than 20, the agent with NAC never arrives at the destination. Note that in order to develop an agent that could plausibly follow both numerical and nonnumeric (e.g. linguistic or iconic) instructions, the instruction stimulus was passed both directly to the agent’s core LSTM and first through the NAC. We hypothesize that the more limited extrapolation (in terms of orders of magnitude) of the NAC here relative with other uses of the NAC was caused by the model still using the LSTM to encode numeracy to some degree.
4.6 MNIST Parity Prediction Task & Ablation Study
Layer Configuration  Test Acc. 

Seguí et al. (2015):  85.1 
Ours:  88.1 
60.0  
87.6  
91.4  
62.5  
88.7  
NAC:  93.1 
Thus far, we have emphasized the extrapolation successes; however, our results indicate that the NAC layer often performs extremely well at interpolation. In our final task, the MNIST parity task Seguí et al. (2015), we look explicitly at interpolation. Also, in this task, neither the input nor the output is directly provided as a number, but it implicitly invovles reasoning about numeric quantities. In these experiments, the NAC or its variants replace the last linear layer in the model proposed by Seguí et al. (2015)
, where it connects output of the convnet to the prediction softmax layer. Since the original model had an affine layer here, and a NAC is a constrained affine layer, we look systematically at the importance of each constraint. Table
4 summarizes the performance of the variant models. As we see, removing the bias and applying nonlinearities to the weights significantly increases the accuracy of the endtoend model, even though the majority of the parameters are not in the NAC itself. The NAC reduces the error of the previous best results by 54%.5 Conclusions
Current approaches to modeling numeracy in neural networks fall short because numerical representations fail to generalize outside of the range observed during training. We have shown how the NAC and NALU can be applied to rectify these two shortcomings across a wide variety of domains, facilitating both numerical representations and functions on numerical representations that generalize outside of the range observed during training. However, it is unlikely that NAC or NALU will be the perfect solution for every task. Rather, they exemplify a general design strategy for creating models that have biases intended for a target class of functions. This design strategy is enabled by the singleneuron number representation we propose, which allows arbitrary (differentiable) numerical functions to be added to the module and controlled via learned gates, as the NALU has exemplified between addition/subtraction and multiplication/division.
6 Acknowledgements
We thank Edward Grefenstette for suggesting the name Neural ALU and for valuable discussion involving numerical tasks. We also thank Stephen Clark, whose presentation on neural counting work directly inspired this one. Finally, we also thank Karl Moritz Hermann, John Hale, Richard Evans, David Saxton, and Angeliki Lazaridou for valuable comments and discussion.
References

Antol et al. [2015]
Stanislaw Antol, Aishwarya Agrawal, Jiasen Lu, Margaret Mitchell, Dhruv Batra,
C Lawrence Zitnick, and Devi Parikh.
VQA: Visual question answering.
In
Proceedings of the IEEE International Conference on Computer Vision
, pages 2425–2433, 2015.  Arteta et al. [2014] Carlos Arteta, Victor Lempitsky, J Alison Noble, and Andrew Zisserman. Interactive object counting. In European Conference on Computer Vision, pages 504–518, 2014.
 Brunton et al. [2016] Steven L Brunton, Joshua L Proctor, and J Nathan Kutz. Discovering governing equations from data by sparse identification of nonlinear dynamical systems. Proceedings of the National Academy of Sciences, 113(15):3932–3937, 2016.
 Chan et al. [2008] Antoni B Chan, ZhangSheng John Liang, and Nuno Vasconcelos. Privacy preserving crowd monitoring: Counting people without people models or tracking. In Proc. CVPR, pages 1–7. IEEE, 2008.
 Dehaene [2011] Stanislas Dehaene. The Number Sense: How the Mind Creates Mathematics. Oxford University Press, 2011.
 Fodor and Pylyshyn [1988] Jerry A. Fodor and Zenon W. Pylyshyn. Connectionism and cognitive architecture: a critical analysis. Cognition, 28(1–2):3–71, 1988.
 Gallistel [2017] C. Randy Gallistel. Finding numbers in the brain. Philosophical Transactions of the Royal Society B, 373, 2017.
 Gelman and Gallistel [1978] Rochel Gelman and C. Randy Gallistel. The child’s understanding of number. Harvard, 1978.
 Gers and Schmidhuber [2001] Felix A Gers and E Schmidhuber. LSTM recurrent networks learn simple contextfree and contextsensitive languages. IEEE Transactions on Neural Networks, 12(6):1333–1340, 2001.

Graves et al. [2013]
Alex Graves, Abdelrahman Mohamed, and Geoffrey Hinton.
Speech recognition with deep recurrent neural networks.
In Acoustics, speech and signal processing (icassp), 2013 ieee international conference on, pages 6645–6649. IEEE, 2013.  Graves et al. [2014] Alex Graves, Greg Wayne, and Ivo Danihelka. Neural turing machines. CoRR, abs/1410.5401, 2014. URL http://arxiv.org/abs/1410.5401.
 Graves et al. [2016] Alex Graves, Greg Wayne, Malcolm Reynolds, Tim Harley, Ivo Danihelka, Agnieszka GrabskaBarwińska, Sergio Gómez Colmenarejo, Edward Grefenstette, Tiago Ramalho, John Agapiou, et al. Hybrid computing using a neural network with dynamic external memory. Nature, 538(7626):471, 2016.
 Grefenstette et al. [2015] Edward Grefenstette, Karl Moritz Hermann, Mustafa Suleyman, and Phil Blunsom. Learning to transduce with unbounded memory. In Proc. NIPS, 2015.
 He et al. [2016] Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Identity mappings in deep residual networks. CoRR, abs/1603.05027, 2016. URL http://arxiv.org/abs/1603.05027.
 Huang et al. [2016] Gao Huang, Zhuang Liu, and Kilian Q. Weinberger. Densely connected convolutional networks. CoRR, abs/1608.06993, 2016. URL http://arxiv.org/abs/1608.06993.
 Jozefowicz et al. [2016] Rafal Jozefowicz, Oriol Vinyals, Mike Schuster, Noam Shazeer, and Yonghui Wu. Exploring the limits of language modeling. arXiv preprint arXiv:1602.02410, 2016.

Krizhevsky and Hinton [2010]
Alex Krizhevsky and Geoff Hinton.
Convolutional deep belief networks on CIFAR10.
Unpublished manuscript, 2010. 
Kumar et al. [2016]
Ankit Kumar, Ozan Irsoy, Peter Ondruska, Mohit Iyyer, James Bradbury, Ishaan
Gulrajani, Victor Zhong, Romain Paulus, and Richard Socher.
Ask me anything: Dynamic memory networks for natural language processing.
In Proc. ICML, pages 1378–1387, 2016.  Lake and Baroni [2018] Brendan Lake and Marco Baroni. Generalization without systematicity: On the compositional skills of sequencetosequence recurrent networks. In Proc. ICML, 2018.
 Marcus [2003] Gary F. Marcus. The Algebraic Mind: Integrating Connectionism and Cognitive Science. MIT Press, 2003.
 Mnih et al. [2016] Volodymyr Mnih, Adria Puigdomenech Badia, Mehdi Mirza, Alex Graves, Timothy Lillicrap, Tim Harley, David Silver, and Koray Kavukcuoglu. Asynchronous methods for deep reinforcement learning. In Proc. ICML, pages 1928–1937, 2016.
 Piazza et al. [2004] Manuela Piazza, Véronique Izard, Philippe Pinel, Denis Le Bihan, and Stanislas Dehaene. Tuning curves for approximate numerosity in the human intraparietal sulcus. Neuron, 44:547–555, 2004.
 Reed and de Freitas [2016] Scott E. Reed and Nando de Freitas. Neural programmerinterpreters. In Proc. ICLR, 2016.
 Rumelhart et al. [1986] David E. Rumelhart, Geoffrey E. Hinton, and Ronald J. Williams. Learning representations by backpropagating errors. Nature, 323(6088):533, 1986.
 Seguí et al. [2015] Santi Seguí, Oriol Pujol, and Jordi Vitrià. Learning to count with deep object features. CoRR, abs/1505.08082, 2015. URL http://arxiv.org/abs/1505.08082.
 Srivastava et al. [2015] Rupesh Kumar Srivastava, Klaus Greff, and Jürgen Schmidhuber. Highway networks. CoRR, abs/1505.00387, 2015. URL http://arxiv.org/abs/1505.00387.
 Sutskever et al. [2014] Ilya Sutskever, Oriol Vinyals, and Quoc V Le. Sequence to sequence learning with neural networks. In Advances in neural information processing systems, pages 3104–3112, 2014.
 Weiss et al. [2018] Gail Weiss, Yoav Goldberg, and Eran Yahav. On the practical computational power of finite precision RNNs for language recognition. In Proc. ACL, 2018.
 Weston et al. [2015a] Jason Weston, Antoine Bordes, Sumit Chopra, and Tomas Mikolov. Towards AIcomplete question answering: A set of prerequisite toy tasks. CoRR, abs/1502.05698, 2015a. URL http://arxiv.org/abs/1502.05698.
 Weston et al. [2015b] Jason Weston, Sumit Chopra, and Antoine Bordes. Memory networks. In Proc. ICLR, 2015b.
 Xie et al. [2018] Weidi Xie, J Alison Noble, and Andrew Zisserman. Microscopy cell counting and detection with fully convolutional regression networks. Computer methods in biomechanics and biomedical engineering: Imaging & Visualization, 6(3):283–292, 2018.
 Zaremba and Sutskever [2014] Wojciech Zaremba and Ilya Sutskever. Learning to execute. CoRR, abs/1410.4615, 2014. URL http://arxiv.org/abs/1410.4615.

Zhang et al. [2015]
Cong Zhang, Hongsheng Li, Xiaogang Wang, and Xiaokang Yang.
Crossscene crowd counting via deep convolutional neural networks.
In Proc. CVPR, pages 833–841. IEEE, 2015.
Appendix A Learning the Identity Function
Name  Error  % Error  Graph 

Hardtanh  495.47  99.1  
ReLU6  494.59  98.9  
Softsign  494.19  98.8  
Tanh  493.65  98.7  
Sigmoid  493.15  98.6  
Threshold  432.97  86.6  
SELU  365.66  73.1  
ELU  142.83  28.4  
Softshrink  119.17  23.8  
ReLU  83.63  16.7  
LeakyReLU  77.02  15.4  
Tanhshrink  57.03  11.4  
Softplus  37.60  7.5  
PReLU  14.66  2.9  
None  <0.0001  0.0 
In Table 5, we show the average error results for each MLP trained on the identity function on the range from 5 to 5 when evaluated on numbers ranging from 1000 to 1000. Pictured on the right is a small graph demonstrating the plotted shape for each nonlinearity. The important thing to note is that nonlinearities which are sharply nonlinear exhibit greater extrapolation error than those which are only mildly nonlinear. An error score of 500 is equivalent to simply predicting 0 for each target prediction.
Appendix B Synthetic Arithmetic Tasks
In the static tasks, a vector is given and the target is a scalar whose value is formed by first taking two random (but consistent) subsections and summing them and . The target is then the result of applying different arithmetic functions to and . The process must be learned endtoend by each model.
This task is challenging because the test set comes in two forms. The first is the interpolation test set, which never proposes an example to the model requiring , , or to represent a number greater than exceeded during training. The extrapolation test set, however, always includes at least one value of , , or that is greater than observed during training in each example.
In order to test the recurrent variant of the NAC, we propose second set of tasks which are only a slight modification of the first. Instead of the first step requiring sums over an input vector , the sum is computed over a timeseries where each step , , and . Training and interpolation testing occur over sequences of length 10. Testing occurs over sequences of length 1000 (forcing , , and to take on values that are much larger than observed during training). All baseline experiments simply use an MLP with one hidden layer containing a nonlinearity. For comparison, we stack two NALUs endtoend. All models use a hidden layer size of 2, the minimally required size to solve the task.
In Table 7, we show the raw scores for MLPs trained with a wide variety of nonlinearities over the hidden layer. As the scores for each task vary significantly, we also show scores for a randomly initialized model in the left column (headed “Random”) for context. Table 8 normalizes all scores reported by dividing them by this “Random” column, making them more easily comparable.
Tasks  Test  Random  Tanh  Sigmoid  Relu6  Softsign  SELU  ELU  ReLU  Crelu  None  NAC  NALU 
I  8.659  .0213  .0086  .0144  .0229  .0031  .0191  .0040  .0020  .0017  <.0001  <.0001  
E  120.9  52.37  51.14  51.56  47.78  .0126  .1064  .0225  .0127  .0013  <.0001  .0012  
I  6.478  .0344  .0183  .0010  .0616  .0046  .0510  .0133  .0035  .0007  <.0001  .0001  
E  42.14  11.54  9.627  12.22  9.734  5.60  3.458  6.108  .0090  .0007  <.0001  <.0001  
I  233.9  21.82  12.06  7.579  20.07  11.99  5.928  6.084  36.28  48.91  50.02  <.0001  
E  3647  971.3  608.9  366.6  998.6  380.7  187.8  183.0  197.9  1076  1215  <.0001  
I  1.175  .0514  .0421  .0499  .0558  .1218  .1233  .1233  .1011  .4113  .4363  .0625  
E  41.56  14.65  13.34  15.46  13.55  7.219  7.286  7.291  6.746  21.75  25.48  .2920  
I  151.7  .3327  .0672  1.027  1.479  6.462  .1744  .7829  .1593  6.495  34.09  <.0001  
E  4674  2168  2142  2195  1938  1237  406.6  569.4  408.4  1172  2490  <.0001  
I  1.476  .0124  .0026  .0080  .0101  .0046  .0055  .0265  .0027  .0329  .0538  <.0001  
E  4.180  .4702  .3571  .4294  .3237  .0739  .1006  .5066  .0622  .8356  .6876  <.0001 
Tasks  Relu6  Softsign  Tanh  Sigmoid  SELU  ELU  ReLU  Crelu  None  NAC  NALU 

Interpolation Test Error  Relative to Random Initialization Baseline  
0.2  0.3  0.2  0.1  0.0  0.2  0.0  0.0  0.0  0.0  0.0  
0.0  1.0  0.5  0.3  0.1  0.8  0.2  0.1  0.0  0.0  0.0  
3.2  8.6  9.3  5.2  5.1  2.5  2.6  15.5  20.9  21.4  0.0  
4.2  4.7  4.4  3.58  10.4  10.5  10.5  8.6  35.0  37.1  5.3  
0.7  1.0  0.2  0.0  4.3  0.1  0.5  0.1  4.3  22.4  0.0  
0.5  0.7  31.6  24.2  0.3  0.4  1.8  0.2  2.2  3.6  0.0  
Extrapolation Test Error  Relative to Random Initialization Baseline  
42.6  39.5  43.3  42.3  0.0  0.0  0.0  0.0  0.0  0.0  0.0  
29.0  23.1  27.3  22.8  13.3  8.2  14.5  0.0  0.0  0.0  0.0  
10.1  27.4  26.6  16.7  10.4  5.1  5.0  5.4  29.5  33.3  0.0  
37.2  32.6  35.3  32.1  17.4  17.5  17.5  16.2  52.3  61.3  0.7  
47.0  41.5  46.4  45.8  26.5  8.7  12.2  8.7  25.1  53.3  0.0  
10.3  7.7  11.2  8.5  1.7  2.4  12.1  1.5  20.0  16.4  0.0 
Model  T/I/E  

Random  I  38.03  30.86  30.17  30.69  29.12  30.92  28.84 
E  330341  336847  336769  336829  336451  336876  336452  
LSTM  T  .001771  .004392  .010654  .017190  .017723  .014044  .006919 
I  .000544  .037351  .015579  .004282  .014142  .016741  .002258  
E  330097  323877  326746  330618  321622  330264  322305  
GRU  T  .002800  .005177  .011987  .020600  .024754  .018683  .006823 
I  .000584  .006819  .026034  .005473  .059921  .029642  .002185  
E  330309  324036  321989  333336  318537  334332  321107  
RNN  TANH  T  .049635  .049862  .072149  .161817  .117094  .076251  .071165 
I  .041697  .020395  .040893  .169256  .115955  .340204  .068180  
E  332760  324180  324000  330137  323908  329339  320840  
RNN  RELU  T  .058062  .037226  .044367  .127584  .096943  .051219  .045705 
I  .005471  .010728  .019738  .091305  .073977  .017803  .007818  
E  325690  287928  238771  329636  2902047  330305  114725  
Neural Accumulator  T  .000002  .000001  .00002  .46505  .34161  .69882  .613314 
I  .00020  <.00001  .000001  .46862  0.350227  .70909  .617613  
E  1.8946  .00004  0.01537  297800  3083319119  416701  2152274742  
Neural ALU  T  <.000001  <.000001  .000171  .000074  .001740  .000164  .000703 
I  <.000001  <.000001  .000001  .000003  .000431  .000001  .003492  
E  <.000001  .013131  .013843  47.0244  >999999  2804.85  1671.81 
Model  

Interpolation Test  
LSTM  0.0  0.0  0.0  0.0  0.0  0.0  0.0 
GRU  0.0  0.0  0.0  0.0  0.0  0.0  0.0 
tanh  0.0  0.0  0.0  0.0  0.0  0.0  0.0 
ReLU  0.0  0.0  0.0  0.0  0.0  0.0  0.0 
NAC  0.0  0.0  0.0  1.5  1.2  2.3  2.1 
NALU  0.0  0.0  0.0  0.0  0.0  0.0  0.0 
Extrapolation Test  
LSTM  100.0  96.1  97.0  98.2  95.6  98.0  95.8 
GRU  100.0  96.2  95.6  99.0  94.7  99.2  95.4 
tanh  100.0  96.2  96.2  98.0  96.3  97.8  95.4 
ReLU  98.6  85.5  70.9  97.9  862.5  98.0  34.1 
NAC  0.0  0.0  0.0  88.4  >999.9  123.7  >999.9 
NALU  0.0  0.0  0.0  0.0  >999.9  0.0  0.0 
In Table 7, we observe that with very strong supervision over a simple task, most nonlinearities cannot learn functions requiring numeracy that generalize outside of the range observed during training. Note the consistency in these results with those in Table 5; common nonlinearities with a very small output range catastrophically fail to extrapolate in any case, even with strong supervision, notably including Sigmoid and Tanh which are ubiquitous in recurrent neural networks, leading to results in Table 9.
In Table 9, we observe the accuracy of various baselines and our models when used recurrently. The tanh and relu in Table 9 refer to vanilla RNNs with tanh and relu respectively applied to hidden states. All models successfully learn to interpolate over learned functions of the input. However, the when attempting to extrapolate a function learned over series of length 10 to series of length 1000, the NAC and NALU significantly outperform the baselines. Division, however, was quite challenging to extrapolate and no models were able to solve the task in a way that extrapolates. Baselines generally predicted a fixed range. However, the NAC and NALU underestimated the denominator, leading to quite significant error during extrapolation.
While these simple experiments are quite numerous, they exist to make clear a simple idea. Across a wide variety of nonlinearities and recurrent combinations of nonlinearities, most architectures can fit a training dataset requiring arithmetic; many can even learn to generalize to a test set if the numbers are within the same bounded range. However, modern neural architectures are illequipped to learn arithmetic in a systematic way. We did observe a few narrow exceptions in Table 7, but we will later show that even these do not hold when the supervised signal is more realistic (such as when this model is merely a piece of a large endtoend architecture). It is the systematic numerical representations present in NAC and NALU that create the desirable learning bias leading to accurate extrapolation. This will continue to be the key to systematic extrapolation in future experiments as well.
Appendix C MNIST Counting
We report the performance of MNIST counting and addition for the full set of models considered in Table 10.
MNIST Digit Counting (test)  MNIST Digit Addition (test)  
Classification  Mean Squared Error  Classification  Mean Squared Error  
Seq Len  1  10  100  1000  1  10  100  1000 
LSTM  98.29%  1.14  181.06  19883  0.0%  168.18  321738  38761851 
GRU  99.02%  1.12  180.95  19886  0.0%  168.09  321826  38784947 
RNNtanh  38.91%  1.53  226.95  20346  0.0%  167.19  321841  38784910 
RNNReLU  9.80%  0.54  160.80  19608  88.18%  4.29  882.87  10969180 
NAC  99.23%  0.03  0.26  3  97.58%  2.82  28.11  280.89 
NALU  97.62%  0.08  0.90  17  77.73%  18.22  1199.12  114303 
Appendix D Language To Number Translation Tasks
For the LSTM, we tried both summing the preceding states as output instead of using the final state and found this to be slightly advantageous to simply outputting the final state, as shown in Table 11. This summed state LSTM is what we use for comparison with the NAC and NALU in table 3. We train all models for 300K steps of gradient descent on the whole training set using Adam.
We selected the best model by validation loss over layer sizes , learning rates , and 10 initializations.
Model  Train MAE  Validation MAE  Test MAE 

LSTM w/ final state  0.0085  32.1  32.2 
LSTM w/ summed states  0.003  29.9  29.4 
Appendix E Program Evaluation
The addition task, which is the simpler variant of the two program evaluation tasks considered, is simple for all models to solve. However all models fail to generalize except for the recurrent NALU. The UGRNN proves much better than the LSTM and DNC, likely due to the simple linear update of the state.
Comments
There are no comments yet.