LLM Reasoning 相关度: 9/10

Learning Chain Of Thoughts Prompts for Predicting Entities, Relations, and even Literals on Knowledge Graphs

Alkid Baci, Luke Friedrichs, Caglar Demir, N'Dah Jean Kouagou, Axel-Cyrille Ngonga Ngomo
arXiv: 2604.12651v1 发布: 2026-04-14 更新: 2026-04-14

AI 摘要

RALP通过学习CoT提示,提升LLM在知识图谱上的实体、关系和文字预测能力。

主要贡献

  • 提出 RALP 方法,利用 LLM 进行知识图谱预测
  • 使用 Bayesian Optimization 寻找有效提示
  • 在多种数据集上验证了 RALP 的有效性

方法论

将链接预测转化为提示学习问题,使用 Bayesian Optimization 优化 CoT 提示,作为三元组的评分函数。

原文摘要

Knowledge graph embedding (KGE) models perform well on link prediction but struggle with unseen entities, relations, and especially literals, limiting their use in dynamic, heterogeneous graphs. In contrast, pretrained large language models (LLMs) generalize effectively through prompting. We reformulate link prediction as a prompt learning problem and introduce RALP, which learns string-based chain-of-thought (CoT) prompts as scoring functions for triples. Using Bayesian Optimization through MIPRO algorithm, RALP identifies effective prompts from fewer than 30 training examples without gradient access. At inference, RALP predicts missing entities, relations or whole triples and assigns confidence scores based on the learned prompt. We evaluate on transductive, numerical, and OWL instance retrieval benchmarks. RALP improves state-of-the-art KGE models by over 5% MRR across datasets and enhances generalization via high-quality inferred triples. On OWL reasoning tasks with complex class expressions (e.g., $\exists hasChild.Female$, $\geq 5 \; hasChild.Female$), it achieves over 88% Jaccard similarity. These results highlight prompt-based LLM reasoning as a flexible alternative to embedding-based methods. We release our implementation, training, and evaluation pipeline as open source: https://github.com/dice-group/RALP .

标签

知识图谱 链接预测 大语言模型 提示学习 Chain-of-Thought

arXiv 分类

cs.CL cs.AI