Hierarchical graph attention network for temporal knowledge graph reasoning

• Published in: Neurocomputing (Amsterdam) Vol. 550; p. 126390
• Main Authors: Shao, Pengpeng, He, Jiayi, Li, Guanjun, Zhang, Dawei, Tao, Jianhua
• Format: Journal Article
• Language: English
• Published: Elsevier B.V 14.09.2023
• Subjects: Graph attention network; Reasoning; Temporal knowledge graphs; Temporal knowledge graphs; Reasoning; Graph attention network
• ISSN: 0925-2312; 1872-8286
• DOI: 10.1016/j.neucom.2023.126390

Temporal knowledge graphs (TKGs) reasoning has attracted increasing research interest in recent years. However, most of the existing TKGs reasoning models aim to learn a dynamic entity representation by binding timestamps information with the entities, neglecting to learn adaptive entity representation that is valuable to the query from relevant historical facts. To this end, we propose a Hierarchical Graph Attention neTwork (HGAT) for the TKGs reasoning task. Specifically, we design a hierarchical neighbor encoder to model the time-oriented and task-oriented roles of the entities. The time-aware mechanism is developed in the first layer to differentiate the contributions of query-relevant historical facts at different timestamps to the query. The designed relation-aware attention is used in the second layer to discern the contributions of the structural neighbors of an entity. Through this hierarchical encoder, our model can absorb valuable knowledge effectively from the relevant historical facts, and thus learn more expressive adaptive entity representation for the query. Finally, we evaluate our model performance on four TKGs datasets and justify its superiority against vaerious state-of-the-art baselines.