arXiv:2606.23539cs.CV2026-06被引 1

用关键词快速筛选文档页,再精准匹配,速度提升数倍。

LightSTAR: Efficient Visual Document Retrieval via Lightweight Selection with Vision-Adaptive Refinement

论文配图:LightSTAR: Efficient Visual Document Retrieval via Lightweight Selection with Vision-Adaptive Refinement
图 1 · 摘自论文原文
  • 不用大模型处理每页,先用关键词和视觉特征粗筛候选页
  • 在少数候选页上做精细匹配,准确率达当前最佳
  • 适合需要高速检索的文档系统,如智能办公、档案管理

视觉文档检索需从大规模多模态语料中快速定位相关页面。现有基于多模态大模型(MLLM)的方法虽精度高,但因对每页都进行密集编码,计算开销巨大。我们观察到用户查询通常以关键词为核心,包含语义丰富的词汇,这些词往往直接出现在相关页面的可见文本中,可作为高效候选筛选线索。基于此,我们提出 LightSTAR:首先通过无需大模型的视觉选择,利用内容感知的查询编码聚焦关键信息,并使用无需大模型的视觉嵌入生成高召回候选集;其次通过视觉自适应语义精炼,在这些候选页上进行细粒度语义匹配,采用区域级特征融合结合文本与版式信息,以硬度感知对比目标优化。实验表明,LightSTAR 在保持最先进检索精度的同时,将端到端延迟降低数倍,为视觉文档检索中的精度-效率权衡提供了实用解决方案。代码已开源。

原文摘要 · Abstract (English)

Visual document retrieval requires rapidly locating relevant pages from large multi-modal corpora in response to user queries. While recent methods powered by Multi-modal Large Language Models (MLLMs) show competitive accuracy, they suffer from prohibitive computational costs by applying intensive MLLM encoding to every single page. Meanwhile, we observe that user queries are typically keyword-anchored, containing semantically rich words that are expected to appear directly in the visible text of relevant pages, offering an efficient cue for quickly narrowing down candidate pages. Building on this insight, we propose LightSTAR, an efficient framework that decomposes visual document retrieval into: 1) LLM-free Visual Selection, which utilizes content-grounded query encoding to focus on informative words and employs LLM-free visual embeddings to produce a high-recall candidate set; and 2) Vision-adaptive Semantic Refinement, which further performs fine-grained semantic matching exclusively on these top candidates via adaptive region-wise feature fusion to effectively combine textual and layout cues, optimized through a hardness-aware contrastive objective. Experimental results demonstrate that LightSTAR achieves state-of-the-art retrieval accuracy while reducing end-to-end latency by several-fold, offering a highly practical solution to the accuracy-efficiency trade-off in visual document retrieval. Code is available at https://github.com/bokufa/LightSTAR.

文档检索视觉理解轻量化多模态

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