基于事件演化图的多标记事件预测模型

doi:10.19678/j.issn.1000-3428.0067498

摘要/Abstract

摘要： 多标记事件预测是指预测多个相关联的事件是否会在未来发生,相比传统单标记事件预测,需要同时预测多个目标事件。现有的事件预测研究忽略各领域存在的多标记事件情境,且对多标记事件预测研究较少。提出一种基于事件演化图的多标记事件预测模型(MLEP),以实现基于事件演化图(EEG)的多标记事件预测研究模式。首先基于事件链构建事件演化图;然后对多标记事件预测问题进行问题转换,将多标记问题转化为单标记问题,利用事件表示学习方法获取所有事件的向量表示,对多标记事件进行编码;最后采用门控图神经网络(GGNN)框架构建多标记事件预测模型,根据相似度匹配出最优的后续事件,实现多标记事件的预测。在真实数据集上的实验结果表明,MLEP模型可以有效地预测出多标记事件,预测准确率达到了65.58%,性能优于大多现有的基准模型,提升幅度达到了4.94%以上。通过消融实验也证明了更好的事件表示学习方法对事件具有较好的表示效果,提升多标记事件预测的性能。

关键词: 多标记, 事件演化图, 事件表示学习, 门控图神经网络, 事件预测

Abstract: Multilabel event prediction refers to the prediction of whether multiple associated events will occur in the future, which requires the simultaneous prediction of multiple target events and comparing it with the conventional single-label event prediction. Because the issue of multi-label event contexts in various fields is yet to be addressed and studies regarding multi-label event prediction are few, this paper proposes a Multi-Label Event Prediction(MLEP) model based on Event-Evolution Graph(EEG). First, an EEG is constructed based on event chains. Subsequently, problem transformation is performed on the multi-label event-prediction problem to transform it into a single-label problem, followed by obtaining vector representations of all events using event-representation learning methods to encode multi-label events. Finally, a multi-label event prediction model is constructed using the Gated Graph Neural Network(GGNN) framework. The optimal subsequent events are matched based on their similarity to predict multi-label events. Experimental results on real datasets show that the proposed MLEP model can effectively predict multi-labeled events with a prediction accuracy of 65.58%, thus outperforming most existing benchmark models with an improvement level exceeding 4.94%. Results of ablation experiments show that better event-representation learning methods provide better event representations and multi-label event predictions.

Key words: multi label, Event-Evolution Graph(EEG), event representation learning, Gated Graph Neural Network(GGNN), event-prediction

中图分类号:

TP391

王华珍, 许泽, 孙悦, 丘斌, 陈坚, 邱强斌. 基于事件演化图的多标记事件预测模型[J]. 计算机工程, 2024, 50(4): 132-140.

WANG Huazhen, XU Ze, SUN Yue, QIU Bin, CHEN Jian, QIU Qiangbin. Multi-Label Event-Prediction Model Based on Event-Evolution Graph[J]. Computer Engineering, 2024, 50(4): 132-140.

https://www.ecice06.com/CN/Y2024/V50/I4/132

参考文献

