arXiv:2411.14762cs.CVcs.AI2024-11CVPR被引 6

用坐标重建方法高效压缩长视频,减少近80%的令牌数量。

Efficient Long Video Tokenization via Coordinate-based Patch Reconstruction

  • 基于坐标采样重建视频补丁,实现长视频低开销训练。
  • 128帧128×128视频仅需1280个令牌,比基线少70%以上。
  • 适合需要高效长视频建模的生成模型研究者使用。

长视频的有效标记化仍是训练可处理长视频视觉模型的挑战。现有标记器在长视频上训练成本高昂,因其需一次性重建所有帧。本文提出CoordTok,一种受3D生成模型启发的视频标记器,将视频编码为因子化的三平面表示,并针对随机采样的 (x,y,t) 坐标重建对应补丁。该方法使大模型能直接在长视频上训练,无需大量资源。实验表明,CoordTok可显著减少令牌数量:例如,128帧、128×128分辨率的视频仅需1280个令牌,而基线需6144或8192个令牌即可达到相似重建质量。此外,该高效标记化支持内存高效的扩散转换器训练,可一次性生成128帧视频。

原文摘要 · Abstract (English)

Efficient tokenization of videos remains a challenge in training vision models that can process long videos. One promising direction is to develop a tokenizer that can encode long video clips, as it would enable the tokenizer to leverage the temporal coherence of videos better for tokenization. However, training existing tokenizers on long videos often incurs a huge training cost as they are trained to reconstruct all the frames at once. In this paper, we introduce CoordTok, a video tokenizer that learns a mapping from coordinate-based representations to the corresponding patches of input videos, inspired by recent advances in 3D generative models. In particular, CoordTok encodes a video into factorized triplane representations and reconstructs patches that correspond to randomly sampled $(x,y,t)$ coordinates. This allows for training large tokenizer models directly on long videos without requiring excessive training resources. Our experiments show that CoordTok can drastically reduce the number of tokens for encoding long video clips. For instance, CoordTok can encode a 128-frame video with 128$\times$128 resolution into 1280 tokens, while baselines need 6144 or 8192 tokens to achieve similar reconstruction quality. We further show that this efficient video tokenization enables memory-efficient training of a diffusion transformer that can generate 128 frames at once.

视频标记化坐标重建扩散模型长视频

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