arXiv:2602.03992cs.IR2026-02被引 8

Nemotron ColEmbed V2 提升文档检索精度,支持大模型高效处理图文内容。

Nemotron ColEmbed V2: Top-Performing Late Interaction Embedding Models for Visual Document Retrieval

  • 采用晚期交互机制融合图文特征,提升视觉文档匹配能力。
  • 80亿参数模型在ViDoRe V3榜单上达63.42的平均NDCG@10,居榜首。
  • 适合需要高精度图文检索的RAG系统开发者使用。

检索增强生成(RAG)系统在生成应用中广受欢迎,通过注入外部知识为语言模型提供支持。企业希望利用其海量文档(如PDF、演示文稿)构建RAG管道,其中检索环节至关重要。密集检索是主流方法,通过嵌入模型生成与相关文本嵌入更接近的用户查询表示。近年来,基于视觉语言模型(VLM)的嵌入模型因其保留视觉信息并简化索引流程而受到青睐。为应对日益增长的视觉文档检索需求,我们推出Nemotron ColEmbed V2系列模型,在ViDoRe基准测试中表现领先。该系列包含30亿、40亿和80亿参数三种版本,分别基于NVIDIA Eagle 2(Llama 3.2 3B主干)、Qwen3-VL-4B-Instruct 和 Qwen3-VL-8B-Instruct。截至2026年2月3日,80亿参数模型在ViDoRe V3排行榜上排名第一,平均NDCG@10达到63.42。我们介绍了数据处理、训练与后训练阶段的核心技术,包括基于聚类采样、难例挖掘、双向注意力、晚期交互和模型融合等方法。同时讨论了晚期交互机制带来的计算与存储挑战,并实验验证了低维嵌入在准确率与存储成本间的平衡策略。

原文摘要 · Abstract (English)

Retrieval-Augmented Generation (RAG) systems have been popular for generative applications, powering language models by injecting external knowledge. Companies have been trying to leverage their large catalog of documents (e.g. PDFs, presentation slides) in such RAG pipelines, whose first step is the retrieval component. Dense retrieval has been a popular approach, where embedding models are used to generate a dense representation of the user query that is closer to relevant content embeddings. More recently, VLM-based embedding models have become popular for visual document retrieval, as they preserve visual information and simplify the indexing pipeline compared to OCR text extraction. Motivated by the growing demand for visual document retrieval, we introduce Nemotron ColEmbed V2, a family of models that achieve state-of-the-art performance on the ViDoRe benchmarks. We release three variants - with 3B, 4B, and 8B parameters - based on pre-trained VLMs: NVIDIA Eagle 2 with Llama 3.2 3B backbone, Qwen3-VL-4B-Instruct and Qwen3-VL-8B-Instruct, respectively. The 8B model ranks first on the ViDoRe V3 leaderboard as of February 03, 2026, achieving an average NDCG@10 of 63.42. We describe the main techniques used across data processing, training, and post-training - such as cluster-based sampling, hard-negative mining, bidirectional attention, late interaction, and model merging - that helped us build our top-performing models. We also discuss compute and storage engineering challenges posed by the late interaction mechanism and present experiments on how to balance accuracy and storage with lower dimension embeddings.

视觉检索嵌入模型RAGVLM

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。