基于图嵌入的关键词抽取方法OA
Graph embedding-based keyword extraction approach
关键词抽取是文本挖掘领域的热点研究主题之一,抽取准确性对于下游任务有显著影响.为了提升面向文本的关键词提取准确性,提出了一种基于图嵌入的关键词抽取(GeKWE)方法.该方法将预处理后的文档表示为候选词图,图中节点表示候选关键词,而边则表示候选词之间的共现关系强度.在此基础上,借鉴 Word2Vec 模型嵌入学习方法,以随机游走策略从图中获取采样节点序列而后训练 skip-gram 模型实现候选词图表征学习,得到候选词的语义向量.进一步地,通过对候选词先分簇后排序得到最终的关键词.在 2 个基准数据集 Inspec 和SemEval2010 上进行了实验验证,并与6 个领域代表性方法进行了对比分析,实验结果表明该文提出的方法在2 个数据集上都获得了最高的 F1 分数,分别为 61.26%和 42.73%,优于选取的对比方法,能有效缓解信息过载.
Keyword extraction is one of hot research topics in the field of text mining,and its accuracy has significant impact on downstream tasks.To enhance the accuracy of keyword extraction for test,this paper proposes a graph embedding-based keyword extraction(GeKWE)approach.A candidate word graph is built for a preprocessed document,where nodes denote candidate keywords while edges denote the strength of co-occurrence relationship between candidate words.On this basis,drawing on the embedding method of the Word2Vec model,this paper uses a random walk strategy to obtain sampling nodes sequences from the graph and then trains a skip-gram model to learn representations of candidate words graph,obtaining the semantic vectors of candidate words.Further,candidate words are divided into several clusters using the embedding representation.For each cluster,candidate words with a high score are chosen as the final keywords.Experiments on two benchmark datasets,Inspec and SemEval2010,are conducted and comparisons with six state-of-the-art approaches are made.Experimental results show that the proposed approach achieves the highest F1 score on both datasets,which are 61.26%and 42.73%respectively,better than the selected comparison approaches,and is able to effectively alleviate information overload.
管维亚;王球;李中烜;朱颀林;徐建
国网江苏省电力有限公司经济技术研究院,江苏 南京 210008国网江苏省电力有限公司经济技术研究院,江苏 南京 210008国网江苏省电力有限公司经济技术研究院,江苏 南京 210008南京理工大学 计算机科学与工程学院,江苏 南京 210094南京理工大学 计算机科学与工程学院,江苏 南京 210094
信息技术与安全科学
关键词抽取图嵌入预训练语言模型表示学习
keyword extractiongraph embeddingpre-trained language modelrepresentation learning
《南京理工大学学报(自然科学版)》 2026 (3)
304-311,8
国家自然科学基金(61872186)
评论