DDPM扩散模型数学推导

扩散过程

x 0 → x 1 → x 2 → ⋯ → x N ∼ N ( 0 , I ) x_0\rightarrow x_1 \rightarrow x_2 \rightarrow\cdots\rightarrow x_N \sim N(0,I) x0x1x2xNN(0,I)

在原始数据分布上添加高斯噪声,使之在最后成为一个纯噪声。
α t = 1 − β t \alpha_t = 1-\beta_t αt=1βt

扩散过程中 β \beta β越来越大, α \alpha α越来越小。

加噪公式:
x t = α t x t − 1 + 1 − α t ϵ t x_t = \sqrt{\alpha_t}x_{t-1}+\sqrt{1-\alpha_t}\epsilon_t xt=αt xt1+1αt ϵt
其中 ϵ ∼ N ( 0 , 1 ) \epsilon\sim N(0,1) ϵN(0,1),噪声参数增大使得在扩散过程中的不同步骤间相对的扩散幅度一致。

对于 x t − 1 x_{t-1} xt1有:
x t − 1 = α t − 1 x t − 2 + 1 − α t − 1 ϵ t − 1 x_{t-1}=\sqrt{\alpha_{t-1}}x_{t-2}+\sqrt{1-\alpha_{t-1}}\epsilon_{t-1} xt1=αt1 xt2+1αt1 ϵt1
代入上式得:
x t = α t α t − 1 x t − 2 + α t ( 1 − α t − 1 ) ϵ t − 1 + 1 − α t ϵ t x_t = \sqrt{\alpha_t\alpha_{t-1}}x_{t-2}+\sqrt{\alpha_t(1-\alpha_{t-1})}\epsilon_{t-1}+\sqrt{1-\alpha_t}\epsilon_t xt=αtαt1 xt2+αt(1αt1) ϵt1+1αt ϵt
这里有两个正态分布 ϵ t , ϵ t − 1 \epsilon_t,\epsilon_{t-1} ϵt,ϵt1,要对他们的结果进行求和,可以使用卷积,当然也可以直接根据正态分布求和的性质,可知求和后的方差为之前两个方差之和:
x t = α t α t − 1 x t − 2 + 1 − α t α t − 1 ϵ x_t = \sqrt{\alpha_t\alpha_{t-1}}x_{t-2}+\sqrt{1-\alpha_t\alpha_{t-1}}\epsilon xt=αtαt1 xt2+1αtαt1 ϵ
不断向后迭代代入,可得:
x t = ∏ i = 0 t α i x 0 + 1 − ∏ i = 0 t α i ϵ x_t = \sqrt{\prod_{i=0}^{t}\alpha_i}x_0+\sqrt{1-\prod_{i=0}^{t}\alpha_i}\epsilon xt=i=0tαi x0+1i=0tαi ϵ这里 ϵ \epsilon ϵ是另一个新的服从 N ( 0 , 1 ) \mathcal{N(0,1)} N(0,1)的分布,记 α ‾ t = ∏ i = 0 t α i \overline{\alpha}_t=\prod_{i=0}^{t}\alpha_i αt=i=0tαi,正向过程的表达式可写为
x t = α ‾ t x 0 + 1 − α ‾ t ϵ x_t = \sqrt{\overline{\alpha}_t}x_0+\sqrt{1-\overline{\alpha}_t}\epsilon xt=αt x0+1αt ϵ

逆向过程

