arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.01649cs.CVcs.AI
收藏

面向贡献的视频理解强化学习token压缩

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

Yinchao Ma, Qiang Zhou, Zhibin Wang, Xianing Chen, Hanqing Yang, Jun Song, Bo Zheng

AI总结:

CaCoVID通过强化学习优化token选择策略,以提高视频理解任务的效率。

摘要:

视频大语言模型在视频理解任务中展现出了显著的能力。然而,视频token的冗余性在推理过程中引入了显著的计算开销,限制了其实际部署。许多压缩算法被提出以优先保留具有最高注意力分数的特征以最小化注意力计算中的扰动。然而,注意力分数与它们对正确答案的实际贡献之间的关系仍然模糊。为了解决上述限制,我们提出了一种新的面向贡献的视频理解强化学习token压缩算法(CaCoVID),该算法基于token对正确预测的贡献来显式优化token选择策略。首先,我们引入了一个基于强化学习的框架,该框架优化了一个策略网络以选择对正确预测贡献最大的视频token组合。这一范式将重点从被动的token保留转移到主动发现最优压缩token组合上。其次,我们提出了一种具有在线组合空间采样的组合策略优化算法,这大大减少了视频token组合的探索空间并加速了策略优化的收敛速度。在多种视频理解基准上的广泛实验证明了CaCoVID的有效性。代码可在https://github.com/LivingFutureLab/CaCoVID上获取。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