Understanding Pruning Regimes in Vision-Language Models Through Domain-Aware Layer Selection
Organizations: Ohio State University
Abstract
Transformer-based vision-language models (VLMs) contain substantial depth redundancy, yet the effect of removing specific decoder layers remains poorly understood, especially for domains that require tight coupling between perception and multi-step reasoning. We study structured decoder layer pruning through the lens of domain-aware activation similarity, measuring how strongly each layer transforms representations for math versus non-math inputs. This yields simple math-aware, non-math-aware, and mixed ranking criteria that identify layers whose input-output activations change least within a target domain. Across two state-of-the-art VLMs and a broad suite of math and general multimodal benchmarks, we uncover a consistent three-regime structure: at low pruning budgets, performance is highly sensitive to which layers are removed; at moderate budgets, methods converge as structural damage accumulates; and at high budgets, structural continuity dominates, favoring spacing-aware strategies. Our domain-aware rankings achieve the strongest stability in the ranking-sensitive regime, while matching or exceeding structure-aware baselines at larger budgets. These results provide a clearer picture of how depth contributes to domain-specific behavior in VLMs and offer a practical, interpretable approach to reducing model depth without sacrificing essential mathematical or general vision-language capabilities.
Figures & tables
| Dataset | Size | Purpose |
|---|---|---|
| Snapask | 300K total (100K train; 2,540 eval) | Real-world homework photos with CoT solutions; primary math supervision. |
| LLaVA-OneVision | 105K total (100K train; 5K eval) | Broad multimodal instruction tuning: charts, diagrams, scenes, VQA. |
| NuminaMath-CoT | 500 eval samples | High-difficulty math reasoning; domain-shift robustness. |
| ChartQA | 32.7K questions (2,500 eval) | Reasoning over charts and structured data visualizations. |
| ScienceQA | 21,208 total (2,017 eval) | Multimodal science questions with diagrams and multi-step reasoning. |
| RealWorldQA | 765 images (full eval) | Real-world spatial and geometric reasoning. |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | 2B Base | 2B Base+SFT | 4B Base | 4B Base+SFT |
|---|---|---|---|---|
| Snapask | 40.47 | 32.83 | 50.75 | 44.96 |
| NuminaMath | 63.45 | 52.61 | 73.49 | 64.06 |
| LLaVA-OneVision | 59.12 | 62.00 | 67.26 | 68.00 |
| ChartQA | 79.72 | 32.60 | 84.00 | 35.12 |
| RealWorldQA | 65.23 | 66.93 | 71.50 | 72.29 |
| ScienceQA | 86.47 | 87.80 | 92.71 | 93.46 |