Rectifier (neural networks) explained

In the context of artificial neural networks, the rectifier or ReLU (rectified linear unit) activation function[1] [2] is an activation function defined as the non-negative part of its argument:

f(x)=x+=max(0,x)=

x+|x|
2

=\begin{cases} x&ifx>0,\\ 0&otherwise, \end{cases}

where

x

is the input to a neuron. This is also known as a ramp function and is analogous to half-wave rectification in electrical engineering. This activation function was introduced by Kunihiko Fukushima in 1969 in the context of visual feature extraction in hierarchical neural networks.[3] [4] [5] It was later argued that it has strong biological motivations and mathematical justifications.[6] [7] In 2011 it was found to enable better training of deeper networks,[8] compared to the widely used activation functions prior to 2011, e.g., the logistic sigmoid (which is inspired by probability theory; see logistic regression) and its more practical[9] counterpart, the hyperbolic tangent. The rectifier is,, the most popular activation function for deep neural networks.[10]

Rectified linear units find applications in computer vision[8] and speech recognition[11] [12] using deep neural nets and computational neuroscience.[13] [14] [15]

Advantages

max(0,ax)=amax(0,x)fora\geq0

.

Rectifying activation functions were used to separate specific excitation and unspecific inhibition in the neural abstraction pyramid, which was trained in a supervised way to learn several computer vision tasks.[16] In 2011,[8] the use of the rectifier as a non-linearity has been shown to enable training deep supervised neural networks without requiring unsupervised pre-training. Rectified linear units, compared to sigmoid function or similar activation functions, allow faster and effective training of deep neural architectures on large and complex datasets.

Potential problems

Variants

Piecewise-linear variants

Leaky ReLU

Leaky ReLUs allow a small, positive gradient when the unit is not active,[12] helping to mitigate the vanishing gradient problem.

f(x)=\begin{cases} x&ifx>0,\\ 0.01x&otherwise. \end{cases}       f'(x)=\begin{cases} 1&ifx>0,\\ 0.01&otherwise. \end{cases}

Parametric ReLU

Parametric ReLUs (PReLUs) take this idea further by making the coefficient of leakage into a parameter that is learned along with the other neural-network parameters.[17]

f(x)=\begin{cases} x&ifx>0,\\ ax&otherwise. \end{cases}          f'(x)=\begin{cases} 1&ifx>0,\\ a&otherwise. \end{cases}

Note that for a ≤ 1, this is equivalent to

f(x)=max(x,ax)

and thus has a relation to "maxout" networks.[17]

Other non-linear variants

Gaussian-error linear unit (GELU)

GELU is a smooth approximation to the rectifier:

f(x)=x\Phi(x),

f'(x)=x\Phi'(x)+\Phi(x),

where

\Phi(x)=P(X\leqslantx)

is the cumulative distribution function of the standard normal distribution.

This activation function is illustrated in the figure at the start of this article. It has a "bump" to the left of x < 0 and serves as the default activation for models such as BERT.[18]

SiLU

See main article: Swish function. The SiLU (sigmoid linear unit) or swish function is another smooth approximation, first coined in the GELU paper:[18]

f(x)=x\operatorname{sigmoid}(x),

f'(x)=x\operatorname{sigmoid}'(x)+\operatorname{sigmoid}(x),

where

\operatorname{sigmoid}(x)

is the sigmoid function.

Softplus

See main article: Softplus. A smooth approximation to the rectifier is the analytic function

f(x)=ln(1+ex),       f'(x)=

ex
1+ex

=

1
1+e-x

,

which is called the softplus[19] or SmoothReLU function.[20] For large negative

x

it is roughly

ln1

, so just above 0, while for large positive

x

it is roughly

ln(ex)

, so just above

x

.

This function can be approximated as:

ln\left(1+ex\right)\begin{cases}ln2,&x=0,\\[6pt]

x
1-e-x/ln2

