LLMs Know What They Need: Leveraging a Missing Information Guided Framework to Empower Retrieval-Augmented Generation

Keheng Wang, Feiyu Duan, Peiguang Li, Sirui Wang, Xunliang Cai
4/22/2024
cs.CL

No official code found. Best implementation: AdelWang/MIGRES (19 stars, last commit 2024-06-14).

Abstract

Retrieval-Augmented Generation (RAG) demonstrates great value in alleviating outdated knowledge or hallucination by supplying LLMs with updated and relevant knowledge. However, there are still several difficulties for RAG in understanding complex multi-hop query and retrieving relevant documents, which require LLMs to perform reasoning and retrieve step by step. Inspired by human's reasoning process in which they gradually search for the required information, it is natural to ask whether the LLMs could notice the missing information in each reasoning step. In this work, we first experimentally verified the ability of LLMs to extract information as well as to know the missing. Based on the above discovery, we propose a Missing Information Guided Retrieve-Extraction-Solving paradigm (MIGRES), where we leverage the identification of missing information to generate a targeted query that steers the subsequent knowledge retrieval. Besides, we design a sentence-level re-ranking filtering approach to filter the irrelevant content out from document, along with the information extraction capability of LLMs to extract useful information from cleaned-up documents, which in turn to bolster the overall efficacy of RAG. Extensive experiments conducted on multiple public datasets reveal the superiority of the proposed MIGRES method, and analytical experiments demonstrate the effectiveness of our proposed modules.

View on arXivView PDF

Code Implementations(1)

AdelWang/MIGRESCommunity100%
192Shell, PythonApr 16, 20242 years ago

Similar Papers

Cite this paper

@article{wang2024llms,
  title  = {LLMs Know What They Need: Leveraging a Missing Information Guided Framework to Empower Retrieval-Augmented Generation},
  author = {Keheng Wang and Feiyu Duan and Peiguang Li and Sirui Wang and Xunliang Cai},
  year   = {2024},
  eprint = {2404.14043},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2404.14043v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2404.14043.svg)](https://codeofpaper.com/paper/2404.14043)

Discussion