AirRAG: Autonomous Strategic Planning and Reasoning Steer Retrieval Augmented Generation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Feng, Wenfeng, Hao, Chuzhan, Zhang, Yuewei, Jiang, Guochao, Song, Jingyi, Wang, Hao
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911123716964352
author Feng, Wenfeng
Hao, Chuzhan
Zhang, Yuewei
Jiang, Guochao
Song, Jingyi
Wang, Hao
author_facet Feng, Wenfeng
Hao, Chuzhan
Zhang, Yuewei
Jiang, Guochao
Song, Jingyi
Wang, Hao
contents Leveraging the autonomous decision-making capabilities of large language models (LLMs) has demonstrated superior performance in reasoning tasks. However, despite the success of iterative or agentic retrieval-augmented generation (RAG) techniques, these methods are often constrained to a single solution space when confronted with complex problems. In this paper, we propose a novel thinking pattern in RAG that integrates autonomous strategic planning with efficient reasoning actions, significantly activating intrinsic reasoning capabilities and expanding the solution space of specific tasks via Monte Carlo Tree Search (MCTS), which we refer to as AirRAG. Specifically, our approach designs five fundamental reasoning actions, which are expanded to a broad tree-based reasoning space using MCTS. The approach also incorporates self-consistency verification to explore potential reasoning paths and inference scaling law. Additionally, computationally optimal strategies are employed to allocate more inference resources to key actions, thereby enhancing overall performance. Experimental results demonstrate the effectiveness of AirRAG, showing significant performance gains on complex question-answering datasets. Furthermore, AirRAG is flexible and lightweight, making it easy to integrate with other advanced technologies and models.
format Preprint
id arxiv_https___arxiv_org_abs_2501_10053
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AirRAG: Autonomous Strategic Planning and Reasoning Steer Retrieval Augmented Generation
Feng, Wenfeng
Hao, Chuzhan
Zhang, Yuewei
Jiang, Guochao
Song, Jingyi
Wang, Hao
Artificial Intelligence
Leveraging the autonomous decision-making capabilities of large language models (LLMs) has demonstrated superior performance in reasoning tasks. However, despite the success of iterative or agentic retrieval-augmented generation (RAG) techniques, these methods are often constrained to a single solution space when confronted with complex problems. In this paper, we propose a novel thinking pattern in RAG that integrates autonomous strategic planning with efficient reasoning actions, significantly activating intrinsic reasoning capabilities and expanding the solution space of specific tasks via Monte Carlo Tree Search (MCTS), which we refer to as AirRAG. Specifically, our approach designs five fundamental reasoning actions, which are expanded to a broad tree-based reasoning space using MCTS. The approach also incorporates self-consistency verification to explore potential reasoning paths and inference scaling law. Additionally, computationally optimal strategies are employed to allocate more inference resources to key actions, thereby enhancing overall performance. Experimental results demonstrate the effectiveness of AirRAG, showing significant performance gains on complex question-answering datasets. Furthermore, AirRAG is flexible and lightweight, making it easy to integrate with other advanced technologies and models.
title AirRAG: Autonomous Strategic Planning and Reasoning Steer Retrieval Augmented Generation
topic Artificial Intelligence
url https://arxiv.org/abs/2501.10053