Papers
arxiv:2608.10636

DistilVDR: A Compact End-to-End Visual Document Retriever via Dual-Student Distillation

Published on Aug 11
· Submitted by
Zhuchenyang Liu
on Aug 12
Authors:
,
,

Abstract

DistilVDR is a compact 524M vision-document retriever distilled from an 8B teacher using cosine alignment without relevance labels, achieving near-teacher accuracy with far smaller indexes and faster indexing.

Visual document retrieval (VDR) is dominated by multi-billion-parameter models that are slow to index at full corpus scale and expensive to serve. Prior compression routes either train a smaller multi-vector encoder from scratch or distil only the query side; neither yields a compact single-vector retriever end-to-end. We present DistilVDR, a 524M end-to-end VDR system distilled bilaterally from a single 8B vision-language teacher under a pointwise cosine alignment loss. All supervision comes from the frozen teacher's embedding space, which was itself trained with relevance supervision, so the student objective needs no relevance labels, negative sampling, or contrastive term. We match VDR's text-query and image-document input asymmetry with an asymmetric encoder-only student that concentrates visual capacity on the document side and keeps the query side at 70M parameters. We release two variants that share the same encoders and training and differ only in the document encoder's visual-tile budget: DistilVDR-HiRes attains 61.74 average NDCG@5 on ViDoRe v1+v2+v3 (86.9% of the 8B teacher) and leads every reproduced sub-1B baseline on the high-resolution-sensitive v3 benchmark, while DistilVDR-Fast attains 59.98 at a 3 times smaller visual-token budget. Both variants store one million documents in a 15.6 times smaller index than the strongest sub-1B multi-vector baseline and index the corpus an order of magnitude faster. The code is available at https://github.com/Ryenhails/NanoVDR.

Community

Paper author Paper submitter

Quick follow-up on NanoVDR: we distilled an 8B visual document retriever down to a 457M document tower and a 70M query tower by simply reproducing the frozen Qwen3-VL-Embedding-8B page embeddings under cosine distance. No relevance labels, no negative mining, no contrastive loss.

Two document tower variants, each paired with the same 70M text-only query tower and averaged over all 22 ViDoRe datasets: HiRes keeps 86.9% of the teacher's NDCG@5 and indexes at 37 pages/s, Fast keeps 84.4% at 99 pages/s, against the teacher's 5.4. Both store 16.4 GB per million pages instead of the 256 GB their multi-vector peers need.

We released the training https://huggingface.co/spaces/nanovdr/distilling-the-document-tower and models https://huggingface.co/nanovdr/NanoVDR-D-Fast-Qwen3VL8B-4096, https://huggingface.co/nanovdr/NanoVDR-D-HiRes-Qwen3VL8B-4096, https://huggingface.co/nanovdr/NanoVDR-Q-DistilBERT-Qwen3VL8B-4096-ML on our NanoVDR space.

Feel free to explore!

Congrats ! I quite liked your nanoVDR work excited to read through this one :)

·

Very happy to hear that! The ViDoRe series, especially v3 is a great infra and inspired a lot of the current work. Hope you enjoy the read :)

Sign up or log in to comment

Models citing this paper 3

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2608.10636 in a dataset README.md to link it from this page.

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.