%0 Journal Article %A LIU Xinhui %A CHEN Wenshi %A ZHOU Ai %A CHEN Fei %A QU Wen %A LU Mingyu %T Multi-label Text Classification Based on Joint Model %D 2020 %R 10.3778/j.issn.1002-8331.1904-0273 %J Computer Engineering and Applications %P 111-117 %V 56 %N 14 %X

At present, the multi-label text classification algorithm ignores the importance of different words in text sequences and the influence of different levels of text features. This paper proposes an ATT-Capsule-BiLSTM method based on multi-head attention, CapsuleNet and the Bidirectional Long Short-Term Memory network(BiLSTM) model. Firstly, the text sequence is vectorized, and the weight distribution of the words is learned by multi-head attention on the basis of the word vector. Then the feature representation of the local spatial information and the context timing information are extracted by the Capsule network and BiLSTM respectively, and the fusion is performed through the fusion layer. After that, it is classified by the sigmoid classifier. The comparison experiments are carried out on two data sets, Reuters-21578 and AAPD. The experimental results show that the proposed joint model achieves better performance based on simple architecture. The [F1] values reach 89.82% and 67.48% respectively.

%U http://cea.ceaj.org/EN/10.3778/j.issn.1002-8331.1904-0273