Enhanced gradient for training restricted Boltzmann machines

Kyunghyun Cho1, Tapani Raiko, Alexander Ilin

  • 1Department of Information and Computer Science, Aalto University School of Science, Espoo, Uusimaa 02150, Finland. kyunghyun.cho@aalto.fi

Neural Computation
|November 15, 2012
PubMed
Summary

Training Restricted Boltzmann machines (RBMs) is challenging due to sensitivity to hyperparameters and data representation. This study introduces an enhanced gradient invariant to bit-flipping, enabling more stable RBM training.