高级检索

    面向时序知识图谱推理的本体-图谱双智能体强化学习方法

    Ontology-Graph Dual-Agent Reinforcement Learning Method for Temporal Knowledge Graph Reasoning

    • 摘要: 时序知识图谱推理旨在基于历史事实预测未来事件,而现有基于强化学习的推理方法仅在实体级进行路径搜索,缺乏高层语义逻辑指导。本体作为图谱的概念级抽象具有很强的逻辑性,而本体构建时依赖静态统计、难以动态适应。为此,提出面向时序知识图谱推理的本体-图谱双智能体强化学习方法ODRT。该方法通过大语言模型语义嵌入与预训练时序知识图谱推理模型结构嵌入联合初始化时序知识图谱本体,并在训练中通过可学习频次权重实现本体动态优化。双智能体强化学习框架中,上层本体智能体在本体图上执行概念级游走,通过历史状态共享为下层图谱智能体提供语义导航,下层智能体在时序图谱上执行实体级搜索并受分层软奖励约束。在4个公开的推理数据集上的广泛实验表明,ODRT在现有基于强化学习的时序推理方法中取得最优性能,并在2个百科型数据集上达到当前最优水平;而在实体行为模式异质且预测依赖历史复用的事件型数据集上,其性能与部分嵌入方法相比存在一定差距。深入实验分析证明了本体引导机制和双智能体强化学习框架的优势。

       

      Abstract: Temporal knowledge graph (TKG) reasoning aims to predict future events based on historical facts. However, existing reinforcement learning (RL)-based reasoning methods merely conduct path searches at the entity level, lacking high-level semantic and logical guidance for the reasoning process. Moreover, ontologies, as conceptual abstractions of graphs, have strong logicality, but their construction typically relies on static statistical rules and is difficult to adapt dynamically to varying data distributions. To effectively address these issues, this paper proposes ODRT, an ontology-graph dual-agent reinforcement learning method for TKG reasoning. Specifically, it jointly initializes the TKG ontology by leveraging both large language model (LLM) semantic embeddings and pretrained TKG reasoning model structural embeddings. Furthermore, it dynamically optimizes the ontology structure through learnable frequency weights during the end-to-end training process. Within the proposed dual-agent RL framework, the upper-level ontology agent performs concept-level walks on the constructed ontology graph, providing effective semantic navigation for the lower-level graph agent through carefully designed shared historical states. Meanwhile, the lower-level graph agent conducts fine-grained entity-level searches on the temporal graph under the constraint of hierarchical soft reward functions. Extensive experiments on four publicly available reasoning datasets demonstrate that ODRT achieves state-of-the-art performance among existing RL-based baselines and reaches the current state-of-the-art level on two encyclopedic datasets. However, on event-based datasets with heterogeneous entity behavior patterns and predictions that rely on historical reuse, its performance still lags behind that of partially embedded methods. Comprehensive experimental analyses further verify the significant advantages of the proposed ontology-guided mechanism and the dual-agent reinforcement learning framework.

       

    /

    返回文章
    返回