Skip-It? Theoretical Conditions for Layer Skipping in Vision-Language Models
Max Hartman, Vidhata Jayaraman, Moulik Choraria, Akhil Bhimaraju, Lav R. Varshney
TL;DR
The paper addresses the heavy compute burden of vision-language models and proposes a principled framework to identify layer- and token-level redundancy that enables skipping. It develops an information- and learning-theoretic approach to define functional and informational redundancy, and links these to geometric and cross-attention measures to locate skip-viable layers. Empirical validation across multiple VLMs shows that skipping layers with high redundancy can accelerate inference with minimal accuracy loss, while skipping non-redundant layers degrades performance. By connecting redundancy concepts to cross-attention analysis and partial information decomposition, the work provides a unified basis for several efficient inference techniques in VLMs.
Abstract
Vision-language models (VLMs) achieve incredible performance across a wide range of tasks, but their large size makes inference costly. Recent work shows that selectively skipping VLM layers can improve efficiency with minimal performance loss or even performance improvements. However, this technique remains underused due to the limited understanding of when layer skipping is beneficial. In this paper, we develop a framework that uses information and learning theory to characterize the conditions under which layer skipping enhances efficiency without sacrificing performance. Motivated by these observations, we analyze the evolution of the VLM's hidden representations through the LLM backbone and show that layers with large redundancy as predicted by our framework coincide with those skipped by popular layer-skipping methods in practice, providing a unified theoretical scaffolding for multiple efficient inference techniques. Our experiments demonstrate that skipping such layers yields faster inference that preserves performance, and also show that applying skipping outside these conditions leads to model degradation.
