Neural Arithmetic Logic Units

08/01/2018
by   Andrew Trask, et al.
6

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.

READ FULL TEXT
research
06/02/2020

Neural Power Units

Conventional Neural Networks can approximate simple arithmetic operation...
research
04/14/2020

Exploring Cell counting with Neural Arithmetic Logic Units

The big problem for neural network models which are trained to count ins...
research
03/17/2020

iNALU: Improved Neural Arithmetic Logic Unit

Neural networks have to capture mathematical relationships in order to l...
research
11/10/2022

Improving the Robustness of Neural Multiplication Units with Reversible Stochasticity

Multilayer Perceptrons struggle to learn certain simple arithmetic tasks...
research
04/15/2020

Neural Status Registers

Neural networks excel at approximating functions and finding patterns in...
research
10/11/2021

Learning Division with Neural Arithmetic Logic Modules

To achieve systematic generalisation, it first makes sense to master sim...
research
10/04/2019

Measuring Arithmetic Extrapolation Performance

The Neural Arithmetic Logic Unit (NALU) is a neural network layer that c...

Please sign up or login with your details

Forgot password? Click here to reset