要倒推的话,根据贝叶斯公式:
P ( x t − 1 ∣ x t ) = P ( x t ∣ x t − 1 ) P ( x t − 1 ) P ( x t ) P(x_{t-1}|x_t) = \frac{P(x_t|x_{t-1})P(x_{t-1})}{P(x_t)} P(xt1xt)=P(xt)P(xtxt1)P(xt1)写的更完整一点,加上given x 0 x_0 x0,得到
P ( x t − 1 ∣ x t , x 0 ) = P ( x t ∣ x t − 1 , x 0 ) P ( x t − 1 ∣ x 0 ) P ( x t ∣ x 0 ) P(x_{t-1}|x_t, x_0) = \frac{P(x_t|x_{t-1},x_0)P(x_{t-1}|x_0)}{P(x_t|x_0)} P(xt1xt,x0)=P(xtx0)P(xtxt1,x0)P(xt1x0)这三个概率可以分别替换成其属于的正态分布:
P ( x t ∣ x t − 1 , x 0 ) = α t x t − 1 + 1 − α t ϵ t ∼ N ( α t x t − 1 , 1 − α t ) P ( x t − 1 ∣ x 0 ) = α ‾ t − 1 x 0 + 1 − α ‾ t − 1 ϵ ∼ N ( α ‾ t − 1 x 0 , 1 − α ‾ t − 1 ) P ( x t ∣ x 0 ) = α ‾ t x 0 + 1 − α ‾ t ϵ ∼ N ( α ‾ t x 0 , 1 − α ‾ t ) P(x_t|x_{t-1},x_0) = \sqrt{\alpha_t}x_{t-1}+\sqrt{1-\alpha_t}\epsilon_t\sim\mathcal{N}(\sqrt{\alpha_t}x_{t-1},1-\alpha_t)\\ P(x_{t-1}|x_0) = \sqrt{\overline{\alpha}}_{t-1}x_{0}+\sqrt{1-\overline{\alpha}_{t-1}}\epsilon\sim\mathcal{N}(\sqrt{\overline{\alpha}}_{t-1}x_{0},1-\overline{\alpha}_{t-1})\\ P(x_{t}|x_0) = \sqrt{\overline{\alpha}}_{t}x_{0}+\sqrt{1-\overline{\alpha}_{t}}\epsilon\sim\mathcal{N}(\sqrt{\overline{\alpha}}_{t}x_{0},1-\overline{\alpha}_{t})\\ P(xtxt1,x0)=αt xt1+1αt ϵtN(αt xt1,1αt)P(xt1x0)=α t1x0+1αt1 ϵN(α t1x0,1αt1)P(xtx0)=α tx0+1αt ϵN(α tx0,1αt)
将这三个分布带入贝叶斯公式进行计算化简,得到 P ( x t − 1 ∣ x t , x 0 ) P(x_{t-1}|x_t,x_0) P(xt1xt,x0)满足的分布
P ( x t − 1 ∣ x t , x 0 ) ∼ N ( a t ( 1 − a ‾ t − 1 ) 1 − a ‾ t x t + a t − 1 ( 1 − a ‾ t ) 1 − a ‾ t x 0 , ( 1 − a t 1 − a ‾ t − 1 1 − a ‾ t ) 2 ) P(x_{t-1}|x_t,x_0)\sim\mathcal{N}(\frac{\sqrt{a_t}(1-\overline{a}_{t-1})}{1-\overline{a}_t}x_t+\frac{\sqrt{a_{t-1}}(1-\overline{a}_{t})}{1-\overline{a}_t}x_0, (\frac{\sqrt{1-a_t}\sqrt{1-\overline{a}_{t-1}}}{\sqrt{1-\overline{a}_t}})^2) P(xt1xt,x0)N(1atat (1at1)xt+1atat1 (1at)x0,(1at 1at 1at1 )2)同时,将 x 0 x_0 x0由上面的公式 x t = α ‾ t x 0 + 1 − α ‾ t ϵ x_t = \sqrt{\overline{\alpha}_t}x_0+\sqrt{1-\overline{\alpha}_t}\epsilon xt=αt x0+1αt ϵ反解带入,得到
P ( x t − 1 ∣ x t , x 0 ) ∼ N ( a t ( 1 − a ‾ t − 1 ) 1 − a ‾ t x t + a t − 1 ( 1 − a ‾ t ) 1 − a ‾ t × x t − 1 − a ‾ t × ϵ a ‾ t , ( 1 − a t 1 − a ‾ t − 1 1 − a ‾ t ) 2 ) P(x_{t-1}|x_t,x_0)\sim\mathcal{N}(\frac{\sqrt{a_t}(1-\overline{a}_{t-1})}{1-\overline{a}_t}x_t+\frac{\sqrt{a_{t-1}}(1-\overline{a}_{t})}{1-\overline{a}_t}\times\frac{x_t-\sqrt{1-\overline{a}_t}\times\epsilon}{\sqrt{\overline{a}_t}}, (\frac{\sqrt{1-a_t}\sqrt{1-\overline{a}_{t-1}}}{\sqrt{1-\overline{a}_t}})^2) P(xt1xt,x0)N(1atat (1at1)xt+1atat1 (1at)×at xt1at ×ϵ,(1at 1at 1at1 )2)
这里面只有从 x 0 x_0 x0 x t x_t xt添加的噪声 ϵ \epsilon ϵ是未知,如果我们使用神经网络来预测这里添加的 ϵ \epsilon ϵ是什么,就可以得到 x t − 1 x_{t-1} xt1遵循的正态分布,再从该分布中进行一次采样,即可得到 x t − 1 x_{t-1} xt1,如此循环以上过程,即可不断倒退到最开始的图片 x 0 x_0 x0

相关推荐

  1. DDPM扩散模型数学推导

    2024-04-06 21:32:02       39 阅读
  2. Pytorch实现扩散模型DDPM代码解读篇1】

    2024-04-06 21:32:02       31 阅读
  3. Pytorch实现扩散模型DDPM代码解读篇2】

    2024-04-06 21:32:02       32 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-04-06 21:32:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-04-06 21:32:02       100 阅读
  3. 在Django里面运行非项目文件

    2024-04-06 21:32:02       82 阅读
  4. Python语言-面向对象

    2024-04-06 21:32:02       91 阅读

热门阅读

  1. 慢SQL问题排查

    2024-04-06 21:32:02       31 阅读
  2. 反转链表(C语言)

    2024-04-06 21:32:02       43 阅读
  3. 蓝桥杯算法题:K倍区间

    2024-04-06 21:32:02       43 阅读
  4. 第六章:CSS最佳实践与优化

    2024-04-06 21:32:02       40 阅读
  5. 第十四届蓝桥杯省赛大学B组(C/C++)整数删除

    2024-04-06 21:32:02       48 阅读
  6. 抖音运营技巧2

    2024-04-06 21:32:02       40 阅读
  7. MyBatis plus 详解

    2024-04-06 21:32:02       141 阅读
  8. 谈谈Python中的正则表达式及其用法。

    2024-04-06 21:32:02       40 阅读