打开网易新闻 查看精彩图片

这项由北京航空航天大学复杂关键软件环境全国重点实验室联合南洋理工大学、中国科学院计算技术研究所、中国科学技术大学及苏黎世联邦理工学院共同完成的研究,以预印本论文形式于2026年7月发布在arXiv平台,论文编号为arXiv:2607.01642。研究的核心成果是一套名为MrFlow的无需训练的图像生成加速方案,专门针对当前主流的文字转图片扩散模型。

当你在手机上用AI软件输入一段描述"一匹白马在春天的草地上奔跑",等待系统生成一张精美图片时,你可能完全意识不到,背后的计算机正在经历一段漫长而繁重的"绘画过程"。以目前顶尖的图像生成模型Qwen-Image-20B为例,在一块顶级的英伟达A100显卡上,生成一张1024×1024像素的图片,需要整整47秒。这不是程序员偷懒,而是这类模型的工作方式本身决定的——它需要反复、细致地从随机噪点中一步步"雕刻"出最终图像,就像一位雕塑家从一块粗糙的大理石开始,一刀一刀地打磨出精细的作品。

北航等机构的研究团队正是为了解决这个"等待太久"的问题而出发的。他们开发的MrFlow方案,让同样的图像生成过程快了10倍以上,同时图像质量几乎没有损失。更重要的是,这套方案完全不需要对原有模型进行任何额外的训练,直接"拿来即用"。当与其他加速技术叠加使用时,甚至能实现高达25倍的加速效果。

一、AI画图究竟慢在哪里,以及为什么过去的加速方法都不够好

要理解MrFlow的聪明之处,需要先搞清楚扩散模型的工作原理,以及前人尝试加速时遇到的困境。

扩散模型(Diffusion Model)的工作方式,可以用一个"去噪"的比喻来理解。给你一张全是随机雪花点的图片,然后一步一步地猜测并消除噪点,最终还原出一张清晰的图像。这个过程通常需要50到100步,每一步都要运行一次庞大的神经网络模型。而现代图像生成模型,比如FLUX和Qwen-Image,使用的是一种叫做"流匹配"(Flow Matching)的技术框架,可以理解为更平滑、更高效版本的去噪过程,但核心思路类似。

当图像分辨率越高,每一步的计算量就越大。具体来说,分辨率翻倍,图像中的像素块数量会增加四倍,而模型中最耗时的"注意力机制"(可以理解为模型在每一步需要分析图像中所有像素块之间的相互关系)的计算量会增加十六倍。换句话说,生成1024×1024的图比生成512×512的图,理论上贵16倍。

研究者们为了提速,大致走了几条路。第一条路是"知识蒸馏",也叫时步蒸馏,让一个小模型去模仿大模型的行为,把原本50步才能完成的生成压缩到4步甚至1步——但这需要大量的额外训练,而且训练好的模型往往只适合生成风格比较单一的图像,多样性会下降。第二条路是"特征缓存",既然相邻两步之间模型的中间计算结果差不多,那就直接复用上一步的结果,跳过部分重复计算——这种方法不需要训练,但加速倍数顶多在4倍左右,再高就会让图像质量明显下滑。第三条路是"多分辨率生成",先在低分辨率下快速生成图像的大体结构,再放大到高分辨率补充细节——这个方向理论上很美好,实际上5倍以上的加速已经很有吸引力,但过去的实现都存在一个棘手的问题:在放大的过程中会产生明显的模糊或伪影(就是图像中出现奇怪的格子状纹路或者模糊块)。

MrFlow的出发点正是这最后一条路,研究团队决定彻底解决多分辨率加速方案中的伪影和模糊问题。他们找到的关键突破口,是把"放大"这个操作从一个不擅长处理像素细节的"潜在空间",挪到更适合处理图像的"像素空间"来做。

二、MrFlow的四步"流水线":像摄影师冲洗照片一样的分阶段处理

MrFlow的整个工作流程,可以类比成一位摄影师在暗房里冲洗胶片的过程。先快速显影出照片的大致轮廓,再用专业设备放大并锐化细节,然后对放大过程中出现的细微失真做一次轻微的"二次曝光"重新捕获,最后做最后一道精修。整个过程分四个明确的阶段。

