前置
本文把后续生成建模所需的概率基础作为自洽的预备知识保留在本章中。关于可测映射、绝对连续性与 Radon–Nikodým 导数的系统论述可参见以上页面。
符号表
记号按章节区分:在本导论与流匹配章节中,真实数据测度记为 ,其密度记为
;在扩散章节中, 表示固定的前向核或变分分布(如
), 表示学习得到的逆向模型,
表示时刻 的边缘密度。每篇扩散笔记会在开头重申局部约定。
扩散与流匹配使用两套时间坐标:扩散时间
中, 为干净数据、
为先验噪声;流匹配时间
中, 为先验、
为数据。令归一化扩散时间
,方向反转关系才是
;不能在 时直接写成 。
| 符号 | 含义 |
|---|---|
| 生成建模时的样本空间 | |
| 上的 Borel -代数 | |
| 数据列向量 | |
| 条件列向量 | |
| 单位矩阵 | |
| 多元高斯;标量情形写作 | |
| 时刻的概率测度 | |
| 时刻的概率密度 | |
| 期望 | |
| 概率 | |
| 扩散时间区间 上的均匀分布 | |
| 流匹配时间区间 上的均匀分布 | |
| SDE 漂移向量场 | |
| SDE 扩散矩阵 | |
| 线性 SDE 的标量漂移系数,即 | |
| 线性 SDE 的标量扩散系数,即 | |
| 允许相等的子集关系 | |
| 随机变量 服从分布 | |
| 期望下标表示 从分布 采样 | |
| 将 定义为 | |
| 映射或可测映射的类型记号 | |
| 对 的梯度;分数为 | |
| 散度;对方阵逐行求散度时得到向量 | |
| 对时间 的偏导 | |
| Lebesgue 积分 | |
| ODE/SDE 时间微分与布朗运动增量 | |
| 极限 | |
| 自然对数 | |
| 指数函数 | |
| 求和,包括可数可加性中的级数 | |
| 连乘 | |
| 最小化、最大化及取得最小值的参数集合 | |
| KL 散度; 表示两个分布之间的散度方向,并非范数 | |
| -散度族 | |
| 熵;在 KL 分解中与模型参数 无关 | |
| 欧氏范数 | |
| 欧氏内积 | |
| 转置 | |
| 幂集 | |
| 全称量词 |
概率基础
概率空间
概率论从以下三个基本对象出发。
定义(-代数) 设样本空间 为底层集合。若 满足:
- ;
- 蕴含 ;
- 对任意至多可数族 ,有 ,
则称 为 上的 -代数。
定义(概率测度) 映射 若满足:
- 非负性:对每个 ,;
- 可数可加性:对任意两两不交的 ,
- ,
则称 为概率测度。
定义(可测空间与概率空间) 称为可测空间; 称为概率空间。
可测映射
概率论通过可测映射把抽象事件转化为可计算的对象;在生成建模中,数据通常是 上的随机向量。
定义(可测映射) 设 与 为可测空间。若映射 满足:对任意 ,都有 ,则称
为可测映射。
定义(Borel 可测映射) 若 是拓扑空间,且分别取 Borel -代数 、,则上述可测映射称为 Borel 可测映射。
像测度
定义(像测度) 设 为测度空间, 可测。对 定义
则 是 上的测度,称为 在 下的像测度(image measure,也称推前测度)。若 是概率测度,则 也是概率测度。
随机元、随机变量与随机向量
定义(随机元、随机变量与随机向量)
- 设 为概率空间, 为度量空间。可测映射
称为 值随机元(random element)。像测度
称为 的概率分布。
- 仅当 时,实值随机元 称为随机变量(random variable)。其分布函数为
单调不减、右连续,并满足
记作 。
- 当 时, 值随机元称为随机向量(random vector),记作 ;其实现值 常简记为 。后文的数据向量均属于这一情形。
- 当 时, 值随机元称为复值随机变量(complex random variable)。
“随机变量”特指实值情形;一般值域上的对象应称为随机元。
采样
目的
生成式建模学习近似分布 ,把它作为真实分布 的代理,从而实现:
- 现实化与可控生成:通过蒙特卡洛等方法从 抽取全新、逼真且可控的样本;
- 似然评估:若模型支持显式密度,则对给定样本 计算 。
逆变换采样
逆变换采样从均匀分布出发,经目标分布 CDF 的逆映射生成样本。先考虑 连续且严格递增、因而通常反函数存在的情形。若
则对任意 ,
因此 服从目标分布。
广义逆与离散或非严格单调分布
CDF 可能存在平台或跳跃,此时传统反函数不存在,需要使用分位数意义下的广义逆。
定义(广义逆或分位数函数) 对任意 CDF ,定义
这里使用 以免与通常的函数反演混淆。对每个 ,集合 非空且有下界,故该定义取有限实值。其关键性质是
证明 记 ,于是 。
- 若 ,则 ,所以 。
- 反之,若 ,则对每个 可取 使
由 单调不减,
令 ,再用 CDF 的右连续性,得到 。证毕。
因此,若 并令 ,则
所以这一构造对离散分布以及非严格单调的分布同样成立。
生成式模型
问题与采样过程
逆变换采样可以处理一维且 CDF 可求的简单分布。然而,对图像、音频等高维、具有复杂流形结构和多峰性质的真实分布 ,通常既没有显式 CDF,也没有解析 PDF。
神经网络本身是确定性函数。为了产生随机样本,必须注入随机性,即从可直接采样的先验分布出发。无论目标分布多复杂,采样源头通常都是简单先验,例如
生成过程因而可以理解为从先验分布到数据分布的变换。核心问题是:如何用神经网络在“简单先验”与“复杂数据分布 ”之间建立桥梁?不同的生成模型给出了不同答案。
深层生成建模
深层生成模型
深层生成模型(deep generative model, DGM)利用深层神经网络学习高维数据的概率分布。
- 已知条件:从真实分布 (若存在密度则记为 )独立同分布抽取的有限数据集 。
- 核心目标:用神经网络参数化可计算的模型分布 (若存在密度则记为 ),使最优参数 对应的模型尽可能逼近 。
训练可抽象为
在两者具有密度时也常写成 。这里 是散度或任务损失。由于 未知,训练目标必须能由有限样本估计;模型优劣最终取决于生成样本与真实数据的统计相似性及具体任务指标。
分布差异的度量
前向 KL 散度
对概率测度 ,KL 散度严格定义为
若 关于同一参考测度 分别有密度 ,则
其中约定 的部分贡献 。因此最小化前向 KL 会迫使模型覆盖真实数据的全部模式:若存在正 -测度的区域满足 ,则 ,KL 直接为无穷大。
KL 最小化与最大似然
若 ,并且以下期望均有定义,则
其中
是微分熵,而常数
与 无关。若 有限,且候选模型满足所需的绝对连续性和可积性,则
不能在 或出现未定义的 时无条件使用这一等价关系。
给定 i.i.d. 样本 ,用样本均值估计期望,得到经验负对数似然
-散度
定义(-散度) 设 是同一可测空间上的概率测度,并关于参考测度 分别有密度 。令 为凸函数且 ,并定义扩展值
则
这等价于采用支撑约定
若额外假设 ,第二项消失,便得到常见的期望形式
定理(非负性与取零条件) 若 凸且 ,则 。仅凭凸性不能推出“取零当且仅当 ”;若 在 上严格凸,则
证明 令
于是 。取任意次梯度 。由凸性,
因此
若 ,显然散度为零。反之,若 严格凸且散度为零,则严格凸性排除 ,并由 Jensen 等号条件得到 ,-几乎处处成立,故 。证毕。
KL 散度是 -散度
取 ,则 。因此当 时散度为 ;当 时,
Fisher 散度
定义(Fisher 散度) 设 为开集, 是 上严格为正且一阶连续可微的密度,并且
定义
严格正性和 条件保证两个分数函数存在;更一般时也可在共同支撑内部以弱导数定义,但必须另行说明边界条件与平方可积性。
合法概率密度与配分函数
相对于参考测度 的合法概率密度必须满足:
- 非负性:,-几乎处处成立;
- 归一化:。
给定能量函数 ,先定义未归一化密度
再定义配分函数
只有在
时,
才是合法概率密度。 通常难以计算;若它发散,则该能量函数并未定义出概率分布。
生成模型分类
以下五类模型构成深层生成建模的主要谱系;后续内容重点讨论与扩散、流相关的模型。
能量模型(Energy-Based Models, EBM)
能量模型利用 定义分布,样本越真实,能量越低:
并要求 。
自回归模型(Autoregressive Models, AR)
由概率链式法则分解联合分布。对 ,
变分自编码器(Variational Autoencoders, VAE)
VAE 引入隐变量 ,以 为先验、 为解码器,并用编码器 近似后验。这里的 是变分后验,与 DDPM 前向核 记号同形但含义不同。
对固定 ,假设证据
满足 ,,且以下对数项可积,则证据下界为
若进一步有
且 KL 项有限,则
训练时最大化
若绝对连续性或可积性不成立,相应 KL 可为 、ELBO 可为 ,不能把上式当作有限实数恒等式使用。
Normalizing Flows(NF)
设先验 ,并令 。为了直接使用换元公式,要求
是 微分同胚,其逆映射也为 ,且 Jacobian 行列式处处非零。令 ,则
从而
若映射不可逆、维数不同或 Jacobian 退化,则上述普通密度换元公式不再直接适用。
生成对抗网络(Generative Adversarial Networks, GAN)
GAN 是不要求显式模型密度的 likelihood-free 模型。生成器 与判别器 进行极小极大博弈: