iNALU: Improved Neural Arithmetic Logic Unit

Daniel Schlör

doi:10.3389/frai.2020.00071

iNALU: Improved Neural Arithmetic Logic Unit

2020-09-29

5

OA

AI

DOI：10.3389/frai.2020.00071

原文链接

查看原文

摘要

En 中文

Neural networks have to capture mathematical relationships in order to learn various tasks. They approximate these relations implicitly and therefore often do not generalize well. The recently proposed Neural Arithmetic Logic Unit (NALU) is a novel neural architecture which is able to explicitly represent the mathematical relationships by the units of the network to learn operations such as summation, subtraction or multiplication. Although NALUs have been shown to perform well on various downstream tasks, an in-depth analysis reveals practical shortcomings by design, such as the inability to multiply or divide negative input values or training stability issues for deeper networks. We address these issues and propose an improved model architecture. We evaluate our model empirically in various settings from learning basic arithmetic operations to more complex functions. Our experiments indicate that our model solves stability issues and outperforms the original NALU model in means of arithmetic precision and convergence.

Keyword:

neural networks

machine learning

arithmetic calculations

neural architecture

experimental evaluation

iNALU: Improved Neural Arithmetic Logic Unit

摘要

AI总结

期刊

机构

引用论文