arXiv:2504.17447cs.CVcs.AI2025-04被引 26

通过选帧生成提升长视频与长文档理解,无需微调现有模型。

FRAG: Frame Selection Augmented Generation for Long Video and Long Document Understanding

  • 先选关键帧再生成,避免长上下文计算负担。
  • 在视频任务上提升5.8%,文档任务上提升超20%。
  • 适配主流大模型,无需额外训练,适合资源受限场景。

大型多模态模型(LMMs)在处理长输入方面取得显著进展,包括多页文档和长视频。然而,由于训练与推理的高计算成本,这些长上下文模型的规模与性能仍受限。本文探索了一条新路径:不依赖长上下文LMMs,而是通过帧选择增强生成(FRAG)。该方法首先独立评分输入中的每一帧,再通过简单的Top-K选择保留高分帧,最后仅基于选中帧生成输出。核心机制无需长上下文处理,可直接应用于现有LMMs,无需微调。实验使用LLaVA-OneVision和InternVL2模型,在长视频和多页文档任务中均实现显著提升:在MLVU上使InternVL2-76B提升5.8%,在Video-MME上提升3.7%;在MP-DocVQA上超越近期专为长文档设计的LMMs,提升超20%。代码已开源。

原文摘要 · Abstract (English)

There has been impressive progress in Large Multimodal Models (LMMs). Recent works extend these models to long inputs, including multi-page documents and long videos. However, the model size and performance of these long context models are still limited due to the computational cost in both training and inference. In this work, we explore an orthogonal direction and process long inputs without long context LMMs. We propose Frame Selection Augmented Generation (FRAG), where the model first selects relevant frames within the input, and then only generates the final outputs based on the selected frames. The core of the selection process is done by scoring each frame independently, which does not require long context processing. The frames with the highest scores are then selected by a simple Top-K selection. We show that this frustratingly simple framework is applicable to both long videos and multi-page documents using existing LMMs without any fine-tuning. We consider two models, LLaVA-OneVision and InternVL2, in our experiments and show that FRAG consistently improves the performance and achieves state-of-the-art performances for both long video and long document understanding. For videos, FRAG substantially improves InternVL2-76B by 5.8% on MLVU and 3.7% on Video-MME. For documents, FRAG achieves over 20% improvements on MP-DocVQA compared with recent LMMs specialized in long document understanding. Code is available at: https://github.com/NVlabs/FRAG

视频理解文档理解轻量推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。