Augmenting Textual Generation via Topology Aware Retrieval

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Yu, Lipka, Nedim, Zhang, Ruiyi, Siu, Alexa, Zhao, Yuying, Ni, Bo, Wang, Xin, Rossi, Ryan, Derr, Tyler
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913365745467392
author Wang, Yu
Lipka, Nedim
Zhang, Ruiyi
Siu, Alexa
Zhao, Yuying
Ni, Bo
Wang, Xin
Rossi, Ryan
Derr, Tyler
author_facet Wang, Yu
Lipka, Nedim
Zhang, Ruiyi
Siu, Alexa
Zhao, Yuying
Ni, Bo
Wang, Xin
Rossi, Ryan
Derr, Tyler
contents Despite the impressive advancements of Large Language Models (LLMs) in generating text, they are often limited by the knowledge contained in the input and prone to producing inaccurate or hallucinated content. To tackle these issues, Retrieval-augmented Generation (RAG) is employed as an effective strategy to enhance the available knowledge base and anchor the responses in reality by pulling additional texts from external databases. In real-world applications, texts are often linked through entities within a graph, such as citations in academic papers or comments in social networks. This paper exploits these topological relationships to guide the retrieval process in RAG. Specifically, we explore two kinds of topological connections: proximity-based, focusing on closely connected nodes, and role-based, which looks at nodes sharing similar subgraph structures. Our empirical research confirms their relevance to text relationships, leading us to develop a Topology-aware Retrieval-augmented Generation framework. This framework includes a retrieval module that selects texts based on their topological relationships and an aggregation module that integrates these texts into prompts to stimulate LLMs for text generation. We have curated established text-attributed networks and conducted comprehensive experiments to validate the effectiveness of this framework, demonstrating its potential to enhance RAG with topological awareness.
format Preprint
id arxiv_https___arxiv_org_abs_2405_17602
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Augmenting Textual Generation via Topology Aware Retrieval
Wang, Yu
Lipka, Nedim
Zhang, Ruiyi
Siu, Alexa
Zhao, Yuying
Ni, Bo
Wang, Xin
Rossi, Ryan
Derr, Tyler
Information Retrieval
Despite the impressive advancements of Large Language Models (LLMs) in generating text, they are often limited by the knowledge contained in the input and prone to producing inaccurate or hallucinated content. To tackle these issues, Retrieval-augmented Generation (RAG) is employed as an effective strategy to enhance the available knowledge base and anchor the responses in reality by pulling additional texts from external databases. In real-world applications, texts are often linked through entities within a graph, such as citations in academic papers or comments in social networks. This paper exploits these topological relationships to guide the retrieval process in RAG. Specifically, we explore two kinds of topological connections: proximity-based, focusing on closely connected nodes, and role-based, which looks at nodes sharing similar subgraph structures. Our empirical research confirms their relevance to text relationships, leading us to develop a Topology-aware Retrieval-augmented Generation framework. This framework includes a retrieval module that selects texts based on their topological relationships and an aggregation module that integrates these texts into prompts to stimulate LLMs for text generation. We have curated established text-attributed networks and conducted comprehensive experiments to validate the effectiveness of this framework, demonstrating its potential to enhance RAG with topological awareness.
title Augmenting Textual Generation via Topology Aware Retrieval
topic Information Retrieval
url https://arxiv.org/abs/2405.17602