一种新的分类受限玻尔兹曼机改进模型

Improved Model Based on Classification Restricted Boltzmann Machine

分类受限玻尔兹曼机(classification restricted boltzmann machine,ClassRBM)在各种分类问题中得到了广泛应用.ClassRBM是一种自带标签信息的神经网络模型,它使用一个神经元标识某类数据的类标.标签神经元总是稀疏的,一个神经元仅能为网络模型参数提供有限的信息.论文在ClassRBM现有的网络结构上,增加标签神经元个数,使每个类标用K个神经元标识,为网络模型参数提供更多的信息,提升模型表达能力,进而改善ClassRBM的分类性能.论文在不同数据集上进行了测试,结果表明改进模型的分类效果可以优于ClassRBM.

The classification restricted boltzmann machine（ ClassRBM） is widely used in various classification applications. It is a type of self-contained neural network model and use a simple neuron to identify a class of data in the label layer. Label neurons are often scarce,such a neuron can only provide limited information for the model parameters. In this paper,Multiple neurons are increased in the label layer on the basis of ClassRBM,and each class is identified by K neurons which provides more information for the model parameters.The improved model is used to improve the classification performance of ClassRBM. The experimental results show that the proposed model can be better than ClassRBM.