融合标签嵌入和知识感知的多标签文本分类方法

作者：冯海; 马甲林^*; 许林杰; 杨宇; 谢乾

来源：南京大学学报(自然科学), 2023, 59(02): 273-281.

DOI：10.13232/j.cnki.jnju.2023.02.010

• 多标签文本分类
• 标签嵌入
• 知识图谱
• 注意力机制

多标签文本分类是自然语言处理领域的重要任务之一.文本的标签语义信息与文本的文档内容有紧密的联系，而传统的多标签文本分类方法存在忽略标签的语义信息以及标签的语义信息不足等问题.针对以上问题，提出一种融合标签嵌入和知识感知的多标签文本分类方法 LEKA (Label Embedding and Knowledge-Aware).该方法依赖于文档文本以及相应的多个标签，通过标签嵌入来获取与标签相关的注意力.考虑标签的语义信息，建立标签与文档内容的联系，将标签应用到文本分类中.另外，为了增强标签的语义信息，通过知识图谱嵌入引入外部感知知识，对标签文本进行语义扩展.在AAPD和RCV1-V2公开数据集上与其他分类模型进行了对比，实验结果表明，与LCFA (Label Combination and Fusion of Attentions)模型相比，LEKA的F1分别提高了3.5%和2.1%.

• 单位
  江苏卓易信息科技股份有限公司; 淮阴工学院