Parametric rectified nonlinear unit (PRenu) for convolution neural networks

• Published in: Signal, image and video processing image and video processing, 2021-03, Vol.15 (2), p.241-246
• Main Authors: El Jaafari, Ilyas, Ellahyani, Ayoub, Charfi, Said
• Format: Article
• Language: English
• Subjects: Artificial neural networks; Back propagation; Back propagation networks; Computer Imaging; Computer Science; Convergence; Convolution; Image Processing and Computer Vision; Multimedia Information Systems; Neural networks; Original Paper; Pattern Recognition and Graphics; Signal,Image and Speech Processing; Vision
• ISSN: 1863-1703; 1863-1711
• DOI: 10.1007/s11760-020-01746-9

Activation function unit is an extremely important part of convolution neural networks; it is the nonlinear transformation that we do over the input data. Using hidden layer incorporating with a well-chosen activation function improves both the accuracy and the CNN convergence speed. This paper proposes a parametric rectified nonlinear function unit (PRenu). The proposed activation function is nearly similar to Relu. It returns x - α log ( x + 1 ) for positive values ( α is between 0 and 1) and zero for negative parts. In contrast to Relu that returns the same received gradient for all positive values in its back-propagation, the PRenu multiplies it by values between 1 - α and 1 depending on the value with which each neuron was involved. The PRenu has been tested on three datasets: CIFAR-10, CIFAR-100 and Oxflower17, and compared to the activation function Relu. The experimental results show that using the proposed activation function PRenu, the CNN convergence is faster and the accuracy is also improved.