用强化学习优化视频采样,让大模型更懂长视频内容
TSPO: Temporal Sampling Policy Optimization for Long-form Video Language Understanding
- 设计可训练的事件感知采样代理,基于概率选关键帧
- 在多个长视频基准上超越现有方法,性能领先12.3%以上
- 适用于各类先进视频-语言模型,可直接迁移使用
多模态大语言模型(MLLMs)在视觉-语言任务中取得显著进展,但在处理长时视频输入时仍面临上下文长度限制和训练成本高的问题。由于需要在输入前对视频进行稀疏帧采样,而现有的采样方式缺乏可训练性与可微性,导致难以优化。为此,我们提出时间采样策略优化(TSPO),通过强化学习提升长视频理解能力。首先,设计一个可训练的事件感知时间代理,捕捉事件与查询的相关性,实现概率化关键帧选择;其次,提出基于强化学习的TSPO框架,将关键帧选择与语言生成联合建模,支持端到端的群体相对优化;进一步,构建双风格长视频训练数据管道,平衡全局时序理解与关键片段定位;最后,引入基于规则的答案准确率与时间定位奖励机制,优化采样策略。大量实验表明,TSPO在多个长视频理解基准上达到当前最优性能,并具备跨不同前沿视频-语言模型的可迁移性。代码已开源。
原文摘要 · Abstract (English)
Multimodal Large Language Models (MLLMs) have demonstrated significant progress in vision-language tasks, yet they still face challenges when processing long-duration video inputs. The limitation arises from MLLMs' context limit and training costs, necessitating sparse frame sampling before feeding videos into MLLMs. However, building a trainable sampling method remains challenging due to the unsupervised and non-differentiable nature of sparse frame sampling in Video-MLLMs. To address these problems, we propose Temporal Sampling Policy Optimization (TSPO), advancing MLLMs' long-form video-language understanding via reinforcement learning. Specifically, we first propose a trainable event-aware temporal agent, which captures event-query correlation for performing probabilistic keyframe selection. Then, we propose the TSPO reinforcement learning paradigm, which models keyframe selection and language generation as a joint decision-making process, enabling end-to-end group relative optimization for the temporal sampling policy. Furthermore, we propose a dual-style long video training data construction pipeline, balancing comprehensive temporal understanding and key segment localization. Finally, we incorporate rule-based answering accuracy and temporal locating reward mechanisms to optimize the temporal sampling policy. Comprehensive experiments show that our TSPO achieves state-of-the-art performance across multiple long video understanding benchmarks, and shows transferable ability across different cutting-edge Video-MLLMs. Our code is available at https://github.com/Hui-design/TSPO
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。