HEED: Density-Weighted Residual Alignment for Hybrid Vision-Language Model Distillation
About
Distilling vision-language models into faster hybrid architectures, such as 3:1 Mamba-2/attention mixes, is now standard practice for making inference efficient. Aggregate benchmarks suggest that this works but they hide selective failures. When we distill Qwen3-VL-8B-Instruct into a 3:1 Mamba-2/attention hybrid, student model stays within 2 points of the teacher across visual reasoning benchmarks like MMStar, MMBench, and MMMU-Pro, while dropping 13 points on optical-character-recognition and document tasks. The student can still understand the scene but loses the fine-grained text needed to answer. We localize much of the failure to a specific kind of position. In a high-resolution image, most patches are sky, wall, or smooth texture, while a small fraction carries text, edges, object boundaries, or other local details. In a token-level diagnostic, the top 10% highest-density patches have 3.6$\times$ larger residual drift than the bottom 10% lowest-density patches and 3.5$\times$ larger teacher-masking answer contribution. Uniform weighting devotes many loss terms to low-information background patches, whereas sparse answer-bearing patches receive no special protection. The required intervention is minimal: we replace uniform residual alignment with density-weighted residual alignment, using patch self-dissimilarity as a training-free proxy for position importance. We call this HEED. Compared with normal end-to-end distillation, HEED increases performance by 8.7 points on OCRBench v2 and 5.13 points on a 10-benchmark average. The gain is realized on different teacher models and hybrid architectures. After standard post-training, the student reaches teacher-level performance on the 10-benchmark average with a 4.12$\times$ throughput and a 68% memory saving at 128k context, with no additional parameters and no inference-time cost.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Mathematical Reasoning | MathVista | Accuracy77.2 | 382 | |
| Reasoning | MMLU-Pro | MMLU-Pro Reasoning Score55 | 36 | |
| Math Reasoning | MathVista | Score76.2 | 30 | |
| General Reasoning | MMBench | Accuracy84.6 | 15 | |
| Reasoning | MMMU-Pro | Accuracy (Reasoning on MMMU-Pro)55.9 | 13 | |
| Fine-Grained Perception | OCRBench v2 | Score63.9 | 10 | |
| Fine-Grained Perception | ChartQA | Score88.6 | 10 | |
| Fine-Grained Perception | TextVQA | Score83.7 | 10 | |
| Fine-Grained Perception | AI2D | Score85.3 | 10 | |
| Reasoning | MMStar | MMStar Reasoning Score71.1 | 10 |