journal article·2024Open access

Knowledge Reasoning Method Based on Deep Transfer Reinforcement Learning: DTRLpath

Shiming Lin, Ling Ye, Yijie Zhuang, Lingyun Lu, Shaoqiu Zheng, Chenxi Huang, Ng Yin Kwee

Computers, Materials & Continua· Volume 80· Issue 1

Abstract

In recent years, with the continuous development of deep learning and knowledge graph reasoning methods, more and more researchers have shown great interest in improving knowledge graph reasoning methods by inferring missing facts through reasoning. By searching paths on the knowledge graph and making fact and link predictions based on these paths, deep learning-based Reinforcement Learning (RL) agents can demonstrate good performance and interpretability. Therefore, deep reinforcement learning-based knowledge reasoning methods have rapidly emerged in recent years and have become a hot research topic. However, even in a small and fixed knowledge graph reasoning action space, there are still a large number of invalid actions. It often leads to the interruption of RL agents’ wandering due to the selection of invalid actions, resulting in a significant decrease in the success rate of path mining. In order to improve the success rate of RL agents in the early stages of path search, this article proposes a knowledge reasoning method based on Deep Transfer Reinforcement Learning path (DTRLpath). Before supervised pre-training and retraining, a pre-task of searching for effective actions in a single step is added. The RL agent is first trained in the pre-task to improve its ability to search for effective actions. Then, the trained agent is transferred to the target reasoning task for path search training, which improves its success rate in searching for target task paths. Finally, based on the comparative experimental results on the FB15K-237 and NELL-995 datasets, it can be concluded that the proposed method significantly improves the success rate of path search and outperforms similar methods in most reasoning tasks.
Source: openalex_inverted_index · Quality: baseline

Research context

OpenAlex topics: Neural Networks and Applications · Advanced Decision-Making Techniques · Fuzzy Logic and Control Systems

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

References (31)

Show source-provided references
1. Knowledge graph embedding based question answering
2. Question answering over freebase with multi-column convolutional neural networks
3. Y. Gao, Y. Li, Y. Lin, H. Gao, and L. Khan, “Deep learning on knowledge graph for recommender system: A survey,” arXiv preprint arXiv:2004.00387, 2020.
4. A survey on knowledge graph-based recommender systems10.1109/TKDE.2020.3028705 ↗
5. W. Xiong, T. Hoang, and W. Yang Wang, “DeepPath: A reinforcement learning method for knowledge graph reasoning,” arXiv preprint arXiv:1707.06690, 2017.10.18653/v1/D17-1060 ↗
6. Temporal knowledge graph reasoning with historical contrastive learning
7. Translating embeddings for modeling multi-relational data
8. Survey on explainable knowledge graph reasoning methods
9. Knowledge graph embedding via dynamic mapping matrix
10. Learning entity and relation embeddings for knowledge graph completion
11. Knowledge graph embedding by translating on hyperplanes
12. A three-way model for collective learning on multi-relational data
13. Embedding entities and relations for learning and inference in knowledge bases
14. Complex embeddings for simple link prediction
15. Analogical inference for multi-relational embeddings
16. Go for a walk and arrive at the answer: reasoning over paths in knowledge bases using reinforcement learning
17. Incorporating graph attention mechanism into knowledge graph reasoning based on deep reinforcement learning
18. Long short-term memory10.1162/neco.1997.9.8.1735 ↗
19. P. Veličković, G. Cucurull, A. Casanova, A. Romero, P. Liò and Y. Bengio, “Graph attention networks,” arXiv preprint arXiv:1710.10903, 2017.
20. Relational retrieval using a combination of path-constrained random walks10.1007/s10994-010-5205-8 ↗
21. Knowledge base completion via coupled path ranking
22. Li · Reinforcement Learning for Sequential Decision and Optimal Control · 2023 · 36510.1007/978-981-19-7784-8_10 ↗
23. Transfer learning in deep reinforcement learning: A survey10.1109/TPAMI.2023.3292075 ↗
24. Deep transfer reinforcement learning for text summarization
25. P. Ammanabrolu and M. O. Riedl, “Transfer in deep reinforcement learning using knowledge graphs,” arXiv preprint arXiv:1908.06556, 2019.10.18653/v1/D19-5301 ↗
26. Value function transfer for deep multi-agent reinforcement learning based on N-Step returns
27. D. P. Kingma and J. Ba, “Adam: a method for stochastic optimization,” arXiv preprint arXiv:1412.6980, 2014.
28. Simple statistical gradient-following algorithms for connectionist reinforcement learning10.1007/BF00992696 ↗
29. OpenKE: An open toolkit for knowledge embedding
30. W. Chen, W. Xiong, X. Yan, and W. Y. Wang, “Variational knowledge graph reasoning,” arXiv preprint arXiv:1803.06581, 2018.10.18653/v1/N18-1165 ↗
31. DIVINE: A generative adversarial imitation learning framework for knowledge graph reasoning