arXiv:2411.13683cs.CV2024-11NeurIPS被引 6

提出自适应掩码策略,让视频自编码器处理128帧长视频并显著提效。

Extending Video Masked Autoencoders to 128 frames

  • 基于重要性优先的自适应掩码,动态选择重建关键帧。
  • 在128帧视频上性能超越32帧模型,Diving48提升3.9分,EPIC-Kitchens提升2.5分。
  • 仅用视频数据预训练,无需图文对或复杂架构,适合通用视频理解任务。

视频理解近年因自监督预训练模型取得显著进展,其中掩码自编码器(MAE)成为主流设计。然而,多数已有工作受限于硬件内存与计算开销,仅能处理短序列视频(如16/32帧)。本文提出一种自适应解码器掩码策略,通过量化令牌并优先重建重要帧,实现128帧长视频的有效训练。该方法依托基于MAGVIT的分词器,联合学习令牌及其重要性。大量消融实验证明,所提长视频掩码自编码器(LVMAE)在128帧下性能优于32帧模型。在Diving48上超过现有最优结果3.9分,在EPIC-Kitchens-100动作分类上提升2.5分,且仅需简单架构与纯视频预训练,无需数百万标注的图文对或专用编码器。

原文摘要 · Abstract (English)

Video understanding has witnessed significant progress with recent video foundation models demonstrating strong performance owing to self-supervised pre-training objectives; Masked Autoencoders (MAE) being the design of choice. Nevertheless, the majority of prior works that leverage MAE pre-training have focused on relatively short video representations (16 / 32 frames in length) largely due to hardware memory and compute limitations that scale poorly with video length due to the dense memory-intensive self-attention decoding. One natural strategy to address these challenges is to subsample tokens to reconstruct during decoding (or decoder masking). In this work, we propose an effective strategy for prioritizing tokens which allows training on longer video sequences (128 frames) and gets better performance than, more typical, random and uniform masking strategies. The core of our approach is an adaptive decoder masking strategy that prioritizes the most important tokens and uses quantized tokens as reconstruction objectives. Our adaptive strategy leverages a powerful MAGVIT-based tokenizer that jointly learns the tokens and their priority. We validate our design choices through exhaustive ablations and observe improved performance of the resulting long-video (128 frames) encoders over short-video (32 frames) counterparts. With our long-video masked autoencoder (LVMAE) strategy, we surpass state-of-the-art on Diving48 by 3.9 points and EPIC-Kitchens-100 verb classification by 2.5 points while relying on a simple core architecture and video-only pre-training (unlike some of the prior works that require millions of labeled video-text pairs or specialized encoders).

视频自编码长视频自适应掩码MAE

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