首个4D视频生成扩散模型!数分钟实现4D内容生成,超81K优质数据集已开源

351081b07b68e284e885df18b0e0ae0e.png

来源:PaperWeekly
本文约1400字,建议阅读5分钟本文介绍各大学团队最新提出的 4D 生成扩散模型。

本文介绍由多伦多大学,北京交通大学,德克萨斯大学奥斯汀分校和剑桥大学团队最新提出的 4D 生成扩散模型,该方法可以在几分钟之内可以完成时空一致的 4D 内容生成。Diffusion4D 整理筛选了约 81K 个 4D assets,利用 8 卡 GPU 共 16 线程,花费超过 30 天渲染得到了约四百万张图片,包括静态 3D 物体环拍、动态 3D 物体环拍以及动态 3D 物体前景视频。

该方法是首个利用大规模数据集,训练视频生成模型生成 4D 内容的框架,目前项目已经开源所有渲染的 4D 数据集以及渲染脚本。

e880b964148daead751cad7a5af5cd15.png

项目地址:

https://vita-group.github.io/Diffusion4D/

论文地址:

https://arxiv.org/abs/2405.16645

1、研究背景

过去的方法采用了 2D、3D 预训练模型在 4D(动态 3D)内容生成上取得了一定的突破,但他们主要依赖于分数蒸馏采样(SDS)或者生成的伪标签进行优化,同时利用多个预训练模型获得监督不可避免的导致时空上的不一致性以及优化速度慢的问题。

4D 内容生成的一致性包含了时间上和空间上的一致性,它们分别在视频生成模型和多视图生成模型中被探索过。基于这个洞见,Diffusion4D 将时空的一致性嵌入在一个模型中,并且一次性获得多时间戳的跨视角监督。

具体来说,使用仔细收集筛选的高质量 4D 数据集,Diffusion4D 训练了一个可以生成动态 3D 物体环拍视图的扩散模型,而后利用已有的 4DGS 算法得到显性的 4D 表征,该方法实现了基于文本、单张图像、3D 到 4D 内容的生成。

d9e26ddd74f4d8b509a73d46b29620d8.png

2、4D数据集

为了训练 4D 视频扩散模型,Diffusion4D 收集筛选了高质量的 4D 数据集。已开源的 Objaverse-1.0 包含了 42K 运动的 3D 物体,在 Objaverse-xl 中包含 323K 动态 3D 物体。然而这些数据包含着大量低质量的样本,研究者们设计了运动程度检测、边界溢出检查等筛选方法,选取了共 81K 的高质量 4D 资产。

对于每一个 4D 资产,渲染得到了 24 个静态视角的图(上图第一行),24 个动态视角的环拍图(上图第二行),以及 24 个正面动态图(上图第三行)。总计得到了超过四百万张图片,总渲染消耗约 300 GPU 天。其他数据集细节可以参考项目主页,目前所有渲染完的数据集和原始渲染脚本已开源,更多数据集使用方法值得探索!

3、方法

有了 4D 数据集之后,Diffusion4D 训练具有 4D 感知的视频扩散模型(4D-aware video diffusion model)。过去的视频生成模型通常不具备 3D 几何先验信息,但近期工作如 SV3D,VideoMV 等探索了利用视频生成模型得到静态 3D 物体的多视图,因此 Diffusion4D 选用了 VideoMV 作为基础模型进行微调训练,使得模型能够输出动态环拍视频。

此外设计了如运动强度(motion magnitude)控制模块、3D-aware classifier-free guidance 等模块增强运动程度和几何质量。得益于视频模态具备更强的连贯性优势,输出的结果具有很强的时空一致性。

341cc77b038e4e81c0ce2e4c9a286338.png

输出得到动态视角环拍视频后,Diffusion4D 借助已有的 4D 重建算法将视频建模得到 4D 表达。具体来说采用了 4DGS 的表征形式,以及使用粗粒度、细粒度的两阶段优化策略得到最终的 4D 内容。从生产环拍视频到重建 4D 内容的两个步骤仅需花费数分钟时间,显著快于过去需要数小时的借助 SDS 的优化式方法。

4、结果

根据提示信息的模态,Diffusion4D 可以实现从文本、图像、3D 到 4D 内容的生成,在定量指标和 user study 上显著优于过往方法。在生成质量上,Diffusion4D 有着更好的细节,更为合理的几何信息以及更丰富的动作。更多可视化结果可以参考项目主页。

1430caa833c89f72e9cc1c7e0e639419.png

5、总结

Diffusion4D 是首个利用视频生成模型来实现 4D 内容生成的框架,通过使用超 81K 的数据集、以及精心设计的模型架构实现了快速且高质量的 4D 内容。未来,如何最大程度发挥 4D 数据集价值,如何生成多物体、复杂场景的 4D 内容仍有很大的探索空间!

编辑:于腾凯

校对:林亦霖

关于我们

数据派THU作为数据科学类公众号,背靠清华大学大数据研究中心,分享前沿数据科学与大数据技术创新研究动态、持续传播数据科学知识,努力建设数据人才聚集平台、打造中国大数据最强集团军。

148e21bd16067487db2f730b14cc77fc.png

新浪微博:@数据派THU

微信视频号:数据派THU

今日头条:数据派THU

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值