Contribution-aware Token Compression for Efficient Video Understanding via Reinforcement Learning

TL;DR

This paper introduces CaCoVID, a reinforcement learning-based method for selecting video tokens that contribute most to accurate understanding, significantly reducing computational costs while maintaining performance.

Contribution

The paper presents a novel reinforcement learning framework with a combinatorial policy optimization algorithm for contribution-aware video token compression.

Findings

Effective token selection improves video understanding accuracy.

Reduces computational overhead in video models.

Accelerates policy convergence with online sampling.

Abstract

Video large language models have demonstrated remarkable capabilities in video understanding tasks. However, the redundancy of video tokens introduces significant computational overhead during inference, limiting their practical deployment. Many compression algorithms are proposed to prioritize retaining features with the highest attention scores to minimize perturbations in attention computations. However, the correlation between attention scores and their actual contribution to correct answers remains ambiguous. To address the above limitation, we propose a novel \textbf{C}ontribution-\textbf{a}ware token \textbf{Co}mpression algorithm for \textbf{VID}eo understanding (\textbf{CaCoVID}) that explicitly optimizes the token selection policy based on the contribution of tokens to correct predictions. First, we introduce a reinforcement learning-based framework that optimizes a policy network to select video token combinations with the greatest contribution to correct predictions. This paradigm shifts the focus from passive token preservation to active discovery of optimal compressed token combinations. Secondly, we propose a combinatorial policy optimization algorithm with online combination space sampling, which dramatically reduces the exploration space for video token combinations and accelerates the convergence speed of policy optimization. Extensive experiments on diverse video understanding benchmarks demonstrate the effectiveness of CaCoVID. Codes are available at https://github.com/LivingFutureLab/CaCoVID.