01研究问题
扩散模型生成高清图片,为什么很费算力?
因为它要反复处理图像;图片越大,每一步要计算的内容就越多。先看生成图片这件事:如果给模型一句“红色杯子放在桌上”,它需要决定杯子在哪里、桌面是什么样,还要填出颜色和纹理。
扩散模型从一张随机噪声图出发,让神经网络一轮轮预测并去掉噪声,逐渐形成新图片。这里的“去噪”是生成步骤,不是在修复一张已有的照片。
难点在于,这样的调整需要重复许多轮。如果每一轮都直接处理高清图片的所有像素,计算量就会很大。可以把它想成修改一幅大画:每调整一点,都要重新检查整张画布;画布越大、调整次数越多,花的时间越多。这只是解释计算成本的类比,模型并不是按人画画的方式操作。
还有一层浪费:像素里既有杯子的形状、位置等重要信息,也有很多人几乎看不出的细微变化。作者因此提出问题:能否先把图片变成一份更小、仍保留重要视觉信息的表示,再在那里反复去噪?
论文把这份学出来的压缩表示所在的空间叫作“潜空间”。它必须能还原成图片;压缩得太少,计算负担仍重,压得太多,又会丢失细节。下一节就看模型如何完成压缩、去噪和还原。
02两阶段方法
先压缩,再去噪:两个模型怎样接力?
第一阶段学习图片与潜表示之间的转换;第二阶段固定转换器,在潜表示上学习生成。
第一阶段:学会压缩和还原
编码器 E 把真实图片变成较小的潜表示 z;解码器 D 再把 z 还原成图片。训练这对模型时,作者既要求重建图像在观感上接近原图,也用局部判别训练帮助保留真实纹理。之后训练扩散模型时,这个压缩空间保持固定。
论文把空间下采样倍数记为 f。例如 f=4 时,256 × 256 图片对应的潜表示宽高是 64 × 64;这是尺寸换算示例。潜表示还有自己的通道数,因此不能简单说“总数据量必定缩为 1/16”。
第二阶段:只在潜空间里学去噪
训练时,模型先取得图片的潜表示,给它加入不同程度的噪声,再让 UNet 预测噪声。UNet 是一种能同时结合局部与整体信息的图像网络。生成时则从随机的潜空间噪声开始,反复去噪得到新的 z,最后由解码器 D 一次性还原成图片。
想象整理一幅画:先学会用一张较小的“结构草图”保留主体和布局,再在草图层面反复调整,最后展开成完整画面。这只是帮助理解的示意;论文实际使用的是连续数值表示,不是人工绘制的草图。
03如何控制生成
文字提示怎样影响正在生成的图像?
模型先把文字变成一组可供读取的向量,再让去噪网络在生成过程中参考它们。论文把这套连接方式称为交叉注意力:图像潜表示中的位置可以从文字表示里取与自己相关的信息。
比如输入“红色杯子在桌上”,文字编码器会把这些词变成表示;UNet 在多步去噪时,便可参考与“杯子”“红色”“桌上”相关的信息。这是机制讲解示意,不是论文展示的某张生成图片,也不保证每个词都被准确落实。
论文还研究了其他条件输入。文字和布局等可以借助交叉注意力;超分辨率、图像修补等与图像位置一一对应的条件,则可以把条件图与潜表示沿通道拼接。条件并不是直接替代去噪过程,而是在每一步影响它。
04实验结果
实验显示:省了多少计算,画质如何?
在论文的图像修补设置里,潜空间模型训练和采样都更快,样本分布指标也更好。图像修补是把图片被遮住的区域补出来。下表摘自原论文表 6:对比直接在像素上去噪的 LDM-1,和空间下采样 4 倍、使用 VQ 型压缩器的 LDM-4。VQ 会把连续表示归入一组学得的编码。该组比较固定了模型参数量。
| 指标 | 像素空间 LDM-1 | 潜空间 LDM-4 |
|---|---|---|
| 训练吞吐量(张/秒) | 0.11 | 0.33 |
| 512 × 512 采样吞吐量(张/秒) | 0.07 | 0.34 |
| 第 6 轮验证集 FID | 24.74 | 14.99 |
在这组条件下,LDM-4 每秒训练约 3 倍图片,512 × 512 采样每秒约 4.9 倍图片。FID 衡量的是生成样本与真实样本的整体特征分布差距,越低通常越好;它不能告诉我们单张图片是否忠实于某个指令。表 6 的不同实验设置会影响速度,因此这些倍数不是所有硬件、任务或模型都适用的加速承诺。
“压得越小越好”也没有得到支持。论文图 6、图 7 对比不同下采样倍数:f=1、2 的训练推进慢,f=32 则因过度压缩而限制画质;f=4、8 在所测图像生成任务上取得较好的效率与质量平衡。
文字生成任务提供了另一类证据:论文在 LAION 数据上训练文字条件模型,并在 MS-COCO 上评估。表 2 中,加入无分类器引导(一种在采样时加强文字提示影响的策略)的 LDM-KL-8-G 的 FID 为 12.63,未加入该引导的 LDM-KL-8 为 23.31。这个比较也包含了引导策略的影响,不能把改进全部归因于“在潜空间扩散”。
05适用边界
潜空间更省算力,哪些问题仍然存在?
压缩并没有让生成变成一步完成,也可能丢掉任务需要的精确像素信息。论文指出,多步采样仍比 GAN 慢;当任务要求像素级精确时,编码器能否完整重建图片会成为上限。作者尤其提醒超分辨率任务可能受此影响。
- 质量取决于第一阶段。如果压缩器没有保存某些细节,后面的去噪网络无法保证把原有细节准确恢复。f=4、8 的优势来自论文所测数据和配置。
- 条件生成不等于完全可控。交叉注意力提供了一种接入文字或布局的方法;论文展示了多类生成任务,但没有证明任意复杂指令都能被准确遵循。
- 降低成本也带来使用风险。作者讨论了篡改图像、虚假信息、训练数据隐私及偏见等问题;降低生成门槛不会自动解决这些风险。
这篇论文的核心不是“把图片缩小后再放大”,而是把压缩学习和生成学习分开:让昂贵的多步去噪发生在更合适的表示空间,最后再回到像素空间。
产品启发:在真实图像产品中,除了看成图观感,还应按任务检查细节保真、提示词遵循、单张生成耗时和潜在误用。这是根据论文机制提出的评估建议,不是论文已经完成的产品验证。
依据:原论文 §5:局限与社会影响、§6:结论。