Papers
arxiv:2609.34899

ColNanoVDR: Document-Free Query Distillation for Multi-Vector Visual Document Retrieval via Optimal Transport

Published on Sep 28
· Submitted by
Zhuchenyang Liu
on Sep 29
Authors:
,
,

Abstract

Multi-vector retrievers built on vision-language models lead visual document retrieval (VDR), but they run a multi-billion-parameter query encoder on every search. Distilling this encoder into a small student that queries the teacher's existing index would remove the bottleneck. The standard recipe, however, matches the teacher's MaxSim scores and so requires encoding and caching every training page, which can reach terabytes of page tokens. NanoVDR avoids pages entirely by training on the teacher's query embeddings alone, but only for single-vector retrievers. We present ColNanoVDR, to our knowledge the first framework to bring this document-free distillation to multi-vector VDR. Its objective, OTW (Optimal Transport with Learned Weights), aligns the student's query tokens with the teacher's by entropic optimal transport, with a learned weight for each student token, and needs no correspondence between the two tokenizations. We prove that the resulting alignment cost bounds the MaxSim score difference on every page. Distilled from five state-of-the-art teachers, the 149M text-only students retain about 95% of their teachers' NDCG@5 on ViDoRe v1-v3 while encoding queries up to 26x faster. Under identical training, OTW matches score distillation while encoding no page and reading 12.6x less cached teacher data.

Community

Paper author Paper submitter

ColNanoVDR brings NanoVDR-style document-free distillation to multi-vector visual document retrieval. We distill multi-billion-parameter multi-vector VLM retrievers (4.5B–8.8B) into a 149M text-only query encoder that scores directly against the teacher's existing page index, with no re-indexing. OTW (Optimal Transport with Learned Weights) aligns the student's query tokens with the teacher's, so training reads no pages at all.

  • ViDoRe v3: 93.0–96.0% of the teacher's NDCG@5 across five teachers (up to 60.1 with ColVec1.1-8b)
  • 26× faster query encoding on a single CPU thread
  • Matches score distillation while reading 12.6× less cached teacher data

Code: https://github.com/Ryenhails/NanoVDR
Checkpoints (all five students): https://huggingface.co/nanovdr

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2609.34899
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 6

Browse 6 models citing this paper

Datasets citing this paper 1

Spaces citing this paper 1

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.