National Laboratory of Pattern Recognition, Institute of Automation, Chinese Academy of Sciences, Beijing, China.
National Laboratory of Pattern Recognition, Institute of Automation, Chinese Academy of Sciences, Beijing, China.
Neural Netw. 2023 Mar;160:192-201. doi: 10.1016/j.neunet.2023.01.004. Epub 2023 Jan 14.
Temporal knowledge prediction is a crucial task for early event warning, which has gained increasing attention recently. It aims to predict future facts based on relevant historical facts using temporal knowledge graphs. There are two main difficulties associated with the prediction task: from the perspective of historical facts, modeling the evolutionary patterns of facts to accurately predict the query and from the query perspective, handling the two cases where the query contains seen and unseen entities in a unified framework. Driven by these two problems, we propose a novel adaptive pseudo-Siamese policy network for temporal knowledge prediction based on reinforcement learning. Specifically, we design the policy network in our model as a pseudo-Siamese network consisting of two sub-policy networks. In the sub-policy network I, the agent searches for the answer to the query along the entity-relation paths to capture static evolutionary patterns. In sub-policy network II, the agent searches for the answer to the query along relation-time paths to deal with unseen entities. Moreover, we develop a temporal relation encoder to capture the temporal evolutionary patterns. Finally, we design a gating mechanism to adaptively integrate the results of the two sub-policy networks to help the agent focus on the destination answer. To assess the performance of our model, we conduct link prediction on four benchmark datasets, and extensive experimental results demonstrate that our method achieves considerable performance compared with existing methods.
时间知识预测是早期事件预警的关键任务,最近受到越来越多的关注。它旨在使用时间知识图根据相关历史事实预测未来事实。预测任务主要有两个困难:从历史事实的角度来看,建模事实的演化模式以准确预测查询;从查询的角度来看,在统一框架中处理查询中包含已见和未见实体的两种情况。受这两个问题的驱动,我们提出了一种基于强化学习的新的自适应伪孪生策略网络用于时间知识预测。具体来说,我们在模型中设计了策略网络作为一个由两个子策略网络组成的伪孪生网络。在子策略网络 I 中,代理沿着实体关系路径搜索查询的答案,以捕获静态演化模式。在子策略网络 II 中,代理沿着关系时间路径搜索查询的答案以处理未见实体。此外,我们开发了一个时间关系编码器来捕获时间演化模式。最后,我们设计了一个门控机制来自适应地整合两个子策略网络的结果,以帮助代理专注于目标答案。为了评估我们模型的性能,我们在四个基准数据集上进行了链接预测,大量实验结果表明,与现有方法相比,我们的方法取得了相当大的性能提升。