Why 1 + 1 < 1 in Visual Token Pruning: Beyond Naive Integration via Multi-Objective Balanced Covering
Yangfu Li, Hongjian Zhan, Tianyi Chen, Qi Liu, Yue Lu
TL;DR
This work tackles visual token pruning in multimodal LLMs by modeling the trade-off between visual preservation and prompt alignment through a closed-form error bound based on the Hausdorff distance $d_H$, and an $\\ extepsilon$-covering framework to quantify prompt-visual coupling. It introduces Multi-Objective Balanced Covering (MoB), a training-free pruning method that recasts pruning as bi-objective covering with budgets $K_p$ and $K-K_p$, solved via greedy radius trading using a two-stage process (k-fold NN covering for prompts and FPS for visuals) and backed by provable guarantees with complexity $T_{MoB}=\mathcal{O}(N(L+K)d)$. Empirically, MoB preserves up to $96.4\%$ of performance with only $11.1\%$ of visual tokens on LLaVA-1.5-7B and accelerates LLaVA-Next-7B by $1.3$–$1.5\times$, with strong generalization to Qwen2-VL and Video-LLaVA across 14 benchmarks. The approach offers a scalable, training-free framework for adaptive pruning in advanced MLLMs, with potential applicability to other redundancy-heavy domains.
Abstract
Existing visual token pruning methods target prompt alignment and visual preservation with static strategies, overlooking the varying relative importance of these objectives across tasks, which leads to inconsistent performance. To address this, we derive the first closed-form error bound for visual token pruning based on the Hausdorff distance, uniformly characterizing the contributions of both objectives. Moreover, leveraging $ε$-covering theory, we reveal an intrinsic trade-off between these objectives and quantify their optimal attainment levels under a fixed budget. To practically handle this trade-off, we propose Multi-Objective Balanced Covering (MoB), which reformulates visual token pruning as a bi-objective covering problem. In this framework, the attainment trade-off reduces to budget allocation via greedy radius trading. MoB offers a provable performance bound and linear scalability with respect to the number of input visual tokens, enabling adaptation to challenging pruning scenarios. Extensive experiments show that MoB preserves 96.4% of performance for LLaVA-1.5-7B using only 11.1% of the original visual tokens and accelerates LLaVA-Next-7B by 1.3-1.5$\times$ with negligible performance loss. Additionally, evaluations on Qwen2-VL and Video-LLaVA confirm that MoB integrates seamlessly into advanced MLLMs and diverse vision-language tasks.
