Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

RSRank: Learning Relevance from Representational Shifts

About

As enterprises deploy RAG-based systems to provide grounded responses to user queries, reranking has become a critical component for the final filtering step that separates relevant from distracting or irrelevant documents. Existing rerankers often rely on heuristic thresholds to achieve optimal filtering. Moreover, for relevance scoring, state-of-the-art methods use a language model's logit signals, which are designed for next-token prediction, not for assessing relevance. To address these limitations, we identify a principled signal for relevance: the representational shift (RS) induced in a query's internal state when conditioned on a document. We observe that the alignment between (a) RS induced by a candidate document and (b) RS induced by an oracle document-set provides a robust indicator of relevance. Building on this insight, we introduce a lightweight training framework that learns projections mapping RS to calibrated relevance scores. Our training objectives naturally filter irrelevant content at a zero threshold, reducing dependence on heuristic tuning. Across diverse retrieval datasets, our method delivers gains over SOTA rerankers.

Archit Gupta, Sai Sundaresan, Debabrata Mahapatra• 2026

Related benchmarks

TaskDatasetResultRank
Document RerankingFEVER
NDCG@599.2
19
Document Reranking2WikiMQA
NDCG@580.2
5
Document RerankingMuSiQue
NDCG@584
5
Document RerankingHotpotQA
NDCG@579.6
5
Document RerankingFiQA
NDCG@597.5
5
Document RerankingNFCorpus
NDCG@583.4
5
Showing 6 of 6 rows

Other info

Follow for update