arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2211.14575cs.CV

基于稀疏条件流匹配的高效视频预测

Efficient Video Prediction via Sparsely Conditioned Flow Matching

University of Bern(伯尔尼大学)

Aram Davtyan, Sepehr Sameni, Paolo Favaro

AI总结:

本研究提出名为RIVER的新型视频预测生成模型,基于潜在流匹配,通过稀疏随机过去帧条件、潜在空间运算及带噪帧近似流ODE初始条件,在性能持平或更优的同时大幅降低计算资源需求。

摘要:

我们提出了一种基于潜在流匹配的新型视频预测生成模型,该方法是扩散模型的高效替代方案。与现有研究不同,我们在图像生成过程的每个积分步中仅以少量随机选取的过去帧为条件,从而避免了训练和推理期间对过去信息建模的高昂成本。此外,为实现高分辨率视频生成并加快训练速度,我们在预训练VQGAN(矢量量化生成对抗网络)的潜在空间中进行运算。最后,我们提出用前一帧带噪帧来近似流常微分方程(ODE)的初始条件,这一设计可减少积分步数量,进而提升推理时的采样速度。我们将该模型命名为Random frame conditioned flow Integration for VidEo pRediction,简称RIVER。实验表明,在常用视频预测基准测试中,RIVER的性能优于或持平于现有方法,同时所需计算资源减少了一个数量级。

英文摘要:

We introduce a novel generative model for video prediction based on latent flow matching, an efficient alternative to diffusion-based models. In contrast to prior work, we keep the high costs of modeling the past during training and inference at bay by conditioning only on a small random set of past frames at each integration step of the image generation process. Moreover, to enable the generation of high-resolution videos and to speed up the training, we work in the latent space of a pretrained VQGAN. Finally, we propose to approximate the initial condition of the flow ODE with the previous noisy frame. This allows to reduce the number of integration steps and hence, speed up the sampling at inference time. We call our model Random frame conditioned flow Integration for VidEo pRediction, or, in short, RIVER. We show that RIVER achieves superior or on par performance compared to prior work on common video prediction benchmarks, while requiring an order of magnitude fewer computational resources.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