Towards Storage-Efficient Visual Document Retrieval: An Empirical Study on Reducing Patch-Level Embeddings

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ma, Yubo, Li, Jinsong, Zang, Yuhang, Wu, Xiaobao, Dong, Xiaoyi, Zhang, Pan, Cao, Yuhang, Duan, Haodong, Wang, Jiaqi, Cao, Yixin, Sun, Aixin
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913878013640704
author Ma, Yubo
Li, Jinsong
Zang, Yuhang
Wu, Xiaobao
Dong, Xiaoyi
Zhang, Pan
Cao, Yuhang
Duan, Haodong
Wang, Jiaqi
Cao, Yixin
Sun, Aixin
author_facet Ma, Yubo
Li, Jinsong
Zang, Yuhang
Wu, Xiaobao
Dong, Xiaoyi
Zhang, Pan
Cao, Yuhang
Duan, Haodong
Wang, Jiaqi
Cao, Yixin
Sun, Aixin
contents Despite the strong performance of ColPali/ColQwen2 in Visualized Document Retrieval (VDR), it encodes each page into multiple patch-level embeddings and leads to excessive memory usage. This empirical study investigates methods to reduce patch embeddings per page at minimum performance degradation. We evaluate two token-reduction strategies: token pruning and token merging. Regarding token pruning, we surprisingly observe that a simple random strategy outperforms other sophisticated pruning methods, though still far from satisfactory. Further analysis reveals that pruning is inherently unsuitable for VDR as it requires removing certain page embeddings without query-specific information. Turning to token merging (more suitable for VDR), we search for the optimal combinations of merging strategy across three dimensions and develop Light-ColPali/ColQwen2. It maintains 98.2% of retrieval performance with only 11.8% of original memory usage, and preserves 94.6% effectiveness at 2.8% memory footprint. We expect our empirical findings and resulting Light-ColPali/ColQwen2 offer valuable insights and establish a competitive baseline for future research towards efficient VDR.
format Preprint
id arxiv_https___arxiv_org_abs_2506_04997
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Towards Storage-Efficient Visual Document Retrieval: An Empirical Study on Reducing Patch-Level Embeddings
Ma, Yubo
Li, Jinsong
Zang, Yuhang
Wu, Xiaobao
Dong, Xiaoyi
Zhang, Pan
Cao, Yuhang
Duan, Haodong
Wang, Jiaqi
Cao, Yixin
Sun, Aixin
Information Retrieval
Computation and Language
Despite the strong performance of ColPali/ColQwen2 in Visualized Document Retrieval (VDR), it encodes each page into multiple patch-level embeddings and leads to excessive memory usage. This empirical study investigates methods to reduce patch embeddings per page at minimum performance degradation. We evaluate two token-reduction strategies: token pruning and token merging. Regarding token pruning, we surprisingly observe that a simple random strategy outperforms other sophisticated pruning methods, though still far from satisfactory. Further analysis reveals that pruning is inherently unsuitable for VDR as it requires removing certain page embeddings without query-specific information. Turning to token merging (more suitable for VDR), we search for the optimal combinations of merging strategy across three dimensions and develop Light-ColPali/ColQwen2. It maintains 98.2% of retrieval performance with only 11.8% of original memory usage, and preserves 94.6% effectiveness at 2.8% memory footprint. We expect our empirical findings and resulting Light-ColPali/ColQwen2 offer valuable insights and establish a competitive baseline for future research towards efficient VDR.
title Towards Storage-Efficient Visual Document Retrieval: An Empirical Study on Reducing Patch-Level Embeddings
topic Information Retrieval
Computation and Language
url https://arxiv.org/abs/2506.04997