·阅读摘要:
本文在已有的SGM和BERT模型上改进,提出了SGM+BERT模型、混合模型。实验证明SGM+BERT模型收敛比BERT快很多,混合模型的效果最好。
·参考文献:
[1] BERT for Sequence-to-Sequence Multi-Label Text Classification
[2] SGM模型讲解,参考博客::【多标签文本分类】SGM: Sequence Generation Model for Multi-Label Classification
[3] Bert模型讲解,参考博客:【文本分类】BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding
本文是改进模型类型的论文,重点全在模型上,其他部分可以不看。
本文提出了SGM+BERT模型、混合模型一共两个模型。
[1] SGM+BERT模型
如下图,是论文中的SGM+BERT模型:
如下图,是SGM模型:
对比上述两个模型,可以看到,SGM+BERT
模型只是在文字转词向量的时候,使用了BERT来获取文本嵌入的语言表示。其他和SGM
模型是一模一样的。
论文中仅仅提到,SGM+BERT
模型比SGM
模型的好处在于,多标签BERT通常需要几十个epoch才能收敛,这与BERT+SGM模型不同,后者只需几百次迭代(不到半个epoch)就可以得到令人满意的结果。
SGM+BERT
模型的具体流程如下:
【注】:这一步叫做Mask softmax
,是在SGM那篇论文里的,感兴趣可以参考【多标签文本分类】SGM: Sequence Generation Model for Multi-Label Classification
· 6、softmax预测。
· 7、找出对应标签,添加到预测的标签集 中。
[2] 混合模型
首先明确,混合的是BERT模型
和SGM+BERT模型
。
SGM+BERT模型
比BERT模型
快,但是效果并不好,论文探究了把两者的输出混合的算法,实验发现效果更好。算法如下:
· 3、以L个标签来遍历,纵向从第2步的每个时间步中,找出最大的概率。
· 4、通过参数来把第1步和第3步的概率中和一下。
· 5、以第4步的概率为准,大于0.5的就算是该文本x xx有此标签。
【注】:这个混合模型看起来很不靠谱……