改进大模型注意力机制,让图文模型更好理解视频时序信息
TC-LLaVA: Rethinking the Transfer from Image to Video Understanding with Temporal Considerations
- 用时序感知的旋转位置编码增强视频时序建模能力
- 通过帧级掩码扩大视觉标记跨帧交互,提升因果推理性能
- 仅需监督微调即在多个视频基准上达新最优,适合视频理解研究者
多模态大语言模型(MLLM)在图像-语言任务中表现显著提升。近期研究致力于将预训练于图像的MLLM迁移到视频任务,但多数工作集中在视觉编码器和投影模块的优化,而核心的大语言模型(LLM)部分仍缺乏深入探索。本文提出两种策略,通过改进LLM中的层间注意力计算来增强视频理解能力:第一种是引入时序感知双旋转位置编码(Temporal-Aware Dual RoPE),在保留视觉与文本标记相对位置关系的同时,加入时间位置信息以强化时序建模;第二种是采用帧级块因果注意力掩码(Frame-wise Block Causal Attention Mask),在保持因果推理机制的前提下,扩展视频帧内及跨帧的视觉标记交互。基于这些方法,我们将LLaVA适配为时序考虑的视频理解模型——TC-LLaVA。实验表明,仅通过视频数据的监督微调(SFT),TC-LLaVA在多个视频理解基准上达到新最优性能。
原文摘要 · Abstract (English)
Multimodal Large Language Models (MLLMs) have significantly improved performance across various image-language applications. Recently, there has been a growing interest in adapting image pre-trained MLLMs for video-related tasks. However, most efforts concentrate on enhancing the vision encoder and projector components, while the core part, Large Language Models (LLMs), remains comparatively under-explored. In this paper, we propose two strategies to enhance the model's capability in video understanding tasks by improving inter-layer attention computation in LLMs. Specifically, the first approach focuses on the enhancement of Rotary Position Embedding (RoPE) with Temporal-Aware Dual RoPE, which introduces temporal position information to strengthen the MLLM's temporal modeling capabilities while preserving the relative position relationships of both visual and text tokens. The second approach involves enhancing the Attention Mask with the Frame-wise Block Causal Attention Mask, a simple yet effective method that broadens visual token interactions within and across video frames while maintaining the causal inference mechanism. Based on these proposed methods, we adapt LLaVA for video understanding tasks, naming it Temporal-Considered LLaVA (TC-LLaVA). Our TC-LLaVA achieves new state-of-the-art performance across various video understanding benchmarks with only supervised fine-tuning (SFT) on video-related datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。