Plan*RAG: Efficient Test-Time Planning for Retrieval Augmented Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Verma, Prakhar, Midigeshi, Sukruta Prakash, Sinha, Gaurav, Solin, Arno, Natarajan, Nagarajan, Sharma, Amit
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916596589527040
author Verma, Prakhar
Midigeshi, Sukruta Prakash
Sinha, Gaurav
Solin, Arno
Natarajan, Nagarajan
Sharma, Amit
author_facet Verma, Prakhar
Midigeshi, Sukruta Prakash
Sinha, Gaurav
Solin, Arno
Natarajan, Nagarajan
Sharma, Amit
contents We introduce Plan*RAG, a novel framework that enables structured multi-hop reasoning in retrieval-augmented generation (RAG) through test-time reasoning plan generation. While existing approaches such as ReAct maintain reasoning chains within the language model's context window, we observe that this often leads to plan fragmentation and execution failures. Our key insight is that by isolating the reasoning plan as a directed acyclic graph (DAG) outside the LM's working memory, we can enable (1) systematic exploration of reasoning paths, (2) atomic subqueries enabling precise retrievals and grounding, and (3) efficiency through parallel execution and bounded context window utilization. Moreover, Plan*RAG's modular design allows it to be integrated with existing RAG methods, thus providing a practical solution to improve current RAG systems. On standard multi-hop reasoning benchmarks, Plan*RAG consistently achieves improvements over recently proposed methods such as RQ-RAG and Self-RAG, while maintaining comparable computational costs.
format Preprint
id arxiv_https___arxiv_org_abs_2410_20753
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Plan*RAG: Efficient Test-Time Planning for Retrieval Augmented Generation
Verma, Prakhar
Midigeshi, Sukruta Prakash
Sinha, Gaurav
Solin, Arno
Natarajan, Nagarajan
Sharma, Amit
Computation and Language
Machine Learning
We introduce Plan*RAG, a novel framework that enables structured multi-hop reasoning in retrieval-augmented generation (RAG) through test-time reasoning plan generation. While existing approaches such as ReAct maintain reasoning chains within the language model's context window, we observe that this often leads to plan fragmentation and execution failures. Our key insight is that by isolating the reasoning plan as a directed acyclic graph (DAG) outside the LM's working memory, we can enable (1) systematic exploration of reasoning paths, (2) atomic subqueries enabling precise retrievals and grounding, and (3) efficiency through parallel execution and bounded context window utilization. Moreover, Plan*RAG's modular design allows it to be integrated with existing RAG methods, thus providing a practical solution to improve current RAG systems. On standard multi-hop reasoning benchmarks, Plan*RAG consistently achieves improvements over recently proposed methods such as RQ-RAG and Self-RAG, while maintaining comparable computational costs.
title Plan*RAG: Efficient Test-Time Planning for Retrieval Augmented Generation
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2410.20753