This paper addresses the inefficiency of processing visual tokens in multimodal large language models by improving token pruning strategies. While prior work uses attention from a fixed middle layer to identify important tokens, the optimal layer varies significantly per sample. The authors propose learning to predict the appropriate middle-layer attention dynamically, rather than relying on a static layer selection.
- Fixed middle-layer attention is suboptimal because relevance varies by input sample
- Dynamic prediction of attention layers improves visual token pruning accuracy
- Reduces inference cost without sacrificing multimodal reasoning performance
- Bypasses the need for computationally expensive full attention maps