arXiv:2502.07737cs.CVcs.AI2025-02被引 30

用块预测提升视频生成速度,11倍加速且质量更好

Next Block Prediction: Video Generation via Semi-Autoregressive Modeling

  • 将视频分块并行预测,打破逐个令牌生成限制
  • 在UCF101上FVD达103.3,比传统方法快11倍
  • 适合追求高效视频生成的研究者和开发者

当前自回归视频生成普遍采用逐标记预测(NTP),存在单向依赖与推理缓慢的问题。本文提出半自回归框架Next-Block Prediction(NBP),将视频均匀划分为等大小块(如行或帧),将生成单位从单个标记改为块,使当前块内每个标记可同时预测下一区块对应标记。该框架在块内引入双向注意力,增强空间依赖建模能力;通过并行预测多个标记,显著减少生成步数,实现更高效推理。模型在UCF101上取得103.3的FVD分数,在K600上为25.5,较基线平均提升4.4分;生成速度达每秒8.89帧(128x128分辨率),提速11倍。模型规模从700M扩展至3B参数后,UCF101的FVD降至55.3,K600降至19.5,验证了方法的可扩展性。

原文摘要 · Abstract (English)

Next-Token Prediction (NTP) is a de facto approach for autoregressive (AR) video generation, but it suffers from suboptimal unidirectional dependencies and slow inference speed. In this work, we propose a semi-autoregressive (semi-AR) framework, called Next-Block Prediction (NBP), for video generation. By uniformly decomposing video content into equal-sized blocks (e.g., rows or frames), we shift the generation unit from individual tokens to blocks, allowing each token in the current block to simultaneously predict the corresponding token in the next block. Unlike traditional AR modeling, our framework employs bidirectional attention within each block, enabling tokens to capture more robust spatial dependencies. By predicting multiple tokens in parallel, NBP models significantly reduce the number of generation steps, leading to faster and more efficient inference. Our model achieves FVD scores of 103.3 on UCF101 and 25.5 on K600, outperforming the vanilla NTP model by an average of 4.4. Furthermore, thanks to the reduced number of inference steps, the NBP model generates 8.89 frames (128x128 resolution) per second, achieving an 11x speedup. We also explored model scales ranging from 700M to 3B parameters, observing significant improvements in generation quality, with FVD scores dropping from 103.3 to 55.3 on UCF101 and from 25.5 to 19.5 on K600, demonstrating the scalability of our approach.

视频生成半自回归加速推理扩散模型

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