第一阶段是在低分辨率下生成图像的主体结构。具体来说,模型先在目标尺寸的一半大小(比如512×512,而非最终的1024×1024)的"潜在空间"里,从一团随机噪声开始,跑12步扩散去噪流程,得到一张低分辨率的干净潜在表示。然后用VAE解码器(可以理解为把模型内部的"草图"翻译成人眼能看见的像素图片的工具)把它转换成一张512×512的像素图。

这个阶段的速度优势来自两个相互独立的来源。其一是单步计算本身快了大约4倍,因为图像尺寸缩小一半后,图像块数量减少了四分之三,注意力计算量大幅降低。其二更微妙但同样重要:低分辨率下的模型只需要更少的步骤就能把图像的整体结构和语义内容表达清楚,这是因为在低分辨率下,模型中的图像部分对文字描述的"注意力"更高——实验数据显示,在模型的浅层和中间层,低分辨率生成时分配给文字条件的注意力权重明显高于高分辨率生成,意味着每一步推理都能更高效地把文字描述转化为图像结构。研究者还测量了图像在潜在空间中的"ODE路径长度"(可以理解为从纯噪声到干净图像,模型的"思路"需要走多长的路),发现低频部分(即图像的大轮廓和整体布局)只占总路径的约58%,高频细节(纹理、笔触、文字细节)占另外42%。低分辨率只需要走那58%的路就够了,高分辨率才需要走完全程。实验也证实,用10步低分辨率加1步高分辨率的组合,比直接用11步高分辨率生成质量更好,延迟还少了一半以上。

第二阶段是在像素空间做超分辨率放大。拿到512×512的低分辨率图之后,MrFlow调用一个预训练好的Real-ESRGAN模型(一个基于生成对抗网络的图像超分辨率模型),把图片从512×512放大到1024×1024。

为什么要在像素空间做这一步,而不是直接在模型内部的潜在空间里插值放大?研究团队给出了清晰的技术理由。潜在空间是模型训练时专门构建的一种抽象表示,它的每一个"格子"都依赖特定的空间统计关系。如果直接在潜在空间里把特征图插值放大,这种统计关系就会被破坏,解码出来的图像必然出现规律性的格子状伪影——对比实验中,LSSGen等在潜在空间做放大的方法确实出现了这个问题。像素空间则是超分辨率研究积累了数十年的主战场,各种先验知识(边缘、纹理、自然图像的频谱特性)都以像素为基础建立,预训练好的超分辨率模型能直接复用。

另一个设计选择是为什么要选GAN(生成对抗网络)类的超分辨率模型,而不是更常见的L2损失训练的回归模型(比如SwinIR)或者基于扩散的超分辨率(比如OSEDiff)。回归类模型虽然结构更清晰、保留整体布局,但它的"职业病"是高频细节模糊——它追求的是像素级别的最小均方误差,导致输出的纹理很平滑、很"安全",但也很糊。GAN类模型则相反,它放弃了像素级别的精确,但换来了自然图像般的锐利细节,即便细节有时不完全符合文字描述(比如文字笔画有些偏差),但这些误差主要集中在高频方向,而高频误差正是后续第四阶段能够纠正的。如果用模糊的超分辨率结果,后续纠错的负担会更重,需要更大的噪声注入,反而会破坏好不容易保留下来的低分辨率结构。

第三阶段是注入低强度噪声,实现高频重采样。把放大后的1024×1024像素图,再通过VAE编码器(解码器的逆操作,把像素图翻译回模型的内部表示)转换成高分辨率的潜在表示。此时得到的潜在表示已经有了正确的整体结构,但可能含有一些超分辨率引入的高频伪影(比如文字笔画的小偏差)。

为了让下一阶段的高分辨率精修能纠正这些高频误差,研究团队在这个潜在表示上混入了一点点随机噪声——噪声强度通常只有0.1到0.15(在0到1的范围内,1代表完全的纯噪声,0代表没有噪声)。这个操作的逻辑可以用一个类比来理解:超分辨率放大后的图像就像一个大体正确但局部细节有些不准的草稿,直接拿去精修的话,精修程序不知道哪些细节是可信的、哪些是错的。加入一点点噪声,相当于告诉精修程序"高频细节部分你不用太相信这个草稿,按照你自己的判断重新来",同时因为噪声很小,低频的整体结构(颜色分布、构图、语义布局)完全不受影响。

