An Improved Convolutional Neural Network Algorithm for Multi-Label Classification
Xinsheng Wang, Lijun Sun, Zhihua Wei · 2018
Recent years conventional neural network(CNN) has been applied to different natural language processing(NLP) tasks such as sentence classification, sentence modeling, etc. Some researchers use CNN to do multi-label classification but their work mainly focus on image rather than text. In this paper, we propose an improved CNN via hierarchical dirichlet process(HDP) model to deal with the multi-label classification problem in NLP. We first apply an HDP model to discard some words which are less important semantically. Then we use word embedding methods to transform words to vectors. Finally, we train CNN based on word vectors. Experimental results demonstrate that our method is superior to most traditional multi-label classification methods and TextCNN in terms of performance.