Improving Neural Retrieval with Attribution-Guided Query Rewriting

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Garouani, Moncef, Mothe, Josiane
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908829528096768
author Garouani, Moncef
Mothe, Josiane
author_facet Garouani, Moncef
Mothe, Josiane
contents Neural retrievers are effective but brittle: underspecified or ambiguous queries can misdirect ranking even when relevant documents exist. Existing approaches address this brittleness only partially: LLMs rewrite queries without retriever feedback, and explainability methods identify misleading tokens but are used for post-hoc analysis. We close this loop and propose an attribution-guided query rewriting method that uses token-level explanations to guide query rewriting. For each query, we compute gradient-based token attributions from the retriever and then use these scores as soft guidance in a structured prompt to an LLM that clarifies weak or misleading query components while preserving intent. Evaluated on BEIR collections, the resulting rewrites consistently improve retrieval effectiveness over strong baselines, with larger gains for implicit or ambiguous information needs.
format Preprint
id arxiv_https___arxiv_org_abs_2602_11841
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Improving Neural Retrieval with Attribution-Guided Query Rewriting
Garouani, Moncef
Mothe, Josiane
Information Retrieval
Artificial Intelligence
Machine Learning
Neural retrievers are effective but brittle: underspecified or ambiguous queries can misdirect ranking even when relevant documents exist. Existing approaches address this brittleness only partially: LLMs rewrite queries without retriever feedback, and explainability methods identify misleading tokens but are used for post-hoc analysis. We close this loop and propose an attribution-guided query rewriting method that uses token-level explanations to guide query rewriting. For each query, we compute gradient-based token attributions from the retriever and then use these scores as soft guidance in a structured prompt to an LLM that clarifies weak or misleading query components while preserving intent. Evaluated on BEIR collections, the resulting rewrites consistently improve retrieval effectiveness over strong baselines, with larger gains for implicit or ambiguous information needs.
title Improving Neural Retrieval with Attribution-Guided Query Rewriting
topic Information Retrieval
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2602.11841