研究团队从信号处理的角度给出了严格的理论推导。他们证明,对于图像高频频段,干净图像潜在表示的"高频功率"(即高频信息的方差)大约只有整体的8%(对应标准差约0.08),由此推导出让模型的高分辨率先验"接管"高频决策权所需的最低噪声强度约为0.075,实验中选用的0.1到0.15略高于此下界,足以让高频部分得到充分的重采样,同时又不至于干扰低频结构。如果使用的是模糊类超分辨率(其误差扩散到中低频),对应的最低噪声强度就会跳升到0.16甚至0.24,需要更大的噪声注入,这样反而会冲刷掉超分辨率好不容易提供的有用细节,得不偿失。这也从理论上解释了为什么GAN类超分辨率与低强度噪声这两个设计选择必须"配套"使用。

第四阶段是在高分辨率下进行单步细节精修。注入噪声之后,把带噪的1024×1024潜在表示送回原来的扩散模型,从当前的噪声强度(比如0.1)对应的时间点出发,跑1步(默认配置)到少数几步的扩散去噪,让模型在高分辨率下完成最后的细节填充。最后再用VAE解码器把结果转换成最终的像素图。

为什么只需要1步就够?研究者从速度场的特性上给出了解释。扩散过程从纯噪声到干净图像,就像一条从山顶到山脚的河流。在靠近山顶(高噪声)的地方,河流弯弯曲曲,走一步可能走错很远;而在靠近山脚(低噪声,接近干净图像)的地方,河流趋于平直,走一步的误差极小。当噪声强度只有0.1时,出发点已经非常靠近终点,速度场几乎是一条直线,因此单步Euler积分(可以理解为沿着当前斜率走一步)的误差极小。实验数据证实,在强度0.1的噪声起点,单步精修的结果与8步精修的结果的CLIP图像相似度高达0.9974,两者几乎无差别。为验证步骤分配的最优性,研究者还比较了三种步骤分布策略:均匀分步、前密后疏(更多步骤集中在高噪声区域)和后密前疏(更多步骤集中在低噪声区域)。结果证明,在靠近图像端集中步骤的"后密前疏"策略,总轨迹路径长度比前密后疏短了37%,效率最优。

三、实验结果:数字和图像都说明了什么

研究团队在两款主流模型——FLUX.1-dev和Qwen-Image-20B——上进行了系统测试,生成分辨率统一设为1024×1024,使用三类指标衡量图像质量:Geneval(评估图像与文字描述在构图和语义上的匹配程度)、DPG-Bench(针对长描述文字的对齐质量)和OneIG-Bench(综合评估中英文双语的字符渲染、风格一致性等多维度属性)。

对比的方法涵盖了多个类别。特征缓存类方法有TeaCache和DB-Taylor;token融合压缩类有ToMA;同属多分辨率方向的训练免除方法有RALU和SPEED;训练依赖的最先进时步蒸馏方法则是SenseFlow和Pi-Flow。

在无需训练的方法中,MrFlow表现出了明显领先的综合优势。以Qwen-Image为例,用20步低分辨率加1步高分辨率的配置,MrFlow实现了约7倍加速,同时Geneval分数从基准的0.88下降到0.87,DPG-Bench从88.67下降到88.00,OneIG-En保持0.54(与基准持平),表现几乎与不加速时没有区别。TeaCache在同等加速比下,Geneval已经跌到0.81,损失了接近8个百分点;DB-Taylor保留了大部分Geneval,但DPG有所下滑;RALU的表现在视觉检查中更差。

当把加速配置调整到更激进的12步低分辨率加1步高分辨率时,MrFlow在Qwen-Image上达到10.3倍加速,Geneval仍保持0.86(仅比基准低0.02),DPG-Bench保持87.10,OneIG达到0.52。相比之下,同等速度区间内,TeaCache的Geneval已经崩溃到0.26,DB-Taylor更跌到0.09,生成质量基本不可用;RALU的Geneval在0.84还算可接受,但实际图像质量经视觉检验明显更差;SPEED在FLUX上有不错的指标,但在Qwen-Image上同样出现了明显质量下滑。

