English

RaFe: Ranking Feedback Improves Query Rewriting for RAG

Computation and Language 2024-05-24 v1 Artificial Intelligence Information Retrieval

Abstract

As Large Language Models (LLMs) and Retrieval Augmentation Generation (RAG) techniques have evolved, query rewriting has been widely incorporated into the RAG system for downstream tasks like open-domain QA. Many works have attempted to utilize small models with reinforcement learning rather than costly LLMs to improve query rewriting. However, current methods require annotations (e.g., labeled relevant documents or downstream answers) or predesigned rewards for feedback, which lack generalization, and fail to utilize signals tailored for query rewriting. In this paper, we propose ours, a framework for training query rewriting models free of annotations. By leveraging a publicly available reranker, ours~provides feedback aligned well with the rewriting objectives. Experimental results demonstrate that ours~can obtain better performance than baselines.

Keywords

Cite

@article{arxiv.2405.14431,
  title  = {RaFe: Ranking Feedback Improves Query Rewriting for RAG},
  author = {Shengyu Mao and Yong Jiang and Boli Chen and Xiao Li and Peng Wang and Xinyu Wang and Pengjun Xie and Fei Huang and Huajun Chen and Ningyu Zhang},
  journal= {arXiv preprint arXiv:2405.14431},
  year   = {2024}
}

Comments

16 pages

R2 v1 2026-06-28T16:37:02.716Z