Title :
Training Restricted Boltzmann Machines with auxiliary function approach
Author :
Kameoka, Hirokazu ; Takamune, Norihiro
Author_Institution :
Grad. Sch. of Inf. Sci. & Technol., Univ. of Tokyo, Tokyo, Japan
Abstract :
Restricted Boltzmann Machines (RBMs) are neural network models for unsupervised learning, but have recently found a wide range of applications as feature extractors for supervised learning algorithms. They have also received a lot of attention recently after being proposed as building blocks for deep belief networks. The success of these models raises the issue of how best to train them. At present, the most popular training algorithm for RBMs is the Contrastive Divergence (CD) learning algorithm. The aim of this paper is to seek for a new optimization algorithm tailored for training RBMs in the hope of obtaining a faster algorithm than the CD algorithm. We propose deriving a new training algorithm for RBMs based on an auxiliary function approach. Through an experiment on parameter training of an RBM, we confirmed that the present algorithm converged faster and to a better solution than the CD algorithm.
Keywords :
Boltzmann machines; optimisation; unsupervised learning; CD learning algorithm; RBM; auxiliary function; contrastive divergence; faster algorithm; feature extractors; neural network models; optimization algorithm; restricted Boltzmann machines; supervised learning algorithms; training algorithm; unsupervised learning; Feature extraction; Linear programming; Mathematical model; Neural networks; Optimization; Signal processing algorithms; Training; Restricted Boltzmann machines; auxiliary function; contrastive divergence learning algorithm; deep belief networks; minorization-maximization;
Conference_Titel :
Machine Learning for Signal Processing (MLSP), 2014 IEEE International Workshop on
Conference_Location :
Reims
DOI :
10.1109/MLSP.2014.6958879