在与训练依赖方法的对比中,MrFlow同样毫不逊色。在FLUX上,免训练的12+1步MrFlow(8.25倍加速),Geneval达到0.63,与需要大量训练的SenseFlow 4步版本(11倍加速,Geneval 0.63)几乎持平。当MrFlow与Pi-Flow蒸馏模型结合(无需额外训练,直接加载Pi-Flow权重在低分辨率和高分辨率阶段都使用蒸馏模型)时,FLUX上达到11.3倍加速,Geneval提升到0.69;Qwen-Image上的MrFlow+Pi-Flow组合达到25.1倍加速,Geneval 0.85,OneIG-Bench损失不超过1%。

研究者还在更新的FLUX.2 Klein(4B和9B两个规模版本)和Z-Image系列上验证了MrFlow的迁移能力。结论一致:在FLUX.2 Klein Base上,20+1步配置能达到5倍以上加速且几乎无质量损失;在Z-Image上,12+1步配置达到10.8倍加速;Z-Image-Turbo与MrFlow结合后达到21倍加速。这表明MrFlow的方案与具体的模型架构无关,具备良好的通用性。

从实际运行时间来看,Qwen-Image的12+1步MrFlow总耗时4.77秒,而原始50步生成需要49.32秒,端到端加速比10.35倍。在这4.77秒中,低分辨率采样占3.2秒,高分辨率精修占1.03秒,中间的VAE解码、Real-ESRGAN超分辨率、VAE编码等固定开销合计只有约0.3秒,占比很小,不会成为瓶颈。

四、消融实验:每一个设计选择都有充分理由

研究团队不仅展示了最终结果,还通过细致的消融实验证明了每个设计选择的必要性。

在超分辨率模型的选择上,研究者在Qwen-Image上对比了双线性插值、SwinIR(L2回归模型)、OSEDiff(扩散类超分辨率)和Real-ESRGAN(GAN类)。从自动化指标上看,四种方法的DPG分数相差不大(86.97到87.25之间),仿佛差别可以忽略。但当把生成的图像放大来看,差别就非常明显了:插值和SwinIR的放大图仍然模糊,局部文字笔画不清;OSEDiff在放大时引入了字符变形;只有Real-ESRGAN的放大结果锐利自然,经过高分辨率精修后质量最佳。这说明自动化指标对高频细节的评估并不充分,视觉质量的判断仍然不可替代。值得注意的是,SwinIR版本的整体推理速度也更慢(4.67倍对应Real-ESRGAN的10.3倍),因为SwinIR本身推理更慢,在加速总账上得不偿失。

在噪声注入方向的选择上,研究者设计了一个精巧的实验:在高分辨率潜在空间注入噪声时,把通常的高斯噪声替换成三种结构化噪声——高频带通、低频带通和通道相关噪声,然后测量精修后输出到干净高分辨率图像"云"(在DINOv2特征空间中的分布)的k近邻距离,以此衡量精修步骤的纠正能力。结果非常清晰:注入高斯噪声和高频噪声后,精修结果的距离与不加噪声直接精修几乎相同(约0.154到0.155),说明这两类误差在精修步骤的可纠正范围内;而注入低频噪声后,距离飙升到0.498(在强度0.3时),是默认值的3倍以上。这证明精修阶段的速度场只有修正高频误差的能力,对低频偏差几乎无能为力。这个发现与GAN超分辨率的设计选择形成了完美的呼应:GAN的残差误差主要集中在高频,恰好在精修的能力范围内。

在步数配置的灵活性上,研究者在Qwen-Image上穷举了低分辨率步数(10、12、16、20步)与高分辨率步数(1、2、3步)的所有组合,绘制出了Geneval与加速倍数的折中曲线。结果显示,MrFlow是唯一一种在两个模型上都持续超越"直接调整原始模型步数"这一天真基准的方法,并且可以灵活调配配置,向质量或速度任意倾斜。从曲线形状来看,增加低分辨率步数对指标提升明显,而增加高分辨率步数几乎没有额外收益,这与理论分析完全一致。

