Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

IMRNNs: An Efficient Method for Interpretable Dense Retrieval via Embedding Modulation

About

Interpretability in black-box dense retrievers remains a central challenge in Retrieval-Augmented Generation (RAG). Understanding how queries and documents semantically interact is critical for diagnosing retrieval behavior and improving model design. However, existing dense retrievers rely on static embeddings for both queries and documents, which obscures this bidirectional relationship. Post-hoc approaches such as re-rankers are computationally expensive, add inference latency, and still fail to reveal the underlying semantic alignment. To address these limitations, we propose Interpretable Modular Retrieval Neural Networks (IMRNNs), a lightweight framework that augments any dense retriever with dynamic, bidirectional modulation at inference time. IMRNNs employ two independent adapters: one conditions document embeddings on the current query, while the other refines the query embedding using corpus-level feedback from initially retrieved documents. This iterative modulation process enables the model to adapt representations dynamically and expose interpretable semantic dependencies between queries and documents. Empirically, IMRNNs not only enhance interpretability but also improve retrieval effectiveness. Across seven benchmark datasets, applying our method to standard dense retrievers yields average gains of +6.35% nDCG, +7.14% recall, and +7.04% MRR over state-of-the-art baselines. These results demonstrate that incorporating interpretability-driven modulation can both explain and enhance retrieval in RAG systems.

Yash Saxena, Ankur Padia, Kalpa Gunaratna, Manas Gaur• 2026

Related benchmarks

TaskDatasetResultRank
Information RetrievalMS Marco
NDCG@1093
56
Information RetrievalTREC-COVID
NDCG@1085
30
Information RetrievalFiQA
MRR0.42
22
Information RetrievalHotpotQA
NDCG@1063
19
Information RetrievalNatural Questions
nDCG91
15
Information RetrievalSciFact
nDCG82
15
Information RetrievalWebis-Touche 2020
nDCG62
15
Information RetrievalArguana BEIR
nDCG90
11
Information RetrievalQuora BEIR
nDCG92
11
Information RetrievalScidocs BEIR
nDCG0.29
6
Showing 10 of 11 rows

Other info

Follow for update