Abstract
Temporal knowledge graphs (TKGs) extend traditional knowledge graphs by incorporating temporal information to represent valid time. TKG reasoning involves inferring and predicting missing or future facts within various time intervals, and applications in critical domains include political event prediction and financial analysis. Current TKG reasoning methods rely on path-based or subgraph-based strategies to achieve enhanced interpretability. However, they mostly focus on entity interaction information within paths or subgraphs, and they fail to exploit the semantic and temporal information of relations during reasoning. To comprehensively utilize such information for refining the reasoning process, we propose a subgraph reasoning model based on time-aware relation representation (TiAR) for TKGs. TiAR learns the subgraph structures around query nodes and reasons on facts happening in future timestamps. It encodes the relation features by combining the temporal displacement values between events and extracts the semantic correlations between query and adjacency relations via an attention-based neighborhood feature aggregation module. TiAR extracts a query subgraph of interest by iteratively sampling and diffusing attention through the whole graph. In particular, TiAR includes a path constraint module to further capture the semantic correlations between the structural features of queries and subgraphs. Our evaluation conducted on four benchmark datasets demonstrates that TiAR outperforms the current state-of-the-art models, indicating its effectiveness in achieving improved TKG reasoning. In particular, for the two large ICEWS05-15/18 datasets, our model yields better performance than the latest model. Moreover, The development of TKG reasoning methods is essential for improving the reliability and interpretability of models in such domains, and our work contributes to this area of research.
Similar content being viewed by others
Availability of data and materials
Data will be made public on publication.
Code Availability
Code will be made public on publication.
References
Wang X, Wang D, Xu C, He X, Cao Y, Chua T-S (2019) Explainable reasoning over knowledge graphs for recommendation. Proceedings of the AAAI Conference on Artificial Intelligence 33:5329–5336
Sun J, Zhang Y (2022) Recommendation system with biclustering. Big Data Mining and Analytics 5(4):282–293. https://doi.org/10.26599/BDMA.2022.9020012
Liu Z, Xiong C, Sun M, Liu Z (2018) Entity-duet neural ranking: Understanding the role of knowledge graph semantics in neural information retrieval. In: Proceedings of the 56th Annual Meeting of the Association for Computational Linguistics (Volume 1:Long Papers), pp 2395–2405
Rahman MS, Reza H (2022) A systematic review towards big data analytics in social media. Big Data Mining and Analytics 5(3):228–244. https://doi.org/10.26599/BDMA.2022.9020009
Zhang Y, Dai H, Kozareva Z, Smola AJ, Song L (2018) Variational reasoning for question answering with knowledge graph. In: Thirty-second AAAI Conference on Artificial Intelligence
Zhang J, Zhang L, Hui B, Tian L (2022) Improving complex knowledge base question answering via structural information learning. Knowledge-Based Systems 242:108252
Jung J, Jung J, Kang U (2020) T-gap: Learning to walk across time for temporal knowledge graph completion. ArXiv arXiv:2012.10595
Zheng X, Cai Z (2020) Privacy-preserved data sharing towards multiple parties in industrial iots. IEEE Journal on Selected Areas in Communications 38(5):968–979. https://doi.org/10.1109/JSAC.2020.2980802
Bollen J, Mao H, Zeng X (2011) Twitter mood predicts the stock market. J Comput Sci 2(1):1–8. https://doi.org/10.1016/j.jocs.2010.12.007
Dasgupta SS, Ray SN, Talukdar P (2018) Hyte: Hyperplane-based temporally aware knowledge graph embedding. In: Proceedings of the 2018 Conference on Empirical Methods in Natural Language Processing, pp 2001–2011
Lacroix T, Obozinski G, Usunier N (2020) Tensor decompositions for temporal knowledge base completion. In: International Conference on Learning Representations . https://openreview.net/forum?id=rke2P1BFwS
Trivedi R, Dai H, Wang Y, Song L (2017) Know-evolve: Deep temporal reasoning for dynamic knowledge graphs. In: International Conference on Machine Learning, pp 3462–3471. PMLR
Das R, Dhuliawala, S, Zaheer M, Vilnis L, Durugkar I, Krishnamurthy A, Smola A, McCallum A (2018) Go for a walk and arrive at the answer: Reasoning over paths in knowledge bases using reinforcement learning. In: International Conference on Learning Representations. https://openreview.net/forum?id=Syg-YfWCW
Lin XV, Socher R, Xiong C (2018) Multi-hop knowledge graph reasoning with reward shaping. In: Proceedings of the 2018 Conference on Empirical Methods in Natural Language Processing, EMNLP 2018, Brussels,Belgium, October 31-November 4, 2018
Yang Y, Song L (2020) Learn to explain efficiently via neural logic inductive learning. In: International Conference on Learning Representations. https://openreview.net/forum?id=SJlh8CEYDB
Xu X, Feng W, Jiang Y, Xie X, Sun Z, Deng Z-H (2020) Dynamically pruned message passing networks for large-scale knowledge graph reasoning. In: International Conference on Learning Representations. https://openreview.net/forum?id=rkeuAhVKvB
Du Z, Zhou C, Yao J, Tu T, Cheng L, Yang H, Zhou J, Tang J (2021) Cogkr: Cognitive graph for multi-hop knowledge reasoning. IEEE Trans Knowl Data Eng 01:1–1. https://doi.org/10.1109/TKDE.2021.3104310
Han Z, Chen P, Ma Y, Tresp V (2021) Explainable subgraph reasoning for forecasting on temporal knowledge graphs. In: International Conference on Learning Representations
Bordes A, Usunier N, Garcia-Duran A, Weston J, Yakhnenko O (2013) Translating embeddings for modeling multi-relational data. In: Neural Information Processing Systems
Yang B, Yih W, He X, Gao J, Deng L (2015) Embedding entities and relations for learning and inference in knowledge bases. In: International Conference on Learning Representations
Dettmers T, Minervini P, Stenetorp P, Riedel S (2018) Convolutional 2d knowledge graph embeddings. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 32
Schlichtkrull M, Kipf TN, Bloem P, Berg Rvd, Titov I, Welling M (2018) Modeling relational data with graph convolutional networks. In: European Semantic Web Conference. Springer, pp 593–607
García-Durán A, Dumancic S, Niepert M (2018) Learning sequence encoders for temporal knowledge graph completion. In: EMNLP
Leblay J, Chekol MW (2018) Deriving validity time in knowledge graph. Companion Proceedings of the the Web Conference 2018:1771–1776
Goel R, Kazemi SM, Brubaker M, Poupart P (2020) Diachronic embedding for temporal knowledge graph completion. Proceedings of the AAAI Conference on Artificial Intelligence 34:3988-3995
Duan Y, Wang J, Ma H, Sun Y (2022) Residual convolutional graph neural network with subgraph attention pooling. Tsinghua Sci Technol 27(4):653–663. https://doi.org/10.26599/TST.2021.9010058
Chen Q, Cai Z, Cheng L, Gao H (2022) Structure-free general data aggregation scheduling for multihop battery-free wireless networks. IEEE Trans Mob Comput 21(9):3342–3359. https://doi.org/10.1109/TMC.2021.3053557
Zhang J, Liang S, Deng Z, Shao J (2021) Spatial-temporal attention network for temporal knowledge graph completion. In: International Conference on Database Systems for Advanced Applications
Sun H, Zhong J, Ma Y, Han Z, He K (2021) Timetraveler: Reinforcement learning for temporal knowledge graph forecasting. In: Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing, pp 8306–8319
Jin W, Qu M, Jin X, Ren X (2020) Recurrent event network: Autoregressive structure inferenceover temporal knowledge graphs. In: Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing.Association for Computational Linguistics, pp 6669–6683 https://doi.org/10.18653/v1/2020.emnlp-main.541
Zhu C, Chen M, Fan C, Cheng G, Zhang Y (2021) Learning from history: Modeling temporal knowledge graphs with sequential copy-generation networks. Proceedings of the AAAI Conference on Artificial Intelligence 35:4732–4740
Hildebrandt M, Serna JAQ, Ma Y, Ringsquandl M, Joblin M, Tresp V (2020) Reasoning on knowledge graphs with debate dynamics. Proceedings of the AAAI Conference on Artificial Intelligence 34:4123–4131
Liu Y, Ma Y, Hildebrandt M, Joblin M, Tresp V (2022) Tlogic: Temporal logical rules for explainable link forecasting on temporal knowledge graphs. Proceedings of the AAAI Conference on Artificial Intelligence 36:4120–4127
Li Z, Jin X, Li W, Guan S, Guo J, Shen H, Wang Y, Cheng X (2021) Temporal knowledge graph reasoning based on evolutional representation learning. In: Proceedings of the 44th International ACM SIGIR Conference on Research and Development in Information Retrieval, pp 408–417
Veličković P, Cucurull G, Casanova A, Romero A, Lió, P., Bengio, Y (2018) Graph attention networks. In: International Conference on Learning Representations. https://openreview.net/forum?id=rJXMpikCZ
Boschee E, Lautenschlager J, O’Brien S, Shellman S, Starz J, Ward M (2023). ICEWS Coded Event Data. https://doi.org/10.7910/DVN/28075
Mahdisoltani F, Biega J, Suchanek F (2014) Yago3: A knowledge base from multilingual wikipedias. In: 7th Biennial Conference on Innovative Data Systems Research. CIDR Conference
Trouillon T, Welbl J, Riedel S, Gaussier É, Bouchard G (2016) Complex embeddings for simple link prediction. In: International Conference on Machine Learning, pp 2071–2080. PMLR
Paszke A, Gross S, Massa F, Lerer A, Bradbury J, Chanan G, Killeen T, Lin Z, Gimelshein N, Antiga L, et al (2019) Pytorch: An imperative style, high-performance deep learning library. Advances in neural information processing systems 32
Acknowledgements
This research work was partly supported by National Natural Science Foundation of China under grant No. 62271125 and No. 62273071, and by Sichuan Science and Technology Program (No.2022YFG0038 and No.2021YFG0018), and by the Fundamental Research Funds for the Central Universities (No. ZYGX2020ZB034 and No.ZYGX2021J019).
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Competing Interests
All authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Mu, C., Zhang, L., Ma, Y. et al. Temporal knowledge subgraph inference based on time-aware relation representation. Appl Intell 53, 24237–24252 (2023). https://doi.org/10.1007/s10489-023-04833-9
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10489-023-04833-9