Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Neural Power Units
(Submitted on 2 Jun 2020 (this version), latest version 17 Dec 2020 (v4))
Abstract: Conventional Neural Networks can approximate simple arithmetic operations, but fail to generalize beyond the range of numbers that were seen during training. Neural Arithmetic Units aim to overcome this difficulty, but current arithmetic units are either limited to operate on positive numbers or can only represent a subset of arithmetic operations. We introduce the Neural Power Unit (NPU) that operates on the full domain of real numbers and is capable of learning arbitrary power functions in a single layer. The NPU thus fixes the shortcomings of existing arithmetic units and extends their expressivity. This is achieved by internally using complex weights without requiring a conversion of the remaining network to complex numbers. We show where the NPU outperforms its competitors in terms of accuracy and sparsity on artificial arithmetic datasets. Additionally, we demonstrate how the NPU can be used as a highly interpretable model to discover the generating equation of a dynamical system purely from data.
Submission history
From: Niklas Heim [view email][v1] Tue, 2 Jun 2020 14:58:07 GMT (383kb,D)
[v2] Fri, 5 Jun 2020 12:23:45 GMT (358kb,D)
[v3] Fri, 16 Oct 2020 18:44:55 GMT (388kb,D)
[v4] Thu, 17 Dec 2020 14:40:50 GMT (381kb,D)
Link back to: arXiv, form interface, contact.