,&x0\end{cases}

By making the change of variables

x=yln(2)

, this is equivalent to

log2(1+2y)\begin{cases}1,&y=0,\\[6pt]

y
1-e-y

,&y0.\end{cases}

A sharpness parameter

k

may be included:

f(x)=

ln(1+ekx)
k,

      f'(x)=

ekx
1+ekx

=

1
1+e-kx

.

The derivative of softplus is the logistic function.

The logistic sigmoid function is a smooth approximation of the derivative of the rectifier, the Heaviside step function.

The multivariable generalization of single-variable softplus is the LogSumExp with the first argument set to zero:

+(x
\operatorname{LSE
1,

...,xn):=\operatorname{LSE}(0,x1,...,xn)=ln(1+

x1
e

++

xn
e

).

The LogSumExp function is

\operatorname{LSE}(x1,...,xn)=

x1
ln(e

++

xn
e

),

and its gradient is the softmax; the softmax with the first argument set to zero is the multivariable generalization of the logistic function. Both LogSumExp and softmax are used in machine learning.

ELU

Exponential linear units try to make the mean activations closer to zero, which speeds up learning. It has been shown that ELUs can obtain higher classification accuracy than ReLUs.[21]

f(x)=\begin{cases}x&ifx>0,\ a\left(ex-1\right)&otherwise. \end{cases}       f'(x)=\begin{cases}1&ifx>0,\ aex&otherwise. \end{cases}

In these formulas,

a

is a hyper-parameter to be tuned with the constraint

a\geq0

.

The ELU can be viewed as a smoothed version of a shifted ReLU (SReLU), which has the form

f(x)=max(-a,x)

, given the same interpretation of

a

.

Mish

The mish function can also be used as a smooth approximation of the rectifier.[22] It is defined as

f(x)=x\tanh(\operatorname{softplus}(x)),

where

\tanh(x)

is the hyperbolic tangent, and

\operatorname{softplus}(x)

is the softplus function.

Mish is non-monotonic and self-gated.[23] It was inspired by Swish, itself a variant of ReLU.[23]

Squareplus

Squareplus[24] is the function

\operatorname{squareplus}b(x)=

x+\sqrt{x2+b
}where

b\geq0

is a hyperparameter that determines the "size" of the curved region near

x=0

. (For example, letting

b=0

yields ReLU, and letting

b=4

yields the metallic mean function.)Squareplus shares many properties with softplus: It is monotonic, strictly positive, approaches 0 as

x\to-infty

, approaches the identity as

x\to+infty

, and is

Cinfty

smooth. However, squareplus can be computed using only algebraic functions, making it well-suited for settings where computational resources or instruction sets are limited. Additionally, squareplus requires no special consideration to ensure numerical stability when

x

is large.

See also

Notes and References

  1. Web site: Brownlee . Jason . A Gentle Introduction to the Rectified Linear Unit (ReLU) . Machine Learning Mastery . 8 April 2021 . 8 January 2019.
  2. Web site: Liu . Danqing . A Practical Guide to ReLU . Medium . 8 April 2021 . en . 30 November 2017.
  3. K. . Fukushima . Visual feature extraction by a multilayered network of analog threshold elements . IEEE Transactions on Systems Science and Cybernetics . 5 . 4 . 1969 . 322–333 . 10.1109/TSSC.1969.300225.
  4. Book: K. . Fukushima . S. . Miyake . Neocognitron: A Self-Organizing Neural Network Model for a Mechanism of Visual Pattern Recognition . Competition and Cooperation in Neural Nets . In Competition and Cooperation in Neural Nets . Lecture Notes in Biomathematics . 1982 . 45 . Springer . 267–285 . 10.1007/978-3-642-46466-9_18 . 978-3-540-11574-8.
  5. Schmidhuber. Juergen. Juergen Schmidhuber. 2022. Annotated History of Modern AI and Deep Learning . cs.NE. 2212.11279.
  6. R. . Hahnloser . R. . Sarpeshkar . M. A. . Mahowald . R. J. . Douglas . H. S. . Seung . Digital selection and analogue amplification coexist in a cortex-inspired silicon circuit . . 405 . 6789. 2000 . 947–951 . 10.1038/35016072 . 10879535 . 2000Natur.405..947H . 4399014 .
  7. R. . Hahnloser . H. S. . Seung . 2001 . Permitted and Forbidden Sets in Symmetric Threshold-Linear Networks . NIPS 2001.
  8. Xavier Glorot . Antoine Bordes . . 2011 . Deep sparse rectifier neural networks . AISTATS . Rectifier and softplus activation functions. The second one is a smooth version of the first. .
  9. Encyclopedia: . . Genevieve B. Orr . . 1998 . Efficient BackProp . G. Orr . K. Müller . Neural Networks: Tricks of the Trade . Springer.
  10. Ramachandran . Prajit . Barret . Zoph . Quoc . V. Le . October 16, 2017 . Searching for Activation Functions . 1710.05941 . cs.NE.
  11. László Tóth . 2013 . Phone Recognition with Deep Sparse Rectifier Neural Networks . .
  12. Andrew L. Maas, Awni Y. Hannun, Andrew Y. Ng (2014). Rectifier Nonlinearities Improve Neural Network Acoustic Models.
  13. D. . Hansel . C. . van Vreeswijk . How noise contributes to contrast invariance of orientation tuning in cat visual cortex . . 22 . 12. 2002 . 5118–5128 . 10.1523/JNEUROSCI.22-12-05118.2002 . 12077207 . 6757721 .
  14. 10.1103/PhysRevX.5.041030 . 5 . 4 . 041030 . Kadmon . Jonathan . Sompolinsky . Haim . Transition to Chaos in Random Neuronal Networks . Physical Review X . 2015-11-19 . 1508.06486 . 2015PhRvX...5d1030K . 7813832.
  15. Engelken . Rainer . Wolf . Fred . Abbott . L. F. . Lyapunov spectra of chaotic recurrent neural networks . 2020-06-03 . nlin.CD . 2006.02427.
  16. Book: Behnke, Sven . 2003 . Hierarchical Neural Networks for Image Interpretation . Lecture Notes in Computer Science . 2766 . Springer . 10.1007/b11963. 978-3-540-40722-5 . 1304548 .
  17. 1502.01852 . He . Kaiming . Delving Deep into Rectifiers: Surpassing Human-Level Performance on Image Net Classification . Zhang . Xiangyu . Ren . Shaoqing . Sun . Jian . cs.CV . 2015.
  18. 1606.08415 . Gaussian Error Linear Units (GELUs) . Hendrycks . Dan . Gimpel . Kevin . 2016 . cs.LG.
  19. Dugas . Charles . Bengio . Yoshua . Bélisle . François . Nadeau . Claude . Garcia . René . 2000-01-01 . Incorporating second-order functional knowledge for better option pricing . Proceedings of the 13th International Conference on Neural Information Processing Systems (NIPS'00) . MIT Press . 451–457 . Since the sigmoid h has a positive first derivative, its primitive, which we call softplus, is convex. .
  20. Web site: Smooth Rectifier Linear Unit (SmoothReLU) Forward Layer . 2017 . Developer Guide for Intel Data Analytics Acceleration Library . en-US . 2018-12-04.
  21. 1511.07289 . Clevert . Djork-Arné . Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs) . Unterthiner . Thomas . Hochreiter . Sepp . cs.LG . 2015.
  22. .
  23. Web site: Shaw . Sweta . 2020-05-10 . Activation Functions Compared with Experiments . 2022-07-11 . W&B . en.
  24. Barron . Jonathan T. . 2112.11687 . Squareplus: A Softplus-Like Algebraic Rectifier . cs.NE . 22 December 2021.