A Novel Restricted Boltzmann Machine Training Algorithm with Fast Gibbs Sampling Policy

Qianglong Wang, Xiaoguang Gao, Kaifang Wan, Fei Li, Zijian Hu

科研成果: 期刊稿件文章同行评审

14 引用 (Scopus)

摘要

The restricted Boltzmann machine (RBM) is one of the widely used basic models in the field of deep learning. Although many indexes are available for evaluating the advantages of RBM training algorithms, the classification accuracy is the most convincing index that can most effectively reflect its advantages. RBM training algorithms are sampling algorithms essentially based on Gibbs sampling. Studies focused on algorithmic improvements have mainly faced challenges in improving the classification accuracy of the RBM training algorithms. To address the above problem, in this paper, we propose a fast Gibbs sampling (FGS) algorithm to learn the RBM by adding accelerated weights and adjustment coefficient. An important link based on Gibbs sampling theory was established between the update of the network weights and mixing rate of Gibbs sampling chain. The proposed FGS method was used to accelerate the mixing rate of Gibbs sampling chain by adding accelerated weights and adjustment coefficients. To further validate the FGS method, numerous experiments were performed to facilitate comparisons with the classical RBM algorithm. The experiments involved learning the RBM based on standard data. The results showed that the proposed FGS method outperformed the CD, PCD, PT5, PT10, and DGS algorithms, particularly with respect to the handwriting database. The findings of our study suggest the potential applications of FGS to real-world problems and demonstrate that the proposed method can build an improved RBM for classification.

源语言英语
文章编号4206457
期刊Mathematical Problems in Engineering
2020
DOI
出版状态已出版 - 2020

指纹

探究 'A Novel Restricted Boltzmann Machine Training Algorithm with Fast Gibbs Sampling Policy' 的科研主题。它们共同构成独一无二的指纹。

引用此