Visual-Advantage On-Policy Distillation for Vision-Language Models
Summary
Visual-Advantage On-Policy Distillation (VA-OPD) is a novel method addressing the challenge of strengthening visual reliance in vision-language models (VLMs) during on-policy knowledge distillation. Standard distillation often improves output quality but fails to make students depend on fine-grained visual input. Researchers introduce "visual advantage" (VA), a token-level log-probability difference indicating teacher reliance on visual detail. VA is sparse, with the top 10% of tokens carrying ~93% of visual supervision. VA-OPD leverages this by reweighting rollouts based on trajectory-averaged VA and grouping tokens into high-VA and low-VA categories for separate KL averaging. Evaluated on Qwen3-VL models (4B, 8B, 32B teachers to 2B student) across Geometry3K and ViRL39K datasets, VA-OPD consistently outperforms standard on-policy distillation on eight benchmarks, showing gains up to +3.8 on Math Avg and +2.5 on Visual Avg, with improvements scaling with teacher size and data.
Key takeaway
For AI Scientists and Machine Learning Engineers optimizing VLM distillation, you should adopt Visual-Advantage On-Policy Distillation (VA-OPD) to genuinely enhance your student models' reliance on fine-grained visual details. This method, by focusing distillation on vision-critical tokens identified via "visual advantage," prevents the dilution of crucial visual signals. Implementing VA-OPD can yield significant accuracy improvements, up to +3.8 Math Avg and +2.5 Visual Avg, especially on vision-intensive tasks, ensuring your models learn to "read the image" rather than just mimic text.
Key insights
Visual supervision in VLM on-policy distillation is sparse, concentrated in few tokens, and diluted by uniform KL averaging.
Principles
- Visual advantage (VA) quantifies token-level visual dependency.
- High-VA tokens carry the primary visual supervision signal.
- Dilution of visual signal hinders VLM visual reliance.
Method
VA-OPD reweights student rollouts by trajectory-averaged VA and applies token-level KL averaging separately within high-VA and low-VA groups.
In practice
- Use VA to identify vision-critical tokens in VLM outputs.
- Prioritize distillation signal on high-VA tokens.
- Degrade images (e.g., 10% resolution) to compute VA.
Topics
- Vision-Language Models
- Knowledge Distillation
- On-Policy Learning
- Visual Advantage
- Qwen3-VL
- Mathematical Reasoning
Best for: Research Scientist, AI Scientist, Machine Learning Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.