OpenAI推出简化一致性模型sCM,将大幅提升采样速度
报道 10月24日消息,据OpenAI公告,其推出了全新的sCM(简化一致性模型),该模型以简化的连续时间一致性模型为基础,提供更高的训练稳定性和可扩展性。据称,sCM能够在仅需两步采样的情况下生成与领先扩散模型相当的样本质量,大幅提升采样速度。
传统扩散模型生成样本需要几十到上百步,而sCM通过将噪声直接转换为无噪样本,仅需两步即可完成,速度提升约50倍。在ImageNet 512×512数据集上,OpenAI训练了1.5亿参数的sCM,生成单个样本仅需0.11秒。sCM的训练基于预训练的扩散模型,并通过一致性蒸馏技术加以优化,使其在保持样本质量的同时大幅降低计算成本。
OpenAI表示,sCM的进步为高效实时生成AI内容开辟了新可能,将持续推进该领域的发展。