设置
  • 日夜间
    随系统
    浅色
    深色
  • 主题色

北大字节开辟图像生成新范式!超越Sora核心组件DiT,不再预测下一个token

2024-04-15 17:24:49 来源: 量子位

网友:通向Sora的更便宜的潜在途径ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

北大和字节联手搞了个大的:ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

提出图像生成新范式,从预测下一个token变成预测下一级分辨率,效果超越Sora核心组件Diffusion Transformer(DiT)。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

并且代码开源,短短几天已经揽下1.3k标星,登上GitHub趋势榜。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

具体是个什么效果?ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

实验数据上,这个名为VAR(Visual Autoregressive Modeling)的新方法不仅图像生成质量超过DiT等传统SOTA,推理速度也提高了20+倍ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

这也是自回归模型首次在图像生成领域击败DiT。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

直观感受上,话不多说,直接看图:ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

值得一提的是,研究人员还在VAR上,观察到了大语言模型同款的Scaling Laws和零样本任务泛化。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

论文代码上线,已经引发不少专业讨论。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

有网友表示有被惊到,顿时觉得其他扩散架构的论文有点索然无味。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

还有人认为,这是一种通向Sora的更便宜的潜在途径,计算成本可降低一个乃至多个数量级。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

预测下一级分辨率

简单来说,VAR的核心创新,就是用预测下一级分辨率,替代了预测下一个token的传统自回归方法。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

VAR的训练分为两个阶段。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

第一阶段,VAR引入了多尺度离散表示,使用VQ-VAE将连续图像编码为一系列离散的token map,每个token map有不同的分辨率。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

第二阶段,主要是对VAR Transformer的训练,通过预测更高分辨率的图像,来进一步优化模型。具体过程是这样的:ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

从最低分辨率(比如1×1)的token map开始,预测下一级分辨率(比如4×4)的完整token map,并以此类推,直到生成最高分辨率的token map(比如256×256)。在预测每个尺度的token map时,基于Transformer,模型会考虑之前所有步骤生成的映射信息。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

在第二阶段中,之前训练好的VQ-VAE模型发挥了重要作用:为VAR提供了“参考答案”。这能帮助VAR更准确地学习和预测图像。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

另外,在每个尺度内,VAR是并行地预测所有位置的token,而不是线性逐个预测,这大大提高了生成效率。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

研究人员指出,采用这样的方法,VAR更符合人类视觉感知从整体到局部的特点,并能保留图像的空间局部性。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

符合Scaling Laws

从实验结果来看,在图像生成质量、推理速度、数据效率和可扩展性等方面,VAR都超过了DiT。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

在ImageNet 256×256上,VAR将FID从18.65降到了1.8,IS从80.4提高到356.4,显著改善了自回归模型基线。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

注:FID越低,说明生成图像的质量和多样性越接近真实图像。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

推理速度方面,相较于传统自回归模型,VAR实现了约20倍的效率提升。而DiT消耗的时间是VAR的45倍。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

数据效率方面,VAR只需要350个训练周期(epoch),远少于DiT-XL/2的1400个。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

可扩展性方面,研究人员观察到VAR有类似于大语言模型的Scaling Laws:随着模型尺寸和计算资源的增加,模型性能持续提升。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

另外,在图像修补、扩展和编辑等下游任务的零样本评估中,VAR表现出了出色的泛化能力。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

目前,在GitHub仓库中,推理示例、demo、模型权重和训练代码均已上线。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

不过,在更多讨论之中,也有网友提出了一些问题:ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

北大字节联合出品

VAR的作者们,来自字节跳动AI Lab和北大王立威团队。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

一作田柯宇,本科毕业自北航,目前是北大CS研究生,师从北京大学信息科学技术学院教授王立威。2021年开始在字节AI Lab实习。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

论文通讯作者,是字节跳动AI Lab研究员袁泽寰和王立威。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

袁泽寰2017年博士毕业于南京大学,目前专注于计算机视觉和机器学习研究。王立威从事机器学习研究20余年,是首届“优青”获得者。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