[1] 刘玉婷, 刘茗, 王保卫, 等. 脚本事件预测:方法、评测与挑战[J]. 计算机应用研究, 2023, 40(5):1303-1311. LIU Y T, LIU M, WANG B W, et al. Script event prediction:methods, evaluation and challenges[J]. Application Research of Computers, 2023, 40(5):1303-1311. (in Chinese)
[2] CHAMBERS N, JURAFSKY D. Unsupervised learning of narrative event chains[C]//Proceedings of the 46th Annual Meeting of the Association for Computational Linguistics. Washington D. C.,USA:IEEE Press,2008:789-797.
[3] JANS B, BETHARD S, VULĺC I, et al. Skip n-grams and ranking functions for predicting script events[C]//Proceedings of the 25th International Conference on Neural Information Processing Systems. Washington D. C., USA:IEEE Press, 2012:336-344.
[4] PICHOTTA K, MOONEY R. Statistical script learning with multi-argument events[C]//Proceedings of the 14th Conference of the European Chapter of the Association for Computational Linguistics. Berlin, Germany:Springer, 2014:220-229.
[5] PICHOTTA K, MOONEY R J. Learning statistical scripts with LSTM recurrent neural networks[C]//Proceedings of the 30th AAAI Conference on Artificial Intelligence. New York, USA:ACM Press, 2016:2800-2806.
[6] GLAVAŠ G, ŠNAJDER J. Construction and evaluation of event graphs[J]. Natural Language Engineering, 2015, 21(4):607-652.
[7] MIKOLOV T, SUTSKEVER I, CHEN K, et al. Distributed representations of words and phrases and their compositionality[C]//Proceedings of the 26th International Conference on Neural Information Processing Systems. New York, USA:ACM Press, 2013:3111-3119.
[8] GRANROTH-WILDING M, CLARK S. What happens next? Event prediction using a compositional neural network model[C]//Proceedings of the 30th AAAI Conference on Artificial Intelligence. New York, USA:ACM Press, 2016:2727-2733.
[9] WANG Z Q, ZHANG Y, CHANG C Y. Integrating order information and event relation for script event prediction[C]//Proceedings of 2017 Conference on Empirical Methods in Natural Language Processing. Stroudsburg, USA:Association for Computational Linguistics, 2017:57-67.
[10] LÜ S W, QIAN W H, HUANG L T, et al. SAM-Net:integrating event-level and chain-level attentions to predict what happens next[C]//Proceedings of AAAI Conference on Artificial Intelligence.[S. 1.]:AAAI Press, 2019:6802-6809.
[11] LI Z Y, ZHAO S D, DING X, et al. EEG:knowledge base for event evolutionary principles and patterns[M]. Berlin, Germany:Springer, 2017.
[12] 黄承宁, 李娟, 陈嘉政. 基于图神经网络的医疗物资智能调度研究优化[J]. 计算机技术与发展, 2021, 31(9):202-207. HUANG C N, LI J, CHEN J Z. Research and optimization of medical material intelligent scheduling based on graph neural network[J]. Computer Technology and Development, 2021, 31(9):202-207.(in Chinese)
[13] 王健宗, 孔令炜, 黄章成, 等. 图神经网络综述[J]. 计算机工程, 2021, 47(4):1-12. WANG J Z, KONG L W, HUANG Z C, et al. Survey of graph neural network[J]. Computer Engineering, 2021, 47(4):1-12.(in Chinese)
[14] ZHAO L. Event prediction in the big data era:a systematic survey[J]. ACM Computing Surveys, 2021,54(5):94.
[15] ZHAO S D. Mining medical causality for diagnosis assistance[C]//Proceedings of the 10th ACM International Conference on Web Search and Data Mining. New York, USA:ACM Press, 2017:841.
[16] 孙盼, 王琪, 万怀宇. 结合事件链与事理图谱的脚本事件预测模型[J]. 计算机工程, 2022, 48(4):119-125. SUN P, WANG Q, WAN H Y. Script event prediction model combining event chains and event evolutionary graphs[J]. Computer Engineering, 2022, 48(4):119-125.(in Chinese)
[17] SCHAPIRE R E, SINGER Y. BoosTexter:a boosting-based system for text categorization[J]. Machine Learning, 2000, 39(2):135-168.
[18] 李志欣, 卓亚琦, 张灿龙, 等. 多标记学习研究综述[J]. 计算机应用研究, 2014, 31(6):1601-1605. LI Z X, ZHUO Y Q, ZHANG C L, et al. Survey on multi-label learning[J]. Application Research of Computers, 2014, 31(6):1601-1605.(in Chinese)
[19] ZHANG M L, ZHOU Z H. A review on multi-label learning algorithms[J]. IEEE Transactions on Knowledge and Data Engineering, 2014, 26(8):1819-1837.
[20] 王晓莹, 谢钧, 陶性留, 等. 基于嵌入式特征提取的多标记分类算法[J]. 计算机工程, 2019, 45(11):172-176. WANG X Y, XIE J, TAO X L, et al. Multi-label classification algorithm based on embedded feature extraction[J]. Computer Engineering, 2019, 45(11):172-176.(in Chinese)
[21] ZHANG M L, ZHOU Z H. ML-KNN:a lazy learning approach to multi-label learning[J]. Pattern Recognition, 2007, 40(7):2038-2048.
[22] ELISSEEFF A, WESTON J. A kernel method for multi-labelled classification[C]//Proceedings of NIPS'02. Cambridge, USA:MIT Press, 2002:681-688.
[23] BOUTELL M R, LUO J B, SHEN X P, et al. Learning multi-label scene classification[J]. Pattern Recognition, 2004, 37(9):1757-1771.
[24] READ J, PFAHRINGER B, HOLMES G. Multi-label classification using ensembles of pruned sets[C]//Proceedings of the 8th International Conference on Data Mining. New York, USA:ACM Press, 2008:995-1000.
[25] 高振华. 基于标记间相关性的多标记分类算法[D]. 长沙:中南大学, 2013. GAO Z H. Multi-label classification algorithm based on labels'correlation[D].Changsha:Central South University, 2013. (in Chinese)
[26] LI H, YAN G Y, ZHAO X, et al. Tactical mission event logic graph construction for network-centric warfare[J]. Alexandria Engineering Journal, 2022, 61(11):9161-9173.
[27] XIE R B, LIU Z Y, SUN M S. Representation learning of knowledge graphs with hierarchical types[C]//Proceedings of the 25th International Joint Conference on Artificial Intelligence. New York, USA:ACM Press, 2016:2965-2971.
[28] DEVLIN J, CHANG M W, LEE K, et al. BERT:pre-training of deep bidirectional transformers for language understanding[C]//Proceedings of Conference of the North American Chapter of the ACL:Human Language Technologies. Washington D. C.,USA:IEEE Press,2019:4171-4186.
[29] LIN Y K, LIU Z Y, LUAN H B, et al. Modeling relation paths for representation learning of knowledge bases[C]//Proceedings of 2015 Conference on Empirical Methods in Natural Language Processing. Stroudsburg, USA:Association for Computational Linguistics, 2015:705-714.
[30] LI Z Y, DING X, LIU T. Constructing narrative event evolutionary graph for script event prediction[C]//Proceedings of the 27th International Joint Conference on Artificial Intelligence. New York, USA:ACM Press, 2018:4201-4207.
[31] 杨顶, 邓明君, 徐丽萍. 基于时空信息融合学习的路段行程车速短时预测[J]. 计算机工程, 2021, 47(12):78-86. YANG D, DENG M J, XU L P. Short-term prediction of road travel speed based on spatio-temporal information fusion learning[J]. Computer Engineering, 2021, 47(12):78-86.(in Chinese)
[32] SHEN Z R, ZHANG M Y, ZHAO H Y, et al. Efficient attention:attention with linear complexities[C]//Proceedings of IEEE Winter Conference on Applications of Computer Vision. Waikoloa, USA:IEEE Press, 2021:3531-3539.
[33] LIU X L, JIN J H, WANG Q, et al. PatientEG dataset:bringing event graph model with temporal relations to electronic medical records[EB/OL].[2023-03-10]. http://arxiv.org/pdf/1812.09905.

选择文件类型/文献管理软件名称

选择包含的内容