The optimization of parallel convolutional RBM based on Spark

Author(s):  
Yun Jiang ◽  
Junyu Zhuo ◽  
Juan Zhang ◽  
Xiao Xiao

With the extensive attention and research of the scholars in deep learning, the convolutional restricted Boltzmann machine (CRBM) model based on restricted Boltzmann machine (RBM) is widely used in image recognition, speech recognition, etc. However, time consuming training still seems to be an unneglectable issue. To solve this problem, this paper mainly uses optimized parallel CRBM based on Spark, and proposes a parallel comparison divergence algorithm based on Spark and uses it to train the CRBM model to improve the training speed. The experiments show that the method is faster than traditional sequential algorithm. We train the CRBM with the method and apply it to breast X-ray image classification. The experiments show that it can improve the precision and the speed of training compared with traditional algorithm.

2021 ◽  
Author(s):  
Hamid Hassanpour

In this article, State-of-the-art deep learning models are evaluated and their performances in X-ray image classification is reported.


Author(s):  
Christine Dewi ◽  
Rung-Ching Chen ◽  
Hendry ◽  
Hsiu-Te Hung

Restricted Boltzmann machine (RBM) plays an important role in current deep learning techniques, as most of the existing deep networks are based on or related to generative models and image classification. Many applications for RBMs have been developed for a large variety of learning problems. Recent developments have demonstrated the capacity of RBM to be powerful generative models, able to extract useful features from input data or construct deep artificial neural networks. In this work, we propose a learning algorithm to find the optimal model complexity for the RBM by improving the hidden layer (50–750 layers). Then, we compare and analyze the classification performance in depth of regular RBM use RBM () function, classification RBM use stackRBM() function, and Deep Belief Network (DBN) use DBN() function with the different hidden layer. As a result, Stacking RBM and DBN could improve our classification performance compared to regular RBM.


Sign in / Sign up

Export Citation Format

Share Document