该项目的项目主管,是字节跳动广告生成AI研究主管Yi jiang。他硕士毕业于浙江大学,目前的研究重点是视觉基础模型、深度生成模型和大语言模型。ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

参考链接:
[1]论文:https://arxiv.org/abs/2404.02905ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
[2]项目主页:https://github.com/FoundationVision/VARykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

— 完 —ykv速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

本文链接:北大字节开辟图像生成新范式!超越Sora核心组件DiT,不再预测下一个tokenhttp://www.sushuapos.com/show-2-4870-0.html

声明:本网站为非营利性网站,本网页内容由互联网博主自发贡献,不代表本站观点,本站不承担任何法律责任。天上不会到馅饼,请大家谨防诈骗!若有侵权等问题请及时与本网联系,我们将在第一时间删除处理。

上一篇: 游戏周报:网易暴雪破镜重圆;本年度第二批进口版号14枚下发

下一篇: 人在B站,要被AI公司们挤爆了

热门资讯

  • 迄今最大三维宇宙地图发布

    国际天文学家团队绘制了迄今最大的三维宇宙地图,记录了大约130万个活跃类星体在空间和时间上的位置。它将成为探测类星体、暗物质晕和超大质量黑洞

  • 我国首个适应高寒气候肉羊品种通过鉴定

    记者19日从兰州大学获悉,天华肉羊通过国家畜禽遗传资源委员会审定鉴定,成为我国首个适应高寒气候的肉羊品种。该品种由兰州大学草地农业科技学院李发

  • 人类祖先200万年前开始捕鱼

    美国《发现》杂志网站2月7日刊登题为《200万年前,我们的人类祖先开始从水里捞鱼》的文章,作者是科迪·科蒂尔,内容编译如下:捕鱼可能是一种占许

  • 我国科学家揭示东北黑土成因

    记者3月20日从中国科学院地质与地球物理研究所获悉,该所科研人员在东北黑土区开展了大范围的野外调查和样品采集工作,通过分析采集的黑土样品发现,黑

  • 肺癌蛋白质致命弱点发现

    肺癌是全球死亡率最高的恶性肿瘤。其中非小细胞肺癌(NSCLC)占比达到85%。表皮生长因子受体(EGFR)是NSCLC最常见的驱动突变基因。现在,一个国际科学家团

  • 借AI“慧眼”鉴别可疑论文图片

    今年1月,英国分子生物学家肖尔托·戴维发表文章,指控美国哈佛大学医学院附属丹娜-法伯癌症研究所科学家通过修改图片伪造数据。随后该研究所正

  • 专家:警惕常见睡眠认知误区

    3月21日是世界睡眠日,中国主题为“健康睡眠 人人共享”。近日发布的《2023年中国居民睡眠白皮书》显示,我国居民平均睡眠时长6.75小时,平均在零点后入

  • 多模态医疗大模型CARES Copilot 1.0发布 已面向香港神经外科医生开放使用

    记者3月21日获悉,由中国科学院自动化研究所和中国科学院香港创新研究院联合研发的医疗领域AI多模态大模型——CARES Copilot 1.0日前在香

  • 黑场显微镜让单个颗粒实现无标记光学显微成像

    3月21日记者从中国科学技术大学获悉,该校物理学院张斗国教授课题组,提出并实现了一种基于矢量光场调控原理的动量空间偏振滤波器件。科研人员将该滤

  • 25日再次“露脸”,水星为啥这么难看见?

    3月25日6时左右将迎来水星东大距。这是水星今年第二次大距、首次东大距,也是公众尝试观测水星的好机会。届时,水星位于太阳东边,与太阳张角约为18.7度

  • 新技术破解结核病“早发现早治疗”难题

    3月24日是第29个世界防治结核病日,我国的宣传主题是“你我共同努力,终结结核流行”。在北京大学社会化媒体研究中心21日举办的“技术升级,加速我国终

  • 中广核:用创新技术提升核电站“智”力

    在近日开幕的中国国际核工业展览会上,中国核学会理事会党委书记、理事长王寿君表示,中国内地现有在运核电机组55台、居全球第三;在建核电机组26台,保持

推荐资讯

  • 日榜
  • 周榜
  • 月榜