MPNet: temporal knowledge graph completion based on a multi-policy network

被引:2
作者
Wang, Jingbin [1 ]
Wu, Renfei [1 ]
Wu, Yuwei [1 ]
Zhang, Fuyuan [1 ]
Zhang, Sirui [1 ]
Guo, Kun [1 ]
机构
[1] Fuzhou Univ, Coll Comp & Data Sci, Fuzhou 350108, Fujian, Peoples R China
关键词
Knowledge graph completion; Reinforcement learning; Link prediction; Temporal knowledge graphs;
D O I
10.1007/s10489-024-05320-5
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Temporal knowledge graphs completion (TKGC) is a critical task that aims to forecast facts that will occur in future timestamps. It has attracted increasing research interest in recent years. Among the many approaches, reinforcement learning-based methods have gained attention due to their efficient performance and interpretability. However, these methods still face two challenges in the prediction task. First, a single policy network lacks the capability to capture the dynamic and static features of entities and relationships separately. Consequently, it fails to evaluate candidate actions comprehensively from multiple perspectives. Secondly, the composition of the action space is incomplete, often guiding the agent towards distant historical events and missing the answers in recent history. To address these challenges, this paper proposes a Temporal Knowledge Graph Completion Based on a Multi-Policy Network(MPNet). It constructs three policies from the aspects of static entity-relation, dynamic relationships, and dynamic entities, respectively, to evaluate candidate actions comprehensively. In addition, this paper creates a more diverse action space that guides the agent in investigating answers within historical subgraphs more effectively. The effectiveness of MPNet is validated through an extrapolation setting, and extensive experiments conducted on three benchmark datasets demonstrate the superior performance of MPNet compared to existing state-of-the-art methods.
引用
收藏
页码:2491 / 2507
页数:17
相关论文
共 33 条
[1]  
Bordes A., 2013, Advances in Neural Information Processing Systems
[2]  
Das Rajarshi, 2017, ARXIV
[3]  
García-Durán A, 2018, 2018 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING (EMNLP 2018), P4816
[4]  
Goel R, 2020, AAAI CONF ARTIF INTE, V34, P3988
[5]   A Survey on Knowledge Graph-Based Recommender Systems [J].
Guo, Qingyu ;
Zhuang, Fuzhen ;
Qin, Chuan ;
Zhu, Hengshu ;
Xie, Xing ;
Xiong, Hui ;
He, Qing .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2022, 34 (08) :3549-3568
[6]  
Han Z, 2020, PREPRINT
[7]  
Han Z, 2021, PREPRINT
[8]  
Han Z, 2020, GRAPH HAWKES NETWORK
[9]  
Han Zongbo, 2021, INT C LEARN REPR
[10]  
Jain P, 2020, PROCEEDINGS OF THE 2020 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING (EMNLP), P3733