Triage: Hierarchical Visual Budgeting for Efficient Video Reasoning in Vision-Language Models
Anmin Wang, Nan Zhang, Wei Tao, Xiaoyang Qu, Guokuan Li, Jiguang Wan, Jianzong Wang
TL;DR
Triage addresses the heavy computational burden of video reasoning in Vision-Language Models by reframing inference as a hierarchical budgeting problem. It first performs Frame-Level Budgeting to select a compact set of keyframes using a frame importance score that combines visual dynamics and query relevance, producing a prior for downstream processing. It then executes Token-Level Budgeting, allocating a constrained token budget into high-relevance Core Tokens and diverse Context Tokens via a batched MMR method guided by cross-attention, with budget distribution across frames informed by frame importance. Experiments across multiple backbones and benchmarks show that Triage yields significant speedups and memory savings while maintaining or surpassing state-of-the-art baselines, demonstrating a practical, plug-and-play solution for efficient video reasoning in VLMs.
Abstract
Vision-Language Models (VLMs) face significant computational challenges in video processing due to massive data redundancy, which creates prohibitively long token sequences. To address this, we introduce Triage, a training-free, plug-and-play framework that reframes video reasoning as a resource allocation problem via hierarchical visual budgeting. Its first stage, Frame-Level Budgeting, identifies keyframes by evaluating their visual dynamics and relevance, generating a strategic prior based on their importance scores. Guided by this prior, the second stage, Token-Level Budgeting, allocates tokens in two phases: it first secures high-relevance Core Tokens, followed by diverse Context Tokens selected with an efficient batched Maximal Marginal Relevance (MMR) algorithm. Extensive experiments demonstrate that Triage improves inference speed and reduces memory footprint, while maintaining or surpassing the performance of baselines and other methods on various video reasoning benchmarks.
