作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2022, Vol. 48 ›› Issue (4): 119-125. doi: 10.19678/j.issn.1000-3428.0060781

• 人工智能与模式识别 • 上一篇    下一篇

结合事件链与事理图谱的脚本事件预测模型

孙盼, 王琪, 万怀宇   

  1. 北京交通大学 计算机与信息技术学院 交通数据分析与挖掘北京市重点实验室, 北京 100044
  • 收稿日期:2021-02-02 修回日期:2021-03-30 发布日期:2021-04-02
  • 作者简介:孙盼(1996—),女,硕士研究生,主研方向为数据挖掘、信息抽取;王琪,硕士研究生;万怀宇(通信作者),副教授、博士。
  • 基金资助:
    国家重点研发计划(2018YFC0830200)。

Script Event Prediction Model Combining Event Chains and Event Evolutionary Graphs

SUN Pan, WANG Qi, WAN Huaiyu   

  1. Beijing Key Laboratory of Traffic Data Analysis and Mining, School of Computer and Information Technology, Beijing Jiaotong University, Beijing 100044, China
  • Received:2021-02-02 Revised:2021-03-30 Published:2021-04-02

摘要: 现有脚本事件预测模型在事件表示时未充分考虑各个元素之间的相关性,且不能同时利用事件链和事理图谱中的信息进行事件预测。针对事件表示不全面和信息融合不充分的问题,提出一种结合事件链和事理图谱的脚本事件预测模型ECGNet。将每个事件的各个元素构造成一个短句,使用Transformer编码器捕获元素之间的序列信息,从而获得更准确的事件表示。在此基础上,构建一个长程时序模块(LRTO)学习事件链中的时序信息,同时构建一个全局事件演化模块(GEEP)捕获隐藏在事理图谱中的演化模式,通过门控注意力机制动态融合时序信息和演化模式进行脚本事件预测。基于纽约时报和新浪新闻两个数据集的实验结果表明,ECGNet能够有效融合事件链和事理图谱的信息进行脚本事件预测,与PMI、Bigram、SAM-Net、SGNN等模型相比,其准确率较最优值取得了3%以上的提升。

关键词: 脚本事件预测, 事件表示, 事件链, 事理图谱, 注意力机制

Abstract: Given a sequence of events that have occurred, the goal of script event prediction is to infer what happens next.Previous approaches ignore the semantic correlations between the words describing the event and are thus unable to simultaneously consider the information in the event chains and event evolutionary graphs for event prediction.To solve the problem of incomplete event representation and insufficient fusion, this study proposes a script event prediction model combining event chains and event evolutionary graphs, namely ECGNet. First, the words describing an event are treated as a short sentence and the transformer encoder is utilized to capture the order between the words to obtain an accurate event representation.Specifically, a Long Range Temporal Orders(LRTO) module is designed to learn the strong order information in contextual event chains, and a Global Event Evolutionary Patterns(GEEP) module is constructed to capture the global patterns in event evolutionary graphs.Finally, a gated multi-attention mechanism is proposed to dynamically integrate the orders and evolutionary patterns to predict what happens next.Experimental results based on two datasets of New York Times and Sina News show that ECGNet can effectively integrate the information of event chains and event maps for script event prediction.Compared with PMI, Bigram, Semantic Attribute Matching Networks (SAM-Net), Self-Generating Neural Network(SGNN), and other models, the event prediction accuracy is improved by more than 3% compared with the optimal value.

Key words: script event prediction, event representation, event chain, event evolutionary graph, attention mechanism

中图分类号: