首页IT科技常用扩散模型有哪些(Stable diffusion扩散模型相关)

常用扩散模型有哪些(Stable diffusion扩散模型相关)

时间2025-06-17 12:06:31分类IT科技浏览5163
导读:时隔两年半(2年4个月),我又回来研究生成技术了。以前学习研究GAN没结果,不管是技术上,还是应用产品上,结果就放弃了,现在基于diffusion的技术又把生成技术带上了一个新的高度。现在自己又来研究学习这方面的东西了。现在看来,以前还是自己自我定位不清晰,想搞研究,搞出研究成果来,自己能力不够,也没人带,搞不了;然后格局不够,对...

时隔两年半(2年4个月)             ,我又回来研究生成技术了             。以前学习研究GAN没结果                       ,不管是技术上      ,还是应用产品上         ,结果就放弃了                       ,现在基于diffusion的技术又把生成技术带上了一个新的高度                       。现在自己又来研究学习这方面的东西了      。现在看来          ,以前还是自己自我定位不清晰      ,想搞研究                      ,搞出研究成果来              ,自己能力不够   ,也没人带                     ,搞不了;然后格局不够                  ,对事物曲折变化认识不够,坚持力也不够         。现在就还是面向应用进行学习研究了                       。当时的GAN生成效果还差一截                 ,但是现在的完全能够达到真实效果了                      ,并且技术使用难度也没那么大   ,应该能做成受欢迎产品级的效果了          。

一             、基本概念

扩散模型 diffusion

当前主要有四大生成模型:生成对抗模型                       、变微分自动编码器      、流模型以及扩散模型      。扩散模型(diffusion models)是当前深度生成模型中新SOTA                      。

不止于图像生成:

从连续到离散             ,计算机视觉                       ,语音生成      ,NLP         、波形信号处理                       、多模态建模          、分子图建模      、时间序列建模                      、对抗性净化等都有出色表现              。

做研究的话:

其实大部分的研究者都不需要理解扩散模型的数学本质         ,更需要的是对扩散模型的原理的经验化理解                       ,从而应用到research里面去   。

基本数学原理:

它的原理相当复杂          ,很多数学推导                     。然后一些具体的东西也不用知道      ,以俺们的水平                      ,知道了也不能根据原理推导出什么性质              ,从而帮助做什么提升性工作                  。要知道其性质   ,只能直接从经验角度走                     ,以及看网上的资料了。

下面的可以不看吧

视频教程好一点

https://www.bilibili.com/video/BV1r14y1Y7xa/?spm_id_from=333.788.recommend_more_video.9&vd_source=832fdd9b4041ab8d66be70be56266213

diffusion model 最近在图像生成领域大红大紫                  ,如何看待它的风头开始超过 GAN ? - 路橙LuChengTHU的回答 - 知乎

https://www.zhihu.com/question/536012286/answer/2533146567

从最开始说起,深度学习就是拟合概率函数                 ,通过大量数据学习到大量参数                      ,来拟合这个函数                 。(现实中很多的智能任务就是在计算一个概率函数?)

一般的深度学习任务   ,输出维度很低                      。

比如图片分类             ,拟合输入数据到输出分类这么一个函数   。

注意到这里图片数据是多维的                       ,这是一个多维到一维的函数      ,单值函数             。

而生成模型则是反过来         ,输入低维数据                       ,得到高维数据          ,这个可以叫做结构化学习?

这里生成对象可以是不同的东西      ,图片矩阵                      ,文字序列都可以              ,需要具体控制                       。

输入的低维数据就代表要生成对象的特征   ,比如狗                     ,黄色的狗到图片                  ,狗为主题到一篇文章      。然后生成过程就交给深度学习模型去拟合了         。

https://zhuanlan.zhihu.com/p/563543020

https://zhuanlan.zhihu.com/p/34998569 这篇文章最好

我们看回生成模型到底是个啥                       。我们拿到一批样本(比如图片,称为 X )                 ,想要学到它的分布p(X)                      ,这样就能同时学到没被采样到的数据了   ,用这个分布 p(X) 就能随意采样             ,然后获得生成结果          。(这里怎么采样?高维空间上X值在某些点概率很大                       ,某些点概率很低      ,但是这样采样没有意义      。)                      。于是这里引入一个隐变量Z         ,就是前面说得特征                       ,高维数据在低维空间的流形              。不妨假设Z服从正态分布          ,然后根据Z来生成X   。Z和X具有一个条件概率关系p(X|Z)       ,然后利用条件概率和全概率公式计算得到p(X),从而学习到X的分布                      ,同时得到特征隐变量z和x的关系                     。

这个条件概率p(X|Z) 就描述了一个由 Z 来生成 X的模型              ,约等于生成器G                  。

于是   ,从VAE和GAN                     ,他们都是从隐变量 Z(对象特征)生成目标数据X                  ,但是实现上有所不同。

更准确地讲,它们是假设了Z服从某些常见的分布(比如正态分布或均匀分布)                 ,训练一个模型G                      ,这个模型将原来的概率分布映射到训练集的概率分布   ,即X‘=g(Z)                 。也就是说             ,它们的目的都是进行分布之间的变换                      。

最后当然是希望促使模型得到的分布 X’ 和目标分布 X尽量接近   。这个就是生成模型的难题                       ,如何判断生成分布与真实分布的相似度      ,因为我们只知道两者的采样结果         ,不知道它们的分布表达式             。

那现在假设Z服从标准的正态分布                       ,那么我就可以从中采样得到若干个 Z1,Z2,…,Zn          ,然后对它做变换得到 X̂1=g(Z1),X̂2=g(Z2),…,X̂n=g(Zn)      ,我们怎么判断这个通过 G 构造出来的数据集                      ,它的分布跟我们目标的数据集分布是不是一样的呢?

有读者说不是有 KL 散度吗?当然不行              ,因为 KL 散度是根据两个概率分布的表达式来算它们的相似度的   ,然而目前我们并不知道它们的概率分布的表达式                       。

GAN的做法就是直接把这个判断方法也用神经网络学习过来就行                     ,边生成边训练      。但是仍然有问题                  ,后面讲到         。

而VAE 则使用了一个精致迂回的技巧                       。

VAE的核心:

其实,在整个 VAE 模型中                 ,我们并没有去使用 p(Z)(先验分布)是正态分布的假设                      ,我们用的是假设 P(X|Z) (后验分布)是正态分布          。

具体来说   ,给定一个真实样本 Xk             ,我们假设存在一个专属于 Xk 的分布 p(Z|Xk)(学名叫后验分布)                       ,并进一步假设这个分布是(独立的              、多元的)正态分布      。(我理解这里Z是大Z      ,代表集合         ,就是每个z值到x_k的概率形成的分布吧                      。)

为什么要强调“专属               ”呢?因为我们后面要训练一个生成器 X=g(Z)                       ,希望能够把从分布 p(Z|Xk) 采样出来的一个 Zk 还原为 Xk              。

如果假设 p(Z) 是正态分布          ,然后从 p(Z) 中采样一个 Z      ,那么我们怎么知道这个 Z 对应于哪个真实的 X 呢?现在 p(Z|Xk) 专属于 Xk                      ,我们有理由说从这个分布采样出来的 Z 应该要还原到Xk 中去   。

那我怎么找出专属于 Xk 的正态分布 p(Z|Xk) 的均值和方差呢?找出均值方差就知道分布了              ,好像并没有什么直接的思路                     。

那好吧   ,我就用神经网络来拟合出来                  。这就是神经网络时代的哲学:难算的我们都用神经网络来拟合。

于是我们构建两个神经网络

μ

k

=

f

1

(

X

k

)

                     ,

l

o

g

σ

2

=

f

2

(

X

k

)

来算它们了                 。我们选择拟合

l

o

g

σ

2

而不是直接拟合

σ

2

                  ,是因为

σ

2

总是非负的,需要加激活函数处理                 ,而拟合

l

o

g

σ

2

μk=f1(Xk)                      ,logσ^2=f2(Xk)来算它们了                      。我们选择拟合 logσ^2 而不是直接拟合 σ^2   ,是因为 σ^2 总是非负的             ,需要加激活函数处理                       ,而拟合 logσ^2

μk=f1(Xk)      ,logσ2=f2(Xk)来算它们了   。我们选择拟合logσ2而不是直接拟合σ2         ,是因为σ2总是非负的                       ,需要加激活函数处理          ,而拟合logσ2 不需要加激活函数      ,因为它可正可负             。

这里没太明白                      ,反正就是构造神经网络              ,把均值方差算出来了                       。

到这里   ,我能知道专属于 Xk 的均值和方差了                     ,也就知道它的正态分布长什么样了                  ,然后从这个专属分布中采样一个 Zk 出来,然后经过一个生成器得到 X̂k=g(Zk)      。

现在我们可以放心地最小化 D(X̂k,Xk)^2                 ,因为 Zk 是从专属 Xk 的分布中采样出来的                      ,这个生成器应该要把开始的 Xk 还原回来         。于是可以画出 VAE 的示意图:

VAE 的本质是什么

它本质上就是在我们常规的自编码器的基础上   ,对 encoder 的结果(在VAE中对应着计算均值的网络)加上了“高斯噪声                   ”             ,使得结果 decoder 能够对噪声有鲁棒性;而那个额外的 KL loss(目的是让均值为 0                       ,方差为 1)      ,事实上就是相当于对 encoder 的一个正则项         ,希望 encoder 出来的东西均有零均值                       。

模型结构

https://zhuanlan.zhihu.com/p/525106459

diffusion model和其他模型最大的区别是它的latent code(z)和原图是同尺寸大小的(正常情况)          。一句话概括diffusion model                       ,即存在一系列高斯噪声( T 轮)          ,将输入图片 x0 变为纯高斯噪声 xt       。而我们的模型则负责将 xt 复原回图片 x0                       。这个和GAN像      ,只是噪声维度不同              。

Diffusion前向过程

所谓前向过程                      ,即往图片上加噪声的过程   。虽然这个步骤无法做到图片生成              ,但是这是理解diffusion model以及构建训练样本GT至关重要的一步                     。

前向过程介绍结束前   ,需要讲述一下diffusion在实现和推导过程中要用到的两个重要特性                  。

特性1:重参数(reparameterization trick)

重参数技巧在很多工作(gumbel softmax, VAE)中有所引用。如果我们要从某个分布中随机采样(高斯分布)一个样本                     ,这个过程是无法反传梯度的                 。而这个通过高斯噪声采样得到 的过程在diffusion中到处都是                  ,因此我们需要通过重参数技巧来使得他可微                      。

Diffusion逆向(推断)过程

如果说前向过程(forward)是加噪的过程,那么逆向过程(reverse)就是diffusion的去噪推断过程   。

步骤总结:每个时间步通过 xt 和t来预测高斯噪声                 ,然后计算得到均值和方差                      ,再计算一个中间量xxx   ,最后重参数得到xt-1

训练过程简记

训练过程中应该是             ,用UNet                       ,输入前向之后的图像      ,然后让UNet预测加入的噪声是什么         ,比较两个噪声                       ,让其差别最小          ,从而训练UNet具有噪声预测能力             。然后      ,反向时候                      ,从输入UNet的图像去掉噪声就可以获得更加清晰的图像了                       。

所以              ,这里的UNet是一个噪声预测器      。

采样过程

测试(采样)如图右边 Algorithm 2 Sampling部分:

从标准高斯分布采样一个噪声 xT∼N(0,I) ;

从时间步 T 开始正向扩散迭代到时间步 1;

如果时间步不为1   ,则从标准高斯分布采样一个噪声 z∼N(0,I)                      ,否则 z=0;

根据高斯分布计算每个时间步 t 的噪声图;

缺点总结

作者:路橙LuChengTHU

链接:https://www.zhihu.com/question/536012286/answer/2533146567

来源:知乎

变分后验分布的表达能力与计算代价的权衡一直是VAE领域的核心痛点         。

GAN还需要额外训练判别器                  ,这导致训练很困难;

normalizing flow需要模型是可逆函数,不能随便用一个图像分类or分割领域的SOTA神经网络                 ,这也导致模型表达能力受限                       。

Diffusion model最初提出是2015年的这篇Deep Unsupervised Learning using Nonequilibrium Thermodynamics                      ,这篇文章写作上跟目前diffusion model非常不一样   ,而且也没有做work             ,不建议新手读          。真正把diffusion model首次做work的工作是2020年提出的DDPM                       ,其中有许多实现上的细节      。抛开这些实现细节      ,站在2022年回看         ,我们可以揣测出2015年原作者提出这种模型的初衷                      。以下内容是我个人的理解              。

总而言之                       ,从理论角度来看          ,diffusion model的成功在于我们训练的模型只需要“模仿        ”一个简单的前向过程对应的逆向过程      ,而不需要像其它模型那样“黑盒            ”地搜索模型   。并且                      ,这个逆向过程的每一小步都非常简单              ,只需要用一个简单的高斯分布来拟合                     。这为diffusion model的优化带来了诸多便利   ,这也是它empirical performance非常好的原因之一                  。

扩散模型主要包括前向过程和后向过程:

前向过程(扩散过程)不断往输入数据中加噪声(标准高斯分布)                     ,加噪声其实是构建标签的过程                  ,时间步趋近于正无穷最后变成纯噪声。

后向过程是去噪   、不断复原的过程,也是生成目标的过程                 。

diffusion特点

在图像数据中                 ,目前只有UNET这种网络结构(跨越连接)是work的                      ,别的都不work(可以说是“完全不work                   ”)                      。 确实目前diffusion model对于低维表征还没有很solid的研究工作   ,这也是我感兴趣的方向             ,说不定哪天就想出来了[笑哭][笑哭] 我们最新的研究提出了一个叫DPM-Solver的加速采样算法                       ,证明了DDIM是diffusion ODE的一阶ODE solver      ,并且提出了二阶                     、三阶solver         ,可以做到10步采样质量很不错                       ,20步几乎收敛   。不是指噪声过程步数             。

diffusion优点:

Diffusion model最大的优势是训练简单                       。

预测噪声           ”          ,其实就是用一个二范数来训练      。Diffusion model借助了图像分割领域的UNet      ,训练loss稳定                      ,模型效果非常也好         。相比于GAN需要和判别器对抗训练或者VAE需要变分后验              ,diffusion model的loss真的是太简单了                       。

究其本质   ,其实就是我在1中提到的                     ,diffusion model只需要“模仿        ”一个非常简单的前向过程对应的逆过程即可          。

目前的训练技术让diffusion model直接跨越了GAN领域调模型的阶段                  ,而是直接可以用来做下游任务      。

得益于社区的蓬勃发展,SD能支持的图像编辑生成任务太多了                 ,都像一个智能的PS了                      ,各种编辑操作都能来   ,提示词编辑             ,各种嵌入模型                       ,各种提示词控制方法等      ,有人说GAN 能更灵活处理不同任务错误的                      。

diffusion模型缺点:

23.2

推理速度慢         ,因为噪声到图片的过程需要生成很多步                       ,每次都运行一遍神经网络          ,就慢了              。

训练速度极其消耗资源      ,要么慢                      ,要么贵   。

发展方向:

我相信diffusion model的加速采样肯定会在不久的将来彻底被解决              ,从而让diffusion model占据深度生成模型的主导                     。

作者:路橙LuChengTHU

链接:https://www.zhihu.com/question/536012286/answer/2533146567

stable diffusion

资料来源

https://www.ithome.com/0/668/981.htm

https://zhuanlan.zhihu.com/p/583124756

背景

22年8月   ,Stable Diffusion被StabilityAI 和 RunwayML 等提出并开源                  。

因为开源效果惊艳                     ,它吸引了很多人积极参与其生态的建设                  ,催生了很多的其他模型和创新,如 SD webui 项目                  、众多二次元改进项目、可用于训练 SD 新模型的 dreambooth 生态                 、可用于精准控制 SD 画面输出的交叉注意力技术方向等等。

链接:https://www.zhihu.com/question/568563436/answer/2790932301

Stable Diffusion基于Latent Diffusion Models                 。它是Latent Diffusion Models的一个具体实现                 ,具体来说                      ,它特指这样的一个模型架构设置:自动编码器下采样因子为8   ,UNet大小为860M                      。这里看到             ,sd也是扩散模型的一个小标志                       ,这个后面肯定会有更好的模型出来的吧   。

特点: 论文提出一种在潜在表示空间(latent space)上进行diffusion过程的方法      ,从而能够大大减少计算复杂度         ,同时也能达到十分不错的图片生成效果             。 相比于其它空间压缩方法(如)                       ,论文提出的方法可以生成更细致的图像          ,并且在高分辨率图片生成任务(如风景图生成      ,百万像素图像)上表现得也很好                       。 论文将该模型在**无条件图片生成****(unconditional image synthesis), 图片修复(inpainting),图片超分(super-resolution)任务上进行了实验                      ,都取得了不错的效果      。 论文还提出了cross-attention的方法来实现多模态训练              ,使得条件图片生成任务也可以实现         。论文中提到的条件图片生成任务包括类别条件图片生成(class-condition), 文图生成(text-to-image), 布局条件图片生成(layout-to-image)                       。这也为日后Stable Diffusion的开发奠定了基础          。

模型结构

Latent Diffusion Models整体框架如图   ,首先需要训练好一个自编码模型(AutoEncoder                     ,包括一个编码器 和一个解码器 )      。这样一来                  ,我们就可以利用编码器对图片进行压缩,然后在潜在表示空间上做diffusion操作                 ,最后我们再用解码器恢复到原始像素空间即可                      ,论文将这个方法称之为感知压缩(Perceptual Compression)                      。个人认为这种将高维特征压缩到低维   ,然后在低维空间上进行操作的方法具有普适性             ,可以很容易推广到文本                      、音频   、视频等领域              。

在潜在表示空间上做diffusion操作其主要过程和标准的扩散模型没有太大的区别                       ,所用到的扩散模型的具体实现为 time-conditional UNet   。但是有一个重要的地方是论文为diffusion操作引入了条件机制(Conditioning Mechanisms)      ,通过cross-attention的方式来实现多模态训练         ,使得条件图片生成任务也可以实现

                     。

图片感知压缩(Perceptual Image Compression)

感知压缩本质上是一个tradeoff(折中                       ,意思是不是啥创新)                  。

感知压缩主要利用一个预训练的自编码模型          ,该模型能够学习到一个在感知上等同于图像空间的潜在表示空间。这种方法的一个优势是只需要训练一个通用的自编码模型      ,就可以用于不同的扩散模型的训练                      ,在不同的任务上使用                 。这样一来              ,感知压缩的方法除了应用在标准的无条件图片生成外   ,也可以十分方便的拓展到各种图像到图像(inpainting                     ,super-resolution)和文本到图像(text-to-image)任务上                      。

在第一阶段训练自编码器时                  ,为了避免潜在表示空间出现高度的异化,作者使用了两种正则化方法                 ,一种是KL-reg                      ,另一种是VQ-reg   ,因此在官方发布的一阶段预训练模型中             ,会看到KL和VQ两种实现

   。在Stable Diffusion中主要采用AutoencoderKL这种实现             。

条件机制

除了无条件图片生成外                       ,我们也可以进行条件图片生成      ,这主要是通过拓展得到一个条件时序去噪自编码器(conditional denoising autoencoder) 来实现的         ,这样一来我们就可通过条件来控制图片合成的过程                       。具体来说                       ,论文通过在UNet主干网络上增加cross-attention机制来实现      。为了能够从多个不同的模态预处理          ,论文引入了一个领域专用编码器(domain specific encoder)       ,它用来将多模态数据映射为同一种中间表示                       ,这样我们就可以很方便的引入各种形态的条件(文本             、类别                       、layout等等)         。最终模型就可以通过一个cross-attention层映射将控制信息融入到UNet的中间层

这个条件结合的网络结构大概如下              ,UNet内部是多层卷积网络嘛                       。

通过给卷积层(具体说是残差块)之后增加一个attention层   ,attention层就把条件信息和卷积输出嵌入结合起来                     ,然后就实现了条件控制          。注意下图中                  ,条件信息会多次输入,才能更好的控制吧                 ,输入过程中加入了步骤时序信息                      ,所以叫条件+时序+去噪自编码器      。

论文设计了大量的对比实验   ,并分别对感知压缩权衡(下采样因子)      、LDM生成效果对比进行了分析验证                      。并且还在其他任务上进行了实验             ,包括Super-Resolution         、Inpainting                       、layout-condition在内的多种图片生成等任务                       ,这说明说明LDMs中的学习到的潜在表示空间确实具备很强的分布拟合能力      ,能够够适配各种下游任务              。

压缩倍率         ,作者重点推荐了LDM-4和LDM-8   。

文本转换模型

stable diffusion用来将文本嵌入所使用的的模型为clip                     。

CLIP官方对他的介绍就是一个用于连接图像和文本的模型                       ,也是一个里程碑式的模型          ,能够很好的进行图像文本的转换                  。

clip开创了一种新的解决视觉任务的模式      ,已经跳出了传统的分类网络的模式。

openAI使用大力出奇迹的方法                      ,采集了互联网上4亿对图像文本对进行训练                 。

这个时候clip就具有了强大的图像文本互相转换的能力                      。

然后他能做什么了              ,可以进行任意类别的图像分类   ,本且准确率很高                     ,而不是传统分类网络只能分类预设的图                  ,并且对于训练集不常见的图效果差   。所以说它跳出了传统的模式             。

现在它可以作为图像文本关联任务的一个主干了                       。类似于检测,分割等一般视觉任务里面分类主干      。

在底层技术上                 ,都是用的比较常用的技术                      ,主要创新点在于思路和数据量         。

**比如text transformer模型对文本进行编码   ,然后使用ResNet或者Vit对图像进行编码             ,然后对比编码的余弦相似度判断是否配对                       。**自然的                       ,训练时配对标签作为正样本      ,反之作为负样本         ,然后进行优化          。

整体结构                       ,来源知乎

https://zhuanlan.zhihu.com/p/599129815

链接:https://zhuanlan.zhihu.com/p/579538165

依赖

k-diffusion(为SD提供samplers(采样器)SDE(随机微分方程)和ODE(常微分方程))midas(负责为sd 深度模型提供支持)

sd webui

链接:https://zhuanlan.zhihu.com/p/579538165

创心域SEO版权声明:以上内容作者已申请原创保护,未经允许不得转载,侵权必究!授权事宜、对本内容有异议或投诉,敬请联系网站管理员,我们将尽快回复您,谢谢合作!

展开全文READ MORE
win10如何玩蜘蛛纸牌(win10蜘蛛纸牌游戏在哪? win10玩蜘蛛纸牌的技巧) ai如何替换字体(【深度学习】AI一键换天)