【多标签文本分类】《融合注意力与CorNet的多标签文本分类》
创始人
2024-01-31 02:13:17
0

·阅读摘要:
  本文主要提出标签与文本注意力+文本注意力CorNet增强标签预测概率两个创新点,提升了实验精度。
·参考文献:
  [1] 融合注意力与CorNet的多标签文本分类

参考论文信息

  论文名称:《融合注意力与CorNet的多标签文本分类》

  发布期刊:《西北大学学报(自然科学版)》

  期刊信息:CSCD扩展
在这里插入图片描述

[0] 摘要

  目前文本分类存在问题:只关注文本本身的信息,忽略了标签的信息。

  为了解决这个问题:论文提出使用图注意力网络GAT来编码标签信息,然后使用 “文本-标签”注意力机制来强化标签与文本的语义联系,最后把它与文本的注意力向量融合,得到最终编入了标签信息的文本向量。

  论文最后使用CorNet模型增强标签预测概率。

[1] 相关工作

  论文介绍多标签文本分类算法的历史:机器学习算法和深度学习算法。

  机器学习算法有BR、CC、ML-DT、SVM、KNN等老算法;

  深度学习算法有CNN系列、RNN系列模型,比如CNN、LSTM、LSTM_Attention、Seq2Seq、SGM、CNN-RNN等。

【注一】:这里已经被各个论文介绍烂了,老生常谈。

  最后,还介绍了LSAN、GCN等会用到标签信息的模型。

【注二】:2022年的论文,竟然不介绍预训练语言模型,比如ELMo、BERT、XLNet这些。

[2] 模型

  模型图如下:
在这里插入图片描述
  该模型主要分为6个部分:

  1、嵌入层+LSTM层
  2、多标签注意力层
  3、图注意力网络
  4、“文本 -标签”注意力机制
  5、自适应融合机制
  6、CorNet模块

  1、嵌入层+LSTM层

  论文使用的是Glove预训练词向量作为embedding层,然后输入到Bi-LSTM层得到文本表示。

【注三】:使用embedding+Bi-LSTM是比较早的深度学习方法,现在更多的是用预训练语言模型来微调。

  2、多标签注意力层

  论文原文写的是:多标签文本可以由多个标签标记,每个标签对应的最相关文本是不同的。通过多标签注意力机制计算每个标签对应的文本向量hi的线性组合,可以有效地表示文本的重要程度。

  论文给的小标题“多标签注意力层”比较让人困惑,似乎这里用了标签的信息?看原文的意思是用了,但是看他贴的公式并没有用,而且我也想象不到怎么使用标签信息。

【注四】:这里我去论文贴出来的参考论文《A structured self-attentive sentence embedding》看了一下,人家写的是文本数据自注意,如下图:
在这里插入图片描述

  3、图注意力网络

  图注意力网络GAT是比较火的一种图神经网络,不再介绍。

  它的输入是标签嵌入,输出是经过注意力机制后的标签向量。

【注五】:GAT原文 《GRAPH ATTENTION NETWORKS》值得一看

  4、“文本 -标签”注意力机制

  强化标签之间的语义联系,将标签语义信息与文本上下文语义信息进行交互,获得基于标签语义的文本特征表示。

  论文的做法是,把 嵌入层+LSTM层 的输出与 图注意力网络 的输出相乘。

  5、自适应融合机制

  自适应融合机制是把 多标签注意力层 的输出与 **“文本 -标签”注意力机制 ** 的输出加权相乘,得到最终的文本表示。

在这里插入图片描述
  公式13有点疑惑,β\betaβ与γ\gammaγ不应该是公式12的输出嘛?怎么会相加等于1?用Softmax处理一下可以做好,但是论文没有提。

  6、CorNet模块

  CorNet模块能够学习标签相关性,使用相关性知识增强原始标签预测,并输出增强的标签预测。

【注六】:CorNet模型原文:《Correlation Networks for Extreme Multi-label Text Classification》

相关内容

热门资讯

中共中央纪委印发《关于做好20...   央视网消息:据中央纪委国家监委网站消息,12月24日,中共中央纪委印发《关于做好2026年元旦春...
2025年国际十大体育新闻   1.考文垂当选创造历史 国际奥委会迎来首任女性主席  当地时间3月20日,在希腊纳瓦里诺海岸举行...
一批重要法律草案提请审议回应“...   央视网消息:12月22日,一批重要法律草案提请全国人大常委会审议,涉及社会热点和民生关切等重要领...
协助追讨拖欠运费超12亿元 交...   今天(23日),国务院新闻办公室举行新闻发布会,交通运输部副部长李扬表示,货车司机常年以车为家、...
【理响中国·铿锵有理】加强重点...   12月10日至11日,中央经济工作会议在北京举行。会议总结2025年经济工作,分析当前经济形势,...
又一虎!西藏自治区人大常委会副...   西藏自治区人大常委会党组副书记、副主任王峻涉嫌严重违纪违法,目前正接受中央纪委国家监委纪律审查和...
关于治安管理处罚法第136条相...   修订后的治安管理处罚法将于2026年1月1日起施行。近期,一些网络媒体、网友等对治安管理处罚法第...
航行警告:南海部分海域进行火箭...   央视网消息:据中国海事局网站消息,阳江海事局发布航行警告,12月24日8时至26日16时,南海部...
泰国感谢中方   泰国外交部发言人尼康德23日说,泰方感谢中方为推动泰柬边境冲突缓局降温发挥的作用。  尼康德当天...
夫妻利用“占卜”诈骗274万余...   法治日报讯 记者周孝清 通讯员杜亚平 刘爱兰 近日,江西省新干县人民法院审结一起诈骗案,判处被告...