Rational neural networks

Nicolas Boullé, Yuji Nakatsukasa, Alex Townsend
4/4/2020
cs.NEcs.LGmath.NA

Official code: NBoulle/RationalNets (31 stars, MIT, last commit 2021-02-15).

Abstract

We consider neural networks with rational activation functions. The choice of the nonlinear activation function in deep learning architectures is crucial and heavily impacts the performance of a neural network. We establish optimal bounds in terms of network complexity and prove that rational neural networks approximate smooth functions more efficiently than ReLU networks with exponentially smaller depth. The flexibility and smoothness of rational activation functions make them an attractive alternative to ReLU, as we demonstrate with numerical experiments.

View on arXivView PDF

Code Implementations(1)

MIT

Code for the paper "Rational neural networks", NeurIPS 2020

315MATLAB, PythonMar 27, 20205 years agoMIT
approximation-theorydeep-learningrational-functionsrational-neural-networks

Cite this paper

@article{boull2020rational,
  title  = {Rational neural networks},
  author = {Nicolas Boullé and Yuji Nakatsukasa and Alex Townsend},
  year   = {2020},
  eprint = {2004.01902},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2004.01902}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2004.01902.svg)](https://codeofpaper.com/paper/2004.01902)

Discussion