首页 | 本学科首页   官方微博 | 高级检索  
     

基于知识增强的图卷积神经网络的文本分类
引用本文:王婷,朱小飞,唐顾.基于知识增强的图卷积神经网络的文本分类[J].浙江大学学报(自然科学版 ),2022,56(2):322-328.
作者姓名:王婷  朱小飞  唐顾
作者单位:重庆理工大学 计算机科学与工程学院,重庆 400054
基金项目:国家自然科学基金资助项目(62141201);重庆市技术创新与应用发展专项项目(cstc2020jscx-dxwtBX0014);重庆市教委语言文字科研资助项目(yyk20103)
摘    要:针对文本分类问题,提出新的基于知识增强的图卷积神经网络(KEGCN)分类模型. KEGCN模型在整个文本集上构建了一个包含单词节点、文档节点、外部实体节点的文本图,不同类型节点之间使用不同的相似性计算方法;在文本图构建完成后将其输入到2层图卷积网络中学习节点的表示并进行分类. KEGCN模型引入外部知识进行构图,捕获长距离不连续的全局语义信息,是第1个将知识信息引入图卷积网络进行分类任务的工作. 在4个大规模真实数据集20NG、OHSUMED、R52、R8上进行文本分类实验,结果表明,KEGCN模型的分类准确率优于所有的基线模型. 将知识信息融入图卷积神经网络有利于学习到更精准的文本表示,提高文本分类的准确率.

关 键 词:知识嵌入  图卷积网络  神经网络  文本分类  自然语言处理  

Knowledge-enhanced graph convolutional neural networks for text classification
Ting WANG,Xiao-fei ZHU,Gu TANG.Knowledge-enhanced graph convolutional neural networks for text classification[J].Journal of Zhejiang University(Engineering Science),2022,56(2):322-328.
Authors:Ting WANG  Xiao-fei ZHU  Gu TANG
Abstract:A new knowledge-enhanced graph convolutional neural network (KEGCN) classification model was proposed aiming at the problem of text classification. In the KEGCN model, firstly a text graph containing word nodes, document nodes, and external entity nodes was constructed on the entire text set. Different similarity calculation methods were used between different types of nodes. After the text graph was constructed, it was input into the two-layer graph convolutional network to learn the representation of the node and classified. The KEGCN model introduced external knowledge to compose the graph, and captured the long-distance discontinuous global semantic information, and was the first work to introduce knowledge information into the graph convolution network for classification tasks. Text classification experiments were conducted on four large-scale real data sets, 20NG, OHSUMED, R52 and R8, and results showed that the classification accuracy of the KEGCN network model was better than that of all baseline models. Results show that integrating knowledge information into the graph convolutional neural network is conducive to learning more accurate text representations and improving the accuracy of text classification.
Keywords:knowledge embedding  graph convolutional network  neural network  text classification  natural language processing  
点击此处可从《浙江大学学报(自然科学版 )》浏览原始摘要信息
点击此处可从《浙江大学学报(自然科学版 )》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号