DistilVDR: A Compact End-to-End Visual Document Retriever via Dual-Student Distillation
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
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!
Models citing this paper 3
nanovdr/NanoVDR-D-HiRes-Qwen3VL8B-4096
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 1
Collections including this paper 0
No Collection including this paper