IEEE Access (Jan 2020)

Development and Application of Matrix Variate Restricted Boltzmann Machine

  • Guanglei Qi,
  • Yi Sun,
  • Mingang Li,
  • Xiaogang Hou

DOI
https://doi.org/10.1109/ACCESS.2020.3012603
Journal volume & issue
Vol. 8
pp. 137856 – 137866

Abstract

Read online

Matrix Variate Restricted Boltzmann Machine (MVRBM) model could maintain valuable spatial information using much fewer model parameters than the classic RBM, thus it has been used in image analysis, pattern recognition and machine learning for data analysis. This paper introduces two extensions of MVRBM: Matrix Variate Deep Belief Network (MVDBN) and Multimodal MVRBM (MMVRBM), both of them are composed of two or more MVRBMs whose input and latent variables are in matrix form. The MVDBN and MMVRBM have much fewer model parameters and deeper layer with better features to avoid overfitting and more accurate model easier to be learned. We test their efficiency on two real-world applications: image super-resolution and handwritten digit recognition. The experiments show the feasibility and superiority of the proposed models.

Keywords