Unsupervised Pre-training Classifier Based on Restricted Boltzmann Machine with Imbalanced Data

详细信息    查看全文

• 关键词：Semi ; supervised learning ; Classification ; Deep learning ; Restricted boltzmann machine ; Deep neural network
• 刊名：Lecture Notes in Computer Science
• 出版年：2017
• 出版时间：2017
• 年：2017
• 卷：10135
• 期：1
• 页码：102-110
• 丛书名：Smart Computing and Communication
• ISBN：978-3-319-52015-5
• 卷排序：10135

文摘

Many learning algorithms can suffer from a performance bias for classification with imbalanced data. This paper proposes the pre-training the deep structure neural network by restricted Boltzmann machine (RBM) learning algorithm, which is pre-sampled with standard SMOTE methods for imbalanced data classification. Firstly, a new training data set can be generated by a pre-sampling method from original examples; secondly the deep neural network structure is trained on the sampled data and all unlabelled data sets by RBM greedy algorithm, which is called “coarse tuning”. Then the neural networks are fined tuned by BP algorithm. The effectiveness of the RBM pre-training neural network (RBMPT) classifier is demonstrated on a number of benchmark data sets. Compared with only BP classifier, pre-sampling BP classifier and RBMPT classifier, it has shown that pre-training procedure can learn more representations of data better with unlabelled data and has better classification performance for classification with imbalanced data sets.