Research on RBM Accelerating Learning Algorithm with Weight Momentum

Fei LI,Xiao-Guang GAO,Kai-Fang WAN
DOI: https://doi.org/10.16383/j.aas.2017.c160325
2017-01-01
Abstract:Momentum algorithms can accelerate the training speed of restricted Boltzmann machine theoretically.Through a simulation study on existing momentum algorithms,it is found that existing momentum algorithms for training restricted Boltzmann machine have a poor accelerating effect and they began to lose acceleration performance.In the latter part of training process.Focusing on this problem,firstly,this paper gives a theoretical analysis of the algorithms based on Gibbs sampling convergence theorem.It is proved that the acceleration effect of existing momentum algorithms is at the expense of enlarging network weights.Then,a further investigation on network weights shows that the network weights contain a lot of information of the true gradient direction which can be used to train the network.According to this,a weight momentum algorithm is proposed based on the weight of the network.Finally,simulation results demon strate that the proposed algorithm has a better acceleration effect and has the accelerating ability even in the end of the training process.Therefore the proposed algorithm can well make up for the weaknesses of existing momentum algorithms.
What problem does this paper attempt to address?