arXiv:2510.01149cs.IR2025-10被引 18

小模型ModernVBERT让文档检索更快更便宜,性能还更强。

ModernVBERT: Towards Smaller Visual Document Retrievers

  • 用轻量级视觉语言模型直接处理文档截图,提升检索效率。
  • 250万参数模型在任务上超越10倍大的模型,支持CPU高效推理。
  • 优化注意力掩码、图像分辨率与对比学习策略,提升精度。

从海量文档中检索特定信息是现代AI的常见工业应用,尤其得益于检索增强生成(RAG)系统的发展。尽管传统神经文档检索模型仅在文本空间运行,但视觉文档检索(VDR)模型——将大型视觉语言解码器重用于嵌入模型,直接以页面截图为输入——因性能和索引延迟优势而日益流行。本文表明,虽成本低廉,但复用生成式模型会制约检索性能。通过受控实验,我们重新审视整个训练流程,提出一套改进视觉文档检索的系统性方法。关键发现包括注意力掩码、图像分辨率、模态对齐数据设置以及晚期交互对比目标的重要性。基于此,我们发布ModernVBERT,一个仅250M参数的紧凑视觉语言编码器,在文档检索任务微调后表现优于最大达10倍的现有模型,可在廉价CPU上实现高效推理,显著降低延迟与成本,同时保持强性能。模型、代码与数据已开源:https://huggingface.co/ModernVBERT。

原文摘要 · Abstract (English)

Retrieving specific information from a large corpus of documents is a prevalent industrial use case of modern AI, notably due to the popularity of Retrieval-Augmented Generation (RAG) systems. Although neural document retrieval models have historically operated exclusively in the text space, Visual Document Retrieval (VDR) models - large vision-language decoders repurposed as embedding models which directly work with page screenshots as inputs - are increasingly popular due to the performance and indexing latency gains they offer. In this work, we show that, while cost-efficient, this approach of repurposing generative models bottlenecks retrieval performance. Through controlled experiments, we revisit the entire training pipeline, and establish a principled recipe for improving visual document retrieval models. We notably measure the impact of attention masking, image resolution, modality alignment data regimes, and late interaction centered contrastive objectives which emerge as central performance factors. Building on these insights, we release ModernVBERT, a compact 250M-parameter vision-language encoder that outperforms recent models up to 10 times larger when fine-tuned on document retrieval tasks, enabling efficient inference on cheap CPU hardware and greatly reducing latency and costs while maintaining strong performance. Models, code and data are available at https://huggingface.co/ModernVBERT.

视觉文档轻量模型检索优化

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