清华、字节跳动王炸开源!5秒出图+PS级编辑,算力暴降6倍

2026-01-08 阅读量:214

关注CAIE,国内头部AI人才认证、培训体系,助你在职场升职加薪。
CAIE注册人工智能工程师
CAIE注册人工智能工程师
公众号作者
CAIE,全称 Certifed Artifcial Intelligence Engineer(人工智能工程师),简称 CAIE(赛一) ,是人工智能领域的技能等级认证。旨在评估和培养具备人工智能理论基础与实战能力的职业人士。

多模态AI大模型一直是个研究难点,语言模型写文案、做推理是一把好手,但让它生成或编辑图片就抓瞎;扩散模型能出高质量图片,可你让它理解复杂指令、做逻辑推理又完全不行。

而字节跳动联合清华、莫纳什大学的研究人员重磅开源了NextFlow,一个模型就能搞定理解、生成、编辑所有事,1024×1024的高清图5秒就能生成。

比之前那些自回归模型快了不止一点半点,同时算力还暴降了6倍左右。

其实行业里一直在尝试多模态,主要分两派。一派是搞混合架构,把自回归和扩散模型凑一起,看着能兼顾两边的优点,但实际用起来会有额外的计算开销。就像用两台不同系统的电脑协作,文件来回传特别麻烦。

另一派是纯自回归模型,想靠一个架构包打天下,可问题更明显。之前的纯自回归模型生成图片,都是像扫二维码一样逐点生成,分辨率越高,生成时间就成倍增加。

一张1024×1024的图要等10多分钟,完全没法实际用。而且它们对图片的理解也很浅,只能捕捉到像素层面的信息,没法get到高层的语义,比如你让它“把夏天的枫叶改成绿色”,它可能都理解不了夏天枫叶该是什么状态。

NextFlow模型最大变革就是,看透了文本和图片的本质区别。文本是一句一句按顺序来的,但图片是有层级的,先有大致轮廓,再有细节。抓住这个关键点,它就彻底解决了之前的那些痛点。

NextFlow的核心思路就是“统一”,用一个模型搞定所有事。双码本Tokenizer让模型既懂语义又懂像素。就像我们看一张图,既要知道里面画的是什么,也要看清细节好不好。

NextFlow的双码本设计就是干这个的,一个码本专门负责捕捉图片的核心概念,比如这是一只猫、在草地上;另一个码本负责还原细节,比如猫的毛发纹理、草地的颜色深浅。

这俩码本还能互相配合,不会顾此失彼。而且它还支持不同分辨率和宽高比的图片,不用强行把图片拉伸或裁剪成固定尺寸,生成的图片自然更协调。对比之前那些单一码本的模型,它生成的图片不仅像,还更懂你想表达的意思。

Next-scale预测,生成图片像搭积木一样高效。之前的模型生成图片是逐点扫描,就像用针一点点绣一幅画,慢得要死。

NextFlow改成了分层生成,先搭个大致框架,再慢慢补细节,就像盖房子先打地基、搭骨架,再装修内饰。

比如生成一张1024×1024的图,它先生成2×2的粗略轮廓,再扩展到4×4、8×8,一步步把细节补全。这样一来,计算量大大减少,速度自然就上去了,5秒搞定高清图就是这么来的。

为了让模型能分清不同层级的重要性,它还会给早期的粗框架更高的权重,避免出现“细节很精致但整体布局乱七八糟”的情况。

而且它还能自我纠错,前面步骤生成得不够好,后面能及时调整,减少图片里的奇怪伪影。

NextFlow的解码器特别简洁,就一个输出头,既能处理文本又能处理图片,不像有些模型搞一堆专用头,复杂还容易出问题。它还加了一些小优化,让模型能更好地适应不同尺寸的图片,生成的一致性更高。

不过离散编码难免会损失一些高频细节,比如小尺寸的文字、人脸细节。所以它还加了个可选的扩散解码器,就像给图片做后期精修,需要超写实效果的时候打开就行。

但这个功能也有取舍,精修的时候可能会轻微改变一些细节,所以日常使用默认关闭,按需启用就好。

模型再聪明,没有好数据也白搭。NextFlow的训练数据规模是真的大,足足6万亿个文本和图像的离散token,涵盖了各种场景。

它的训练过程也很有讲究,不是一下子就上高分辨率,而是从256分辨率开始,慢慢升到512、1024,让模型循序渐进地学习。

还分了好几个阶段,先对齐文本和图像的语义,再大规模预训练,之后优化美学质量,最后用强化学习微调,确保模型既好用又好看。训练的时候还用了1024块GPU,通过一些优化策略让GPU利用率拉满。

生成1024×1024的高清图只要5秒,比传统自回归模型快了上百倍,比diffusion模型也省了不少算力,推理时的计算量直接减少6倍。

在多个权威测试里,NextFlow的表现都特别突出。比如在GenEval测试中,它对指令的理解和还原度,能跟GPTImage1、Seedream3.0这些顶尖模型看齐。

在DPG测试里,它在实体还原、细节呈现上还排到了第一。生成的图片不仅符合要求,美学质量也很高,想象力、构图都在线。

精准图像编辑是NextFlow的王牌功能,不管是局部修改还是全局调整,都能精准拿捏。

比如你想把图片里的黑色衣服改成红色、给小狗缩小头上的蝴蝶结、把背景的蓝天改成多云,都能完美实现,不会改了一个地方,其他地方变得乱七八糟。

在ImgEdit、OmniContext这些专业编辑测试里,它的分数都是第一,甚至在主题一致性上超过了GPT-4o。

还支持56种细粒度的编辑任务,从文本修改到风格转换,几乎覆盖了所有日常需求,以后做设计、改图片,可能真的不用打开PS了。

开源地址:https://github.com/ByteVisionLab/NextFlow

想转型AI,不被时代淘汰

CAIE注册人工智能工程师认证

岗位能力 × AI工具 ×转型方向 × 场景落地 = 新AI职业价值

扫码免费领取《AI工程师入门学习指南》

点击下方
阅读原文
即刻跳转至CAIE官网,了解更多AI相关信息

确定要退出登录吗?
确定 取消