Llama Nemoretriever Colembed: Top-Performing Text-Image Retrieval Model
About
Motivated by the growing demand for retrieval systems that operate across modalities, we introduce llama-nemoretriever-colembed, a unified text-image retrieval model that delivers state-of-the-art performance across multiple benchmarks. We release two model variants, 1B and 3B. The 3B model achieves state of the art performance, scoring NDCG@5 91.0 on ViDoRe V1 and 63.5 on ViDoRe V2, placing first on both leaderboards as of June 27, 2025. Our approach leverages the NVIDIA Eagle2 Vision-Language model (VLM), modifies its architecture by replacing causal attention with bidirectional attention, and integrates a ColBERT-style late interaction mechanism to enable fine-grained multimodal retrieval in a shared embedding space. While this mechanism delivers superior retrieval accuracy, it introduces trade-offs in storage and efficiency. We provide a comprehensive analysis of these trade-offs. Additionally, we adopt a two-stage training strategy to enhance the model's retrieval capabilities.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Visual document retrieval | ViDoRe V2 | Avg nDCG@563.5 | 39 | |
| Multimodal Question Answering | WebQA Average | F1 Score50.8 | 32 | |
| Visual document retrieval | ViDoRe V3 | HR58.69 | 23 | |
| Visual document retrieval | MIRACL Vision | Chinese0.4355 | 17 | |
| Multimodal Question Answering | WebQA Text-grounded n=2,455 (test) | F1 Score48.6 | 16 | |
| Multimodal Question Answering | WebQA Image | F1 Score11.3 | 16 | |
| Multimodal Question Answering | WebQA Text | F1 Score52.1 | 16 | |
| Multimodal Question Answering | WebQA Image-grounded n=2,511 (test) | F1 Score53 | 16 | |
| Visual document retrieval | JinaVDR | nDCG@1067.8 | 15 | |
| Visual document retrieval | VisRAG | -- | 13 |