arXiv:2410.20280cs.CVcs.AI2024-10被引 46

用掩码自回归规划视频生成,高效实现多场景视频合成

MarDini: Masked Autoregressive Diffusion for Video Generation at Scale

  • 将掩码自回归与扩散模型结合,分步完成时序规划与空间生成
  • 单模型支持插值、图生视频、扩展等多任务,推理步数少且质量高
  • 计算资源集中于低分辨率规划,适合大规模视频生成应用

我们提出MarDini,一种新型视频扩散模型家族,将掩码自回归(MAR)融入统一的扩散模型(DM)框架。其中,MAR负责时序规划,而DM专注于空间生成,采用非对称网络设计:i)包含大部分参数的基于MAR的规划模型,使用低分辨率输入为每个被掩码帧生成规划信号;ii)轻量级生成模型利用这些信号通过扩散去噪生成高分辨率帧。MarDini的MAR机制支持在任意帧位置掩码任意数量帧,单一模型可处理视频插值(如掩码中间帧)、图像到视频生成(如从第二帧起掩码)和视频扩展(如掩码一半帧)。高效设计将主要计算资源分配给低分辨率规划模型,使昂贵但关键的时空注意力可在大规模下实现。MarDini在视频插值任务上达到新SOTA;同时,在少量推理步数内,生成视频质量媲美更昂贵的先进图像到视频模型。

原文摘要 · Abstract (English)

We introduce MarDini, a new family of video diffusion models that integrate the advantages of masked auto-regression (MAR) into a unified diffusion model (DM) framework. Here, MAR handles temporal planning, while DM focuses on spatial generation in an asymmetric network design: i) a MAR-based planning model containing most of the parameters generates planning signals for each masked frame using low-resolution input; ii) a lightweight generation model uses these signals to produce high-resolution frames via diffusion de-noising. MarDini's MAR enables video generation conditioned on any number of masked frames at any frame positions: a single model can handle video interpolation (e.g., masking middle frames), image-to-video generation (e.g., masking from the second frame onward), and video expansion (e.g., masking half the frames). The efficient design allocates most of the computational resources to the low-resolution planning model, making computationally expensive but important spatio-temporal attention feasible at scale. MarDini sets a new state-of-the-art for video interpolation; meanwhile, within few inference steps, it efficiently generates videos on par with those of much more expensive advanced image-to-video models.

视频生成扩散模型掩码建模

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