Tree-of-Traversals: A Zero-Shot Reasoning Algorithm for Augmenting Black-box Language Models with Knowledge Graphs

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Markowitz, Elan, Ramakrishna, Anil, Dhamala, Jwala, Mehrabi, Ninareh, Peris, Charith, Gupta, Rahul, Chang, Kai-Wei, Galstyan, Aram
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929444268015616
author Markowitz, Elan
Ramakrishna, Anil
Dhamala, Jwala
Mehrabi, Ninareh
Peris, Charith
Gupta, Rahul
Chang, Kai-Wei
Galstyan, Aram
author_facet Markowitz, Elan
Ramakrishna, Anil
Dhamala, Jwala
Mehrabi, Ninareh
Peris, Charith
Gupta, Rahul
Chang, Kai-Wei
Galstyan, Aram
contents Knowledge graphs (KGs) complement Large Language Models (LLMs) by providing reliable, structured, domain-specific, and up-to-date external knowledge. However, KGs and LLMs are often developed separately and must be integrated after training. We introduce Tree-of-Traversals, a novel zero-shot reasoning algorithm that enables augmentation of black-box LLMs with one or more KGs. The algorithm equips a LLM with actions for interfacing a KG and enables the LLM to perform tree search over possible thoughts and actions to find high confidence reasoning paths. We evaluate on two popular benchmark datasets. Our results show that Tree-of-Traversals significantly improves performance on question answering and KG question answering tasks. Code is available at \url{https://github.com/amazon-science/tree-of-traversals}
format Preprint
id arxiv_https___arxiv_org_abs_2407_21358
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Tree-of-Traversals: A Zero-Shot Reasoning Algorithm for Augmenting Black-box Language Models with Knowledge Graphs
Markowitz, Elan
Ramakrishna, Anil
Dhamala, Jwala
Mehrabi, Ninareh
Peris, Charith
Gupta, Rahul
Chang, Kai-Wei
Galstyan, Aram
Artificial Intelligence
Knowledge graphs (KGs) complement Large Language Models (LLMs) by providing reliable, structured, domain-specific, and up-to-date external knowledge. However, KGs and LLMs are often developed separately and must be integrated after training. We introduce Tree-of-Traversals, a novel zero-shot reasoning algorithm that enables augmentation of black-box LLMs with one or more KGs. The algorithm equips a LLM with actions for interfacing a KG and enables the LLM to perform tree search over possible thoughts and actions to find high confidence reasoning paths. We evaluate on two popular benchmark datasets. Our results show that Tree-of-Traversals significantly improves performance on question answering and KG question answering tasks. Code is available at \url{https://github.com/amazon-science/tree-of-traversals}
title Tree-of-Traversals: A Zero-Shot Reasoning Algorithm for Augmenting Black-box Language Models with Knowledge Graphs
topic Artificial Intelligence
url https://arxiv.org/abs/2407.21358