Align then Train: Efficient Retrieval Adapter Learning
About
Dense retrieval systems increasingly need to handle complex queries. In many realistic settings, users express intent through long instructions or task-specific descriptions, while target documents remain relatively simple and static. This asymmetry creates a retrieval mismatch: understanding queries may require strong reasoning and instruction-following, whereas efficient document indexing favors lightweight encoders. Existing retrieval systems often address this mismatch by directly improving the embedding model, but fine-tuning large embedding models to better follow such instructions is computationally expensive, memory-intensive, and operationally burdensome. To address this challenge, we propose Efficient Retrieval Adapter (ERA), a label-efficient framework that trains retrieval adapters in two stages: self-supervised alignment and supervised adaptation. Inspired by the pre-training and supervised fine-tuning stages of LLMs, ERA first aligns the embedding spaces of a large query embedder and a lightweight document embedder, and then uses limited labeled data to adapt the query-side representation, bridging both the representation gap between embedding models and the semantic gap between complex queries and simple documents without re-indexing the corpus. Experiments on the MAIR benchmark, spanning 126 retrieval tasks across 6 domains, show that ERA improves retrieval in low-label settings, outperforms methods that rely on larger amounts of labeled data, and effectively combines stronger query embedders with weaker document embedders across domains.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Information Retrieval | MAIR 20% labeled query-document pairs | Avg MAP@10048.55 | 30 | |
| Information Retrieval | MAIR 5% labeled query-document pairs (test) | Average IR Score58.53 | 15 | |
| Query-document retrieval | MAIR (20% labeled) | Average Score77.08 | 15 | |
| Retrieval | MAIR 40% labeled query-document pairs | Avg Score59.99 | 15 | |
| Retrieval | MAIR 10% labeled query-document pairs | Average Score58.92 | 15 | |
| Retrieval | MAIR (test) | nDCG@10 (Avg)59.38 | 15 |