五、一个隐藏的实验,证明了流水线的"分工哲学"是对的

研究团队还专门设计了一个实验,验证MrFlow流水线中"低分辨率决定结构、高分辨率精修细节"这个核心假设是否真的成立。

实验做法是:固定低分辨率阶段的随机种子,只改变高分辨率精修阶段的随机种子(vary HR组),观察最终输出在不同频段的差异;再固定高分辨率精修的种子,只改变低分辨率阶段的种子(vary LR组),同样观察差异。用图像的低频均方差和高频均方差来量化不同频段的变化幅度。

结论令人震惊:改变低分辨率种子,低频均方差约为1.087×10??;改变高分辨率种子,同样是低频均方差,只有1.298×10??,两者相差约8000倍。换句话说,图像的整体结构(低频部分)几乎100%由低分辨率阶段决定,高分辨率阶段的随机性对整体结构几乎没有任何影响。而高分辨率精修引入的变化,其高频/低频比值(19.8)远高于低分辨率阶段引入变化的对应比值(7.4),说明精修阶段的影响确实集中在高频细节,而非整体结构。这个实验以最直接的数据证明了MrFlow的流水线设计哲学是正确的。

说到底,MrFlow的核心贡献是找到了一条多分辨率图像生成加速的正确路径,并且把路上每一个岔口的选择都论证得清清楚楚。与其在模型内部的抽象空间里做各种手脚,不如让每个阶段各司其职——低分辨率阶段快速建立语义骨架,像素空间的超分辨率负责保留结构并补充高频细节,低强度噪声注入完成频率选择性的"授权",高分辨率单步精修最终抹去高频瑕疵。

这套方案不依赖任何专用硬件,不需要对原始模型做任何修改,不需要收集运行时的动态统计信息,也不需要任何额外训练,真正做到了"开箱即用"。更实际的意义在于,它与目前所有主流的时步蒸馏方法正交兼容,可以直接叠加使用,把已经很快的蒸馏模型再加速2到3倍。这意味着随着更多轻量化模型的发布,MrFlow的效果还可以继续叠加增强。

对于普通用户来说,这项研究的直接影响是:以后在手机或网页上用AI生成图片,等待的时间可能从几十秒压缩到几秒钟,而图片质量几乎看不出差别。当然,实际产品中的落地还涉及工程适配、内存优化等额外工作,但研究层面的技术路障已经被清除。有兴趣深入了解技术细节的读者,可以通过arXiv平台查阅编号为2607.01642的完整论文。

Q&A

Q1:MrFlow和其他扩散模型加速方法相比,最主要的优势是什么?

A:MrFlow最大的优势是在10倍以上的加速比下,图像质量几乎不损失,而其他训练免除的方法(如TeaCache、RALU)在达到同样加速倍数时,生成质量通常会明显崩溃。MrFlow不需要任何额外训练,并且可以直接与时步蒸馏模型叠加使用,实现最高25倍的加速,这种"免训练且可叠加"的特性是其他方案难以同时满足的。

Q2:MrFlow为什么不在模型内部的潜在空间做图像放大,而要先解码成像素图?

A:在潜在空间直接插值放大会破坏VAE解码器依赖的空间统计关系,导致解码后图像出现规律性的格子状伪影。像素空间是图像超分辨率研究积累多年的领域,预训练的超分辨率模型(如Real-ESRGAN)在像素域工作效果最好,而且VAE编码器在把放大后的像素图重新编码时,会自然衰减超出训练分布的高频成分,起到轻量级的滤波作用,为后续噪声注入提供更干净的起点。

Q3:MrFlow的高分辨率阶段为什么只需要一步就够了?

A:因为高分辨率精修的起点(经过超分辨率放大再加入0.1强度噪声的潜在表示)已经非常接近最终干净图像,处于扩散轨迹最靠近终点的区间。在这个区间,速度场非常平直,单步Euler积分的误差极小。实验测量显示,在0.1噪声强度下,单步精修结果与8步精修结果的图像相似度高达0.9974,几乎完全一致,多走几步没有实质收益。