COCO-DR: Combating Distribution Shifts in Zero-Shot Dense Retrieval with Contrastive and Distributionally Robust Learning
Not on the map. Why?
Abstract (arXiv)
We present a new zero-shot dense retrieval (ZeroDR) method, COCO-DR, to improve the generalization ability of dense retrieval by combating the distribution shifts between source training tasks and target scenarios. To mitigate the impact of document differences, COCO-DR continues pretraining the language model on the target corpora to adapt the model to target distributions via COtinuous COtrastive learning. To prepare for unseen target queries, COCO-DR leverages implicit Distributionally Robust Optimization (iDRO) to reweight samples from different source query clusters for improving model robustness over rare queries during fine-tuning. COCO-DR achieves superior average performance on BEIR, the zero-shot retrieval benchmark. At BERT Base scale, COCO-DR Base outperforms other ZeroDR models with 60x larger size. At BERT Large scale, COCO-DR Large outperforms the giant GPT-3 embedding model which has 500x more parameters. Our analysis show the correlation between COCO-DR's effectiveness in combating distribution shifts and improving zero-shot accuracy. Our code and model can be found at urlhttps://github.com/OpenMatch/COCO-DR.
Key facts. Cited by 19 papers in the map (data through 7 Oct 2026). Builds on 48 of its 76 references that are in the map. First posted on arXiv 27 Oct 2022.
Builds on
48 of 76 references.
References · 48
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,168
- 2016MS MARCO: A Human Generated MAchine Reading COmprehension Datasetcited by 1,591
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,548
36 more
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,414
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,393
- 2018FEVER: a large-scale dataset for Fact Extraction and VERificationcited by 1,362
- 2021BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Modelscited by 1,225
- 2021Unsupervised Dense Information Retrieval with Contrastive Learningcited by 1,203
- 2020ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERTcited by 1,160
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2020Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text Retrievalcited by 846
- 2019Latent Retrieval for Weakly Supervised Open Domain Question Answeringcited by 691
- 2019Green AIcited by 620
- 2020Fact or Fiction: Verifying Scientific Claimscited by 546
- 2019CCNet: Extracting High Quality Monolingual Datasets from Web Crawl Datacited by 544
- 2021Large Dual Encoders Are Generalizable Retrieverscited by 454
- 2020SPECTER: Document-level Representation Learning using Citation-informed Transformerscited by 426
- 2020RocketQA: An Optimized Training Approach to Dense Passage Retrieval for Open-Domain Question Answeringcited by 410
- 2022Text and Code Embeddings by Contrastive Pre-Trainingcited by 358
- 2021Efficiently Teaching an Effective Dense Retriever with Balanced Topic Aware Samplingcited by 238
- 2020No Subclass Left Behind: Fine-Grained Robustness in Coarse-Grained Classification Problemscited by 208
- 2021Unsupervised Corpus Aware Language Model Pre-training for Dense Passage Retrievalcited by 203
- 2020CLIMATE-FEVER: A Dataset for Verification of Real-World Climate Claimscited by 195
- 2021A Fine-Grained Analysis on Distribution Shiftcited by 171
- 2021Condenser: a Pre-training Architecture for Dense Retrievalcited by 162
- 2019Distributionally Robust Language Modelingcited by 142
- 2021COCO-LM: Correcting and Contrasting Text Sequences for Language Model Pretrainingcited by 126
- 2021GPL: Generative Pseudo Labeling for Unsupervised Domain Adaptation of Dense Retrievalcited by 119
- 2021TSDAE: Using Transformer-based Sequential Denoising Auto-Encoder for Unsupervised Sentence Embedding Learningcited by 116
- 2022From Distillation to Hard Negative Sampling: Making Sparse Neural IR Models More Effectivecited by 98
- 2021Examining and Combating Spurious Features under Distribution Shiftcited by 64
- 2021UDALM: Unsupervised Domain Adaptation through Language Modelingcited by 33
- 2021Zero-Shot Dense Retrieval with Momentum Adversarial Domain Invariant Representationscited by 27
- 2021Focus on the Common Good: Group Distributional Robustness Followscited by 25
- 2022LaPraDoR: Unsupervised Pretrained Dense Retriever for Zero-Shot Text Retrievalcited by 24
- 2022Pre-train a Discriminative Text Encoder for Dense Retrieval via Contrastive Span Predictioncited by 23
- 2022A Thorough Examination on Zero-shot Dense Retrievalcited by 21
- 2021OpenMatch: An Open Source Library for Neu-IR Researchcited by 6
- 2021Domain-Specific Pretraining for Vertical Search: Case Study on Biomedical Literaturecited by 6
Built on by
Citing papers per year: 2022: 1; 2023: 6; 2024: 5; 2025: 3; 2026: 4.
19 papers.
Sort:
By area:
- 2022Precise Zero-Shot Dense Retrieval without Relevance Labelscited by 508
- 2023ReGen: Zero-Shot Text Classification via Training Data Generation with Progressive Dense Retrievalcited by 26
- 2025Collab-RAG: Boosting Retrieval-Augmented Generation for Complex Question Answering via White-Box and Black-Box LLM Collaborationcited by 19
- 2025Collapse of Dense Retrievers: Short, Early, and Literal Biases Outranking Factual Evidencecited by 16
- 2023Synergistic Interplay between Search and Large Language Models for Information Retrievalcited by 11
- 2023Chain-of-Factors Paper-Reviewer Matchingcited by 11
- 2023Weakly-Supervised Scientific Document Classification via Retrieval-Augmented Multi-Stage Trainingcited by 8
- 2024Beyond Borders: Investigating Cross-Jurisdiction Transfer in Legal Case Summarizationcited by 8
- 2023I3: Intent-Introspective Retrieval Conditioned on Instructionscited by 5
- 2023Boot and Switch: Alternating Distillation for Zero-Shot Dense Retrievalcited by 4
- 2024Large Language Model Augmented Exercise Retrieval for Personalized Language Learningcited by 4
- 2024Investigating Mixture of Experts in Dense Retrievalcited by 4
7 more
- 2024PairDistill: Pairwise Relevance Distillation for Dense Retrievalcited by 3
- 2024DREditor: An Time-efficient Approach for Building a Domain-specific Dense Retrieval Modelcited by 2
- 2025A Representation Sharpening Framework for Zero Shot Dense Retrievalcited by 1
- 2026With Argus Eyes: Assessing Retrieval Gaps via Uncertainty Scoring to Detect and Remedy Retrieval Blind Spotscited by 1
- 2026Beyond Paper-to-Paper: Structured Profiling and Rubric Scoring for Paper-Reviewer Matchingcited by 0
- 2026CoHyDE: Iterative Co-Training of LLM Rewriter & Dense Encoder for Tool Retrievalcited by 0
- 2026When Does Dense Retrieval Need Asymmetric Geometry? A Bias-Variance Theory of Shared and Dual Projectionscited by 0
- 2026When Does Dense Retrieval Need Asymmetric Geometry? A Bias-Variance Theory of Shared and Dual Projections26 Sep 2026
- 2026CoHyDE: Iterative Co-Training of LLM Rewriter & Dense Encoder for Tool Retrieval28 May 2026
- 2026Beyond Paper-to-Paper: Structured Profiling and Rubric Scoring for Paper-Reviewer Matching7 Apr 2026
- 2026With Argus Eyes: Assessing Retrieval Gaps via Uncertainty Scoring to Detect and Remedy Retrieval Blind Spots10 Feb 2026
- 2025A Representation Sharpening Framework for Zero Shot Dense Retrieval7 Nov 2025
- 2025Collab-RAG: Boosting Retrieval-Augmented Generation for Complex Question Answering via White-Box and Black-Box LLM Collaboration7 Apr 2025
- 2025Collapse of Dense Retrievers: Short, Early, and Literal Biases Outranking Factual Evidence6 Mar 2025
- 2024Investigating Mixture of Experts in Dense Retrieval16 Dec 2024
- 2024PairDistill: Pairwise Relevance Distillation for Dense Retrieval2 Oct 2024
- 2024Beyond Borders: Investigating Cross-Jurisdiction Transfer in Legal Case Summarization28 Mar 2024
- 2024Large Language Model Augmented Exercise Retrieval for Personalized Language Learning8 Feb 2024
- 2024DREditor: An Time-efficient Approach for Building a Domain-specific Dense Retrieval Model23 Jan 2024
7 more
- 2023Boot and Switch: Alternating Distillation for Zero-Shot Dense Retrieval27 Nov 2023
- 2023Chain-of-Factors Paper-Reviewer Matching23 Oct 2023
- 2023I3: Intent-Introspective Retrieval Conditioned on Instructions19 Aug 2023
- 2023Weakly-Supervised Scientific Document Classification via Retrieval-Augmented Multi-Stage Training12 Jun 2023
- 2023ReGen: Zero-Shot Text Classification via Training Data Generation with Progressive Dense Retrieval18 May 2023
- 2023Synergistic Interplay between Search and Large Language Models for Information Retrieval12 May 2023
- 2022Precise Zero-Shot Dense Retrieval without Relevance Labels20 Dec 2022
- 2025Collapse of Dense Retrievers: Short, Early, and Literal Biases Outranking Factual Evidencecited by 16
- 2023Weakly-Supervised Scientific Document Classification via Retrieval-Augmented Multi-Stage Trainingcited by 8
- 2026With Argus Eyes: Assessing Retrieval Gaps via Uncertainty Scoring to Detect and Remedy Retrieval Blind Spotscited by 1
- 2026When Does Dense Retrieval Need Asymmetric Geometry? A Bias-Variance Theory of Shared and Dual Projectionscited by 0
- 2026When Does Dense Retrieval Need Asymmetric Geometry? A Bias-Variance Theory of Shared and Dual Projections26 Sep 2026
- 2026With Argus Eyes: Assessing Retrieval Gaps via Uncertainty Scoring to Detect and Remedy Retrieval Blind Spots10 Feb 2026
- 2025Collapse of Dense Retrievers: Short, Early, and Literal Biases Outranking Factual Evidence6 Mar 2025
- 2023Weakly-Supervised Scientific Document Classification via Retrieval-Augmented Multi-Stage Training12 Jun 2023
- 2024Investigating Mixture of Experts in Dense Retrievalcited by 4
- 2024Investigating Mixture of Experts in Dense Retrieval16 Dec 2024
- 2022Precise Zero-Shot Dense Retrieval without Relevance Labelscited by 508
- 2023ReGen: Zero-Shot Text Classification via Training Data Generation with Progressive Dense Retrievalcited by 26
- 2023Synergistic Interplay between Search and Large Language Models for Information Retrievalcited by 11
- 2023Chain-of-Factors Paper-Reviewer Matchingcited by 11
- 2024Beyond Borders: Investigating Cross-Jurisdiction Transfer in Legal Case Summarizationcited by 8
- 2023I3: Intent-Introspective Retrieval Conditioned on Instructionscited by 5
- 2023Boot and Switch: Alternating Distillation for Zero-Shot Dense Retrievalcited by 4
- 2024Large Language Model Augmented Exercise Retrieval for Personalized Language Learningcited by 4
- 2024PairDistill: Pairwise Relevance Distillation for Dense Retrievalcited by 3
- 2024DREditor: An Time-efficient Approach for Building a Domain-specific Dense Retrieval Modelcited by 2
- 2025A Representation Sharpening Framework for Zero Shot Dense Retrievalcited by 1
- 2026Beyond Paper-to-Paper: Structured Profiling and Rubric Scoring for Paper-Reviewer Matchingcited by 0
- 2026Beyond Paper-to-Paper: Structured Profiling and Rubric Scoring for Paper-Reviewer Matching7 Apr 2026
- 2025A Representation Sharpening Framework for Zero Shot Dense Retrieval7 Nov 2025
- 2024PairDistill: Pairwise Relevance Distillation for Dense Retrieval2 Oct 2024
- 2024Beyond Borders: Investigating Cross-Jurisdiction Transfer in Legal Case Summarization28 Mar 2024
- 2024Large Language Model Augmented Exercise Retrieval for Personalized Language Learning8 Feb 2024
- 2024DREditor: An Time-efficient Approach for Building a Domain-specific Dense Retrieval Model23 Jan 2024
- 2023Boot and Switch: Alternating Distillation for Zero-Shot Dense Retrieval27 Nov 2023
- 2023Chain-of-Factors Paper-Reviewer Matching23 Oct 2023
- 2023I3: Intent-Introspective Retrieval Conditioned on Instructions19 Aug 2023
- 2023ReGen: Zero-Shot Text Classification via Training Data Generation with Progressive Dense Retrieval18 May 2023
- 2023Synergistic Interplay between Search and Large Language Models for Information Retrieval12 May 2023
- 2022Precise Zero-Shot Dense Retrieval without Relevance Labels20 Dec 2022