English

Sculpting the Vector Space: Towards Efficient Multi-Vector Visual Document Retrieval via Prune-then-Merge Framework

Computation and Language 2026-04-21 v2 Computer Vision and Pattern Recognition Information Retrieval

Abstract

Visual Document Retrieval (VDR), which aims to retrieve relevant pages within vast corpora of visually-rich documents, is of significance in current multimodal retrieval applications. The state-of-the-art multi-vector paradigm excels in performance but suffers from prohibitive overhead, a problem that current efficiency methods like pruning and merging address imperfectly, creating a difficult trade-off between compression rate and feature fidelity. To overcome this dilemma, we introduce Prune-then-Merge, a novel two-stage framework that synergizes these complementary approaches. Our method first employs an adaptive pruning stage to filter out low-information patches, creating a refined, high-signal set of embeddings. Subsequently, a hierarchical merging stage compresses this pre-filtered set, effectively summarizing semantic content without the noise-induced feature dilution seen in single-stage methods. Extensive experiments on 29 VDR datasets demonstrate that our framework consistently outperforms existing methods, significantly extending the near-lossless compression range and providing robust performance at high compression ratios.

Keywords

Cite

@article{arxiv.2602.19549,
  title  = {Sculpting the Vector Space: Towards Efficient Multi-Vector Visual Document Retrieval via Prune-then-Merge Framework},
  author = {Yibo Yan and Mingdong Ou and Yi Cao and Xin Zou and Jiahao Huo and Shuliang Liu and James Kwok and Xuming Hu},
  journal= {arXiv preprint arXiv:2602.19549},
  year   = {2026}
}

Comments

Accepted by The 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026, Findings)

R2 v1 2026-07-01T10:46:56.531Z