Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Yihao Liang, Niraj K. Jha• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMathVista
Accuracy77.2
382
ReasoningMMLU-Pro
MMLU-Pro Reasoning Score55
36
Math ReasoningMathVista
Score76.2
30
General ReasoningMMBench
Accuracy84.6
15
ReasoningMMMU-Pro
Accuracy (Reasoning on MMMU-Pro)55.9
13
Fine-Grained PerceptionOCRBench v2
Score63.9
10
Fine-Grained PerceptionChartQA
Score88.6
10
Fine-Grained PerceptionTextVQA
Score83.7
10
Fine-Grained PerceptionAI2D
Score85.3
10
ReasoningMMStar
MMStar Reasoning Score71.1
10
Showing 10 of 16 rows

Other info

Follow for update