Frontiers in Neuroscience (Apr 2024)

An analytical approach for unsupervised learning rate estimation using rectified linear units

  • Chaoxiang Chen,
  • Chaoxiang Chen,
  • Chaoxiang Chen,
  • Vladimir Golovko,
  • Vladimir Golovko,
  • Aliaksandr Kroshchanka,
  • Egor Mikhno,
  • Marta Chodyka,
  • Piotr Lichograj

DOI
https://doi.org/10.3389/fnins.2024.1362510
Journal volume & issue
Vol. 18

Abstract

Read online

Unsupervised learning based on restricted Boltzmann machine or autoencoders has become an important research domain in the area of neural networks. In this paper mathematical expressions to adaptive learning step calculation for RBM with ReLU transfer function are proposed. As a result, we can automatically estimate the step size that minimizes the loss function of the neural network and correspondingly update the learning step in every iteration. We give a theoretical justification for the proposed adaptive learning rate approach, which is based on the steepest descent method. The proposed technique for adaptive learning rate estimation is compared with the existing constant step and Adam methods in terms of generalization ability and loss function. We demonstrate that the proposed approach provides better performance.

Keywords