Query Optimization for Parametric Knowledge Refinement in Retrieval-Augmented Large Language Models

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Cong, Youan, Akash, Pritom Saha, Wang, Cheng, Chang, Kevin Chen-Chuan
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912594700271616
author Cong, Youan
Akash, Pritom Saha
Wang, Cheng
Chang, Kevin Chen-Chuan
author_facet Cong, Youan
Akash, Pritom Saha
Wang, Cheng
Chang, Kevin Chen-Chuan
contents We introduce the \textit{Extract-Refine-Retrieve-Read} (ERRR) framework, a novel approach designed to bridge the pre-retrieval information gap in Retrieval-Augmented Generation (RAG) systems through query optimization tailored to meet the specific knowledge requirements of Large Language Models (LLMs). Unlike conventional query optimization techniques used in RAG, the ERRR framework begins by extracting parametric knowledge from LLMs, followed by using a specialized query optimizer for refining these queries. This process ensures the retrieval of only the most pertinent information essential for generating accurate responses. Moreover, to enhance flexibility and reduce computational costs, we propose a trainable scheme for our pipeline that utilizes a smaller, tunable model as the query optimizer, which is refined through knowledge distillation from a larger teacher model. Our evaluations on various question-answering (QA) datasets and with different retrieval systems show that ERRR consistently outperforms existing baselines, proving to be a versatile and cost-effective module for improving the utility and accuracy of RAG systems.
format Preprint
id arxiv_https___arxiv_org_abs_2411_07820
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Query Optimization for Parametric Knowledge Refinement in Retrieval-Augmented Large Language Models
Cong, Youan
Akash, Pritom Saha
Wang, Cheng
Chang, Kevin Chen-Chuan
Computation and Language
Information Retrieval
We introduce the \textit{Extract-Refine-Retrieve-Read} (ERRR) framework, a novel approach designed to bridge the pre-retrieval information gap in Retrieval-Augmented Generation (RAG) systems through query optimization tailored to meet the specific knowledge requirements of Large Language Models (LLMs). Unlike conventional query optimization techniques used in RAG, the ERRR framework begins by extracting parametric knowledge from LLMs, followed by using a specialized query optimizer for refining these queries. This process ensures the retrieval of only the most pertinent information essential for generating accurate responses. Moreover, to enhance flexibility and reduce computational costs, we propose a trainable scheme for our pipeline that utilizes a smaller, tunable model as the query optimizer, which is refined through knowledge distillation from a larger teacher model. Our evaluations on various question-answering (QA) datasets and with different retrieval systems show that ERRR consistently outperforms existing baselines, proving to be a versatile and cost-effective module for improving the utility and accuracy of RAG systems.
title Query Optimization for Parametric Knowledge Refinement in Retrieval-Augmented Large Language Models
topic Computation and Language
Information Retrieval
url https://arxiv.org/abs/2411.07820