扬帆创投微信小程序
更聚焦的出海投融资平台
精准高效领先的融资对接服务
微信扫一扫进入小程序

OpenAI发布了一种新的扩散模型方法sCM,该方法能够在保持生成图片质量的同时显著提高推理效率,仅需两步即可生成高质量图片或3D模型,实现了50倍的时钟加速。

摘要

sCM训练了一个具有15亿参数的模型,该模型能在单个A100 GPU上在0.11秒内生成内容,无需任何推理优化。与目前最快的扩散模型相比,sCM在保持质量的同时将推理效率提升了一倍。

sCM采用连续时间框架,避免了离散时间模型中的离散化误差。这种框架使得模型参数化、扩散过程和训练目标能够适应连续时间设置,从而更精确地捕捉数据生成过程中的动态变化。

sCM引入了多项改进,包括改进的时间条件、自适应组归一化、新的激活函数和自适应权重,以提高模型的训练稳定性和生成质量。这些改进有助于模型在训练过程中保持内部特征的稳定性,并动态调整损失函数中的权重。

研究人员在CIFAR-10、ImageNet 64×64和ImageNet 512×512等多个数据集上对sCM进行了评估,证实了其作为目前最高效的扩散生成方法的地位。例如,在ImageNet 512×512数据集上,sCM的模型达到了1.88的FID分数,同时使用的算力更少、更高效。

今天凌晨,OpenAI发布了全新扩散模型方法sCM,仅需2步就能生成高质量图片、3D模型等实现50倍时钟加速,尤其是在高分辨率任务上相当出色。

例如,通过sCM训练了一个15亿参数的模型,在单个A100 GPU上无需任何推理优化0.11秒内就能生成内容。

目前,扩散模型生成图片最快的是Stability AI开源的SD快速版本,4步就能生成高质量图片。而Scm在保证质量的前提下又将推理效率提升了1倍,同时简化了连续时间一致性模型的理论公式,允许模型在更大数据集上进行稳定的训练和扩展。

图片

sCM的核心原理是基于一致性模型思路,通过直接将噪声转换为无噪声样本来生成数据。在传统扩散模型中,数据生成过程被视作一条从噪声到数据的渐进路径,每一步都通过去噪来逐渐恢复数据的清晰度。

一致性模型则恰恰相反,可找到一条更直接的路径,在单步或少数几步内直接从噪声状态跳跃到数据状态。

sCM采用了连续时间框架,使得模型在理论上可以在连续的时间轴上进行操作,从而避免了离散时间模型中的离散化误差。在连续时间于一致性模型中,模型的参数化、扩散过程和训练目标都被重新定义,以适应连续时间的设置。

图片

例如,模型的参数化不再依赖于离散的时间步,而是直接依赖于时间本身。这种连续时间的参数化方式使得模型能够更精确地捕捉数据生成过程中的动态变化。

在网络架构方面,sCM引入了改进的时间条件、自适应组归一化、新的激活函数和自适应权重,以提高模型的训练稳定性和生成质量。改进的时间条件使得模型能够更准确地感知时间t的变化,从而在生成过程中做出更合理的决策。

自适应组归一化则有助于模型在训练过程中保持内部特征的稳定性,减少训练过程中的噪声干扰。新的激活函数被引入以增强模型的非线性表达能力,使得模型能够更好地学习复杂的数据分布。

而自适应权重的引入允许模型根据训练数据的分布动态调整损失函数中的权重,从而减少了不同时间步长之间的损失方差。

图片

为了评估sCM的性能,研究人员在CIFAR-10、ImageNet 64×64和ImageNet 512×512多个数据集上进行了综合评测。结果显示,sCM成为目前最高效的扩散生成方法。

例如,在ImageNet 512×512数据集上,sCM的模型达到了1.88FID,同时使用的算力更少、更高效。

有网友表示,如果把这个方法用在视频领域,那实时视频可能很快会到来。Sora的推理负担也会降低很多。

图片

很高兴又看到OpenAI分享技术论文了。

图片

如果把这个技术用在Sora,它应该快公测了吧?

图片

这种简化的模型非常适合需要快速结果而不影响质量的应用!

图片

这个模型相当有前途啊。

图片

2步就能生成内容,这会再一次改变游戏规则啊。

图片

把这个技术用在Sora,应该就快来了吧?

图片

目前,OpenAI已经分享了该论文方法,是由两位华人提出来的。

论文地址:https://arxiv.org/abs/2410.11081

原文链接:点击前往 >

文章作者: AIGC开放社区

版权申明:文章来源于 AIGC开放社区。该文观点仅代表作者本人,扬帆出海平台仅提供信息存储空间服务,不代表扬帆出海官方立场。因本文所引起的纠纷和损失扬帆出海均不承担侵权行为的连带责任,如若转载请联系原文作者。 更多资讯关注扬帆出海官网:https://www.yfchuhai.com/

{{likeNum}}

好文章,需要你的鼓励

请前往扬帆出海小程序完成个人认证
认证通过后即可申请入驻
扫码添加小助手
微信扫描二维码,添加小助手
微信号:yfch24
扫码添加小助手
微信扫描二维码,添加小助手
微信号:yfch24
APP
小程序
微信公众号
微信小程序
扬帆出海APP
扬帆出海APP
微信
公众号
微信公众号
关注扬帆出海
专注服务互联网出海!
出海人
社群
微信公众号
扫码进群
与10万+出海人同行!