Training for Compositional Sensitivity Reduces Dense Retrieval Generalization
Organizations: Redis, Bulgaria and Redis, USA
Abstract
Dense retrieval compresses texts into single embeddings ranked by cosine similarity. While efficient for recall, this interface is brittle for identity-level matching: minimal compositional edits (negation, role swaps) flip meaning yet retain high similarity. Motivated by geometric results for unit-sphere cosine spaces (Kang et al., 2025), we test this retrieval-composition tension in text-only retrieval. Across four dual-encoder backbones, adding structure-targeted negatives consistently reduces zero-shot NanoBEIR retrieval (8-9% mean nDCG@10 drop on small backbones; up to 40% on medium ones), while only partially improving pooled-space separation. Treating pooled cosine as a recall interface, we then benchmark verifiers scoring token--token cosine maps. MaxSim (late interaction) excels at reranking but fails to reject structural near-misses, whereas a small Transformer over similarity maps reliably separates near-misses under end-to-end training.
Figures & tables
| nDCG@10 | Acc@1 | |||||
|---|---|---|---|---|---|---|
| Backbone | Model A | Model B | (% drop) | Model A | Model B | (% drop) |
| MiniLM-L6 | 0.439 0.000 | 0.401 0.001 | -0.038 (-8.7%) | 0.393 0.002 | 0.346 0.004 | -0.047 (-12.0%) |
| MiniLM-L12 | 0.467 0.001 | 0.424 0.005 | -0.043 (-9.2%) | 0.424 0.003 | 0.369 0.010 | -0.055 (-13.0%) |
| GTE-Small | 0.481 0.002 | 0.442 0.006 | -0.039 (-8.1%) | 0.444 0.001 | 0.389 0.004 | -0.055 (-12.4%) |
| GTE-ModernBERT-base | 0.543 0.001 | 0.324 0.018 | -0.219 (-40.3%) | 0.493 0.005 | 0.275 0.015 | -0.218 (-44.2%) |
Information Retrieval
Information Retrieval studies how to find, rank, and organize relevant information in large collections of data.
See all 471 papers