VC-OPD: Visual Counterfactual On-Policy Distillation for Grounded Vision-Language Reasoning
Organizations: College of Intelligence and Computing, Tianjin University, Tianjin, China
Abstract
GRPO-style reinforcement learning has advanced vision-language reasoning, but its sequence-level rewards provide limited credit assignment for individual reasoning tokens. On-policy distillation (OPD) addresses this issue by providing dense supervision on student-generated trajectories, yet standard OPD assigns similar importance to all teacher-preferred tokens, including template phrases and visually irrelevant continuations. We introduce Visual Counterfactual On-Policy Distillation (VC-OPD), a grounding-aware distillation method that prioritizes tokens whose correctness depends on image-specific evidence. After a teacher-supervised warm start that stabilizes response formats and reduces noisy rollouts, VC-OPD performs on-policy distillation with two teacher evaluations for each student trajectory: one under the original image and one under a counterfactual image with corrupted instance-specific evidence. The original-image teacher distribution determines the distillation target, while the original--counterfactual teacher gap estimates token-level visual dependence and softly reweights the loss. To avoid unstable response-level update scales, VC-OPD further applies mass-preserving normalization to the token weights. This design preserves the dense supervision of OPD while shifting learning capacity toward visually grounded reasoning tokens. Experiments on multimodal reasoning benchmarks show that VC-OPD improves visual reasoning performance and produces more effective token-level distillation.
Figures & tables
| Model | Geo3K | MVer. | MVist. | WeM. | MMMU | Logic | Count | Avg. |
|---|---|---|---|---|---|---|---|---|
| Qwen2.5-VL-3B-Instruct | 24.96 | 38.08 | 58.50 | 53.22 | 24.50 | 34.72 | 51.71 | 40.81 |
| 7B teacher 3B student | ||||||||
| + GRPO | 30.52 | 40.60 | 59.04 | 59.10 | 25.86 | 38.34 | 57.81 | 44.47 |
| + PAPO | 30.95 | 41.50 | 62.38 | 60.09 | 27.11 | 38.67 | 62.56 | 46.18 |
| + RKL | 32.12 | 40.75 | 62.08 | 60.23 | 28.40 | 39.41 | 63.50 | 46.64 |
| + VC-OPD | 34.35 | 41.90 | 64.10 | 61.46 | 32.20 | 42.40 | 65.68 | 48.87 |
| Variant | Warm | CF | Soft | Mass | Math Avg. | Vision Avg. | Overall Avg. |
|---|---|---|---|---|---|---|---|
| Vanilla OPD | – | – | – | – | 48.80 | 43.77 | 46.64 |
| w/o warm-start | – | ✓ | ✓ | ✓ | 49.37 | 45.55 | 47.73 |
| w/o counterfactual teacher | ✓ | – | – | ✓ | 49.42 | 44.63 | 47.37 |
| w/ hard mask | ✓ | ✓ | – | ✓ | 49.86 | 45.92 | 48.17 |
| w/o mass norm. | ✓ | ✓ | ✓ | – | 50.05 | 46.10 | 48.36 |
| VC-OPD | ✓ | ✓ | ✓ | ✓ | 50.45 | 46.76 | 48.87 |
| Method | Geo3K | LogicVista | ||||||
|---|---|---|---|---|---|---|---|---|
| 50 | 100 | 150 | 200 | 50 | 100 | 150 | 200 | |
| GRPO | 26.10 | 27.75 | 28.90 | 30.52 | 34.20 | 36.30 | 37.45 | 38.34 |
| RKL | 29.20 | 30.35 | 31.25 | 32.12 | 37.05 | 38.20 | 38.85 | 39.41 |
| VC-OPD | 31.80 | 32.45 | 33.20 | 34.35 | 39.10 | 40.55 | 41.45 | 42.40 |
| Visual ratio | Math | Vision | Overall |
|---|---|---|---|
| 20% | 49.92 | 46.18 | 48.32 |
| 10% ⋆ | 50.45 | 46.76 | 48.87 |
| 5% | 49.58 | 45.90 | 48.00 |
| Counterfactual | Configuration | Math Avg. | Vision Avg. | Overall Avg. |
|---|---|---|---|---|
| No counterfactual | RKL | 48.80 | 43.77 | 46.64 |
| Random dropout | – | 49.86 | 46.20 | 48.29 (+3.54%) |
| Patch dropout | 20% | 49.72 | 45.98 | 48.12 (+3.17%) |
| Patch dropout | 60% | 50.45 | 46.76 | 48.87 (+4.78%) |
| Patch dropout | 100% | 49.20 | 45.55 | 47.64 (+2.14%) |
| Gaussian blur | Kernel , | 50.06 | 46.35 | 48.47 (+3.92%) |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | Value |
|---|---|
| Student model | Qwen2.5-VL-3B-Instruct |
| Teacher model | Qwen2.5-VL-32B-Instruct |
| Optimizer | AdamW |
| Learning rate | |
| Batch size | 256 |
| Response format | <think> reasoning + \boxed{} answer |
| Hyperparameter | GRPO | VC-OPD |
|---|---|---|
| Optimizer | AdamW | AdamW |
| Learning rate | ||
| Batch size (rollouts) | 5 | 1 |
| Max response length | 8192 | 8192 |
| Max prompt length | 4096 | 4096 |
| Generation temperature | 1.0 | 1.0 |
| Scoring signal | P@5% | P@10% | Recall@10% | AUPRC |
|---|---|---|---|---|
| Random score | 11.8 | 11.6 | 10.1 | 12.0 |
| Token log-probability | 18.7 | 17.4 | 15.2 | 18.6 |
| Student–teacher gap | 36.5 | 31.8 | 27.7 | 34.2 |
| Factual–counterfactual gap | 58.9 | 52.1 | 45.4 | 57.6 |
| Model | Geo3K | MVer. | MVist. | WeM. | MMMU | Logic | Count | Avg. |
|---|---|---|---|---|---|---|---|---|
| Qwen3-VL-2B | ||||||||
| + Base | 28.20 | 39.50 | 59.80 | 56.60 | 27.50 | 36.80 | 55.10 | 43.36 |
| + RKL | 39.50 | 42.80 | 63.00 | 60.50 | 31.40 | 40.90 | 61.20 | 48.47 |
| + VC-OPD | 42.40 | 44.20 | 64.50 | 61.90 | 33.80 | 42.80 | 63.80 | 50.49 |
| (%) | +7.34 | +3.27 | +2.38 | +2.31 | +7.64 | +4.65 | +4.25 | +4.16 |
| Qwen3-VL-4B | ||||||||
| Model | Geo3K | MVer. | MVist. | WeM. | MMMU | Logic | Count | Avg. |
|---|---|---|---|---|---|---|---|---|
| InternVL3.5-2B | ||||||||
| + Base | 30.80 | 39.40 | 60.50 | 28.40 | 31.20 | 41.00 | 56.20 | 41.07 |
| + RKL | 37.90 | 42.10 | 64.90 | 35.20 | 34.00 | 39.80 | 61.00 | 44.99 |
| + VC-OPD | 40.80 | 44.20 | 66.10 | 37.40 | 35.80 | 42.30 | 63.10 | 47.10 |
| (%) | +7.65 | +4.99 | +1.85 | +6.25 | +5.29 | +6.28 | +3.44 | +4.70 |
| InternVL3.5-4B | ||||||||
| Target Accuracy | 29% | 30% | 31% | 32% | 33% | 34% |
|---|---|---|---|---|---|---|
| RKL | 2.5 | 4.0 | 7.0 | 14.5 | – | – |
| VC-OPD | 1.5 | 2.8 | 5.0 | 7.2 | 14.4 | 17.4 |
| Method | MathVista | MathVerse | We-Math | LogicVista | Avg. |
|---|---|---|---|---|---|
| DeepEyes | 70.10 | 47.30 | 38.90 | 47.70 | 51.00 |
| Thyme-VL | 70.00 | 39.10 | 39.30 | 49.00 | 49.35 |
| VC-OPD | 70.90 | 48.90 | 68.10 | 48.60 | 59.13 |
VLM Distillation
Vision-language model distillation transfers multimodal capabilities to smaller, faster models for efficient deployment.
See all 120 papers