RealNVP
RealNVP(Real-valued Non-Volume Preserving,实值非体积保持变换)是一类基于标准化流(normalizing flows)的深度生成模型,由 Laurent Dinh、Jascha Sohl-Dickstein 与 Samy Bengio 于 2017 年提出[1]。它的核心思想可以概括为:用一个完全可逆的神经网络,将复杂的数据分布逐步变换为简单的高斯分布。由于变换可逆,模型既可正向精确计算数据的概率密度,也可反向直接从高斯分布采样生成新数据。在标准化流的谱系中,RealNVP 属于耦合流(coupling flow)分支,与 NICE、Glow 同属一支,三者构成"加性耦合→仿射耦合→引入可逆 [math]\displaystyle{ 1\times1 }[/math] 卷积"的演进序列。RealNVP 的贡献也体现在两个层面:作为完整模型,它是标准化耦合流这一派的代表作;而其标志性的仿射耦合层又常被单独取出,作为通用可逆模块应用于标准化流之外的场合(见"应用"一节)。
历史渊源
RealNVP 的提出是沿着一条从数学工具到建模范式、再到具体架构的线索逐步成形的:先有密度估计的建模需求与变量替换公式这一数学工具,再有标准化流框架的正式提出,最后才落地为 RealNVP 等可训练的深度模型。
深度生成建模的一个核心需求,是既能精确计算数据的概率密度,又能从模型中高效采样。满足这一需求的数学工具是概率论中的变量替换公式:若变换可逆,变换前后的密度便由雅可比行列式精确联系。用可逆变换把复杂数据"归一化"到简单分布的尝试由来已久——从 1990 年代的独立成分分析(Independent Component Analysis,ICA),到 2000 年 Chen 与 Gopinath 的高斯化(Gaussianization)[2],都是这一思路的先驱。
这一思路被正式提炼为建模范式是在 2010 年:Tabak 与 Vanden-Eijnden 正式提出了"标准化流"(normalizing flow)的数学框架[3]:通过一系列可微可逆映射的复合,让基分布(base distribution,变换起点的简单分布,通常取标准高斯;与变换要逼近的目标分布相对)像水流一样"流动"成目标分布,密度则由变量替换公式精确给出。2015 年,Rezende 与 Mohamed 将这一框架引入变分推断[4],"标准化流"一词由此流行。作为一种建模范式,标准化流只规定变换必须可逆、基分布必须指定、密度由变量替换公式计算,而不限定变换的具体结构——这为不同的实现方案留出了空间。
最早把这一框架落到实处的是 2014 年 Dinh 等人提出的 NICE(Non-linear Independent Components Estimation)[5]。NICE 的关键创新是加性耦合层:将输入分成两半,其中一半直接保持不变,另一半则加上一个以第一半为条件的神经网络输出。由于该变换的雅可比行列式恒为 1,它保持体积(volume preserving),但表达能力因此受限。
2017 年的 RealNVP 在 NICE 的基础上将加性耦合扩展为仿射耦合,即以前一半变量为条件,对后一半变量先进行缩放再进行平移。这一改动使变换不再保持体积(non-volume preserving),增强了模型的表达能力,同时保留了计算上的便利性。后续的理论工作(Draxler et al., 2024)证明:体积保持的流不是分布的通用逼近器(universal approximator),而仿射耦合流在一定的良好条件下则是[6]。这从理论上解释了 RealNVP 相对于 NICE 的优势。除耦合流一支外,标准化流还有自回归流、连续流等其他实现路线(见"与相关框架的关系"一节);RealNVP 是耦合流支线上的关键一环。
值得一提的是,RealNVP 原论文通篇并未自称"标准化流"——该词在全文中仅出现于参考文献的标题里,作者当时将模型定位为基于变量替换公式、可精确计算似然的概率模型。将 RealNVP 正式归入标准化流大类下的耦合流分支,是由后来的综述文献确立的分类[7][8]。
模型架构
RealNVP 是一类标准化流。标准化流是一类通过最大似然来估计分布的模型,其构造包含三个要素:一个可微的可逆映射、一个指定的基分布,以及由变量替换公式给出的精确密度。具体地,RealNVP 将一系列可逆映射 [math]\displaystyle{ f_1, \dots, f_n }[/math] 复合为整体变换 [math]\displaystyle{ f = f_n \circ \cdots \circ f_1 }[/math],从而将数据 [math]\displaystyle{ \mathbf{x} }[/math] 映射到基分布——通常取标准正态——[math]\displaystyle{ \mathbf{z} = f(\mathbf{x}) \sim \mathcal{N}(0, \mathbf{I}) }[/math]。每一层变换的雅可比行列式都能高效解析计算,因此整体对数似然 [math]\displaystyle{ \log p_X(\mathbf{x}) }[/math] 可直接求出,无需借助变分下界或对抗训练。由于可逆性,生成样本时只需从正态分布采样 [math]\displaystyle{ \mathbf{z} }[/math],再通过逆映射 [math]\displaystyle{ \mathbf{x} = f^{-1}(\mathbf{z}) }[/math] 还原为数据。
RealNVP 把若干可逆模块按顺序堆叠,形成从数据空间到潜在空间的完整映射。其网络主体由四个功能模块组成:仿射耦合层(affine coupling layer)负责可逆的特征变换,每个耦合层输出后接批量归一化(batch normalization)以稳定训练;压缩(squeeze)负责降低空间分辨率并把信息重排到通道;变量分解(factor out)负责在每个尺度末尾抽取变量作为潜在变量。其中,仿射耦合层通过两种掩码实现输入的对半划分:空间棋盘掩码(checkerboard masking)像国际象棋棋盘一样按像素坐标的奇偶性把像素分成两组,通道级掩码(channel-wise masking)则按通道把变量分成前后两半;二者交替使用。
RealNVP 的多尺度架构如下图所示。输入图像 [math]\displaystyle{ \mathbf{x} }[/math] 首先进入第 1 尺度。每个非末尺度内部依次经过 3 个空间棋盘掩码的耦合层(图中标为 3 × (Coupling (checkerboard) → BN);凡是耦合层,输出后均接批量归一化(即块内的 BN),下文不再重复)、压缩块(图中 Squeeze)、再 3 个通道级掩码的耦合层(图中 3 × (Coupling (channel-wise) → BN)),最后进行变量分解(图中 Factor Out);末尺度则只经过 4 个空间棋盘掩码耦合层(图中 4 × (Coupling (checkerboard) → BN)),随后把全部变量抽出作为 [math]\displaystyle{ \mathbf{z}^{(L)} }[/math],不再保留传递变量。非末尺度的变量分解把当前特征一分为二:一半作为 [math]\displaystyle{ \mathbf{z}^{(i)} }[/math] 抽出,另一半 [math]\displaystyle{ \mathbf{h}^{(i)} }[/math] 继续传入下一尺度。最终把每个 [math]\displaystyle{ \mathbf{z}^{(i)} }[/math] 展平为向量后按顺序拼接(图中 Concatenation)成完整潜在变量 [math]\displaystyle{ \mathbf{z} }[/math]。虽然各 [math]\displaystyle{ \mathbf{z}^{(i)} }[/math] 的展平长度不同(依次为 [math]\displaystyle{ HWC/2, HWC/4, \dots, HWC/2^{L-1} }[/math],其中末尺度因全部抽出而与其前一尺度等长),但它们的总和恰好等于输入维度 [math]\displaystyle{ HWC }[/math],因此整体变换保持维度不变。这种分层设计的意图是:早期尺度保留高分辨率的局部细节,后期尺度处理低分辨率的全局结构,并通过逐步分解减少后续层的计算量。
正向流程可概括为:
[math]\displaystyle{ \mathbf{x} = \mathbf{h}^{(0)} \;\xrightarrow{\text{耦合层} \times n}\; \xrightarrow{\text{压缩}}\; \xrightarrow{\text{耦合层} \times n}\; \xrightarrow{\text{分解出一半}}\; \mathbf{z}^{(1)} \;\longrightarrow\; \cdots \;\longrightarrow\; \mathbf{z} }[/math]
由于每个压缩和分解步骤都要对空间或通道维度进行对半划分,各尺度的空间尺寸与通道数都必须是偶数;实际应用中通常把图像尺寸调整为偶数,或通过填充、裁剪等预处理满足该条件。
仿射耦合层:可逆的"条件变形"
下图给出两个交替堆叠的仿射耦合层的前向流程。仿射耦合层是 RealNVP 的基本可逆变换单元;图中左半部分的符号与公式 (1) 一一对应。
-
[math]\displaystyle{ \begin{cases} \mathbf{y}_1 = \mathbf{x}_1 \\[6pt] \mathbf{y}_2 = \mathbf{x}_2 \odot \exp\!\bigl(\mathbf{s}(\mathbf{x}_1)\bigr) + \mathbf{t}(\mathbf{x}_1) \end{cases} }[/math]
(1)
这里 [math]\displaystyle{ \mathbf{s} }[/math](scale,缩放)和 [math]\displaystyle{ \mathbf{t} }[/math](translation,平移)是两个深度神经网络(可以是任意结构;图像任务中通常用卷积神经网络(Convolutional Neural Network,CNN))。[math]\displaystyle{ \odot }[/math] 为逐元素乘积。其中 [math]\displaystyle{ \mathbf{x}_1 }[/math]、[math]\displaystyle{ \mathbf{x}_2 }[/math] 并非第 1、2 个坐标,而是由掩码划分出的两个子向量:[math]\displaystyle{ \mathbf{x}_1 }[/math] 是保持不变的被动部分,[math]\displaystyle{ \mathbf{x}_2 }[/math] 是被变换的主动部分。通道级掩码下二者即前、后两半通道,棋盘掩码下则是按坐标奇偶性分出的两组像素;具体规则见下文"两种掩码分区模式"。由于 [math]\displaystyle{ \mathbf{s} }[/math] 和 [math]\displaystyle{ \mathbf{t} }[/math] 的输入只来自不变的那一半,输出只作用在可变的那一半,整个变换的雅可比矩阵是下三角块结构,行列式即为对角线元素的乘积——一般的 [math]\displaystyle{ D }[/math] 维矩阵求行列式需要 [math]\displaystyle{ O(D^3) }[/math] 的计算量,而这里只需把对角元相乘,计算复杂度降为 [math]\displaystyle{ O(D) }[/math]。此外,逆变换为:
-
[math]\displaystyle{ \begin{cases} \mathbf{x}_1 = \mathbf{y}_1 \\[6pt] \mathbf{x}_2 = \bigl(\mathbf{y}_2 - \mathbf{t}(\mathbf{y}_1)\bigr) \odot \exp\!\bigl(-\mathbf{s}(\mathbf{y}_1)\bigr) \end{cases} }[/math]
(2)
无需对 [math]\displaystyle{ \mathbf{s} }[/math] 或 [math]\displaystyle{ \mathbf{t} }[/math] 求逆,只需按公式 (2) 进行减法和除法即可。因此编码(求密度)和解码(采样)同样高效。
两种掩码分区模式
仿射耦合层要先把输入分成两部分:保持不变的一半称为被动部分,被变换的一半称为主动部分。针对图像数据,RealNVP 使用两种掩码来决定划分方式:
- 空间棋盘掩码(checkerboard masking):像国际象棋棋盘一样,按坐标 [math]\displaystyle{ (i+j) }[/math] 的奇偶性决定像素是否属于"被动"集合。相邻像素被分到不同集合,能充分利用局部相关性。
- 通道级掩码(channel-wise masking):将前一半通道设为被动,后一半为主动。这通常在"压缩"操作后使用,因为此时空间分辨率已降低,通道间的语义分工更明确。
通过交替堆叠使用不同掩码的耦合层,每个维度都能在不同层中分别充当被动与主动变量,信息得以充分混合。原论文的多尺度架构中,每个尺度内同一掩码的耦合层固定为 3 层(末尺度为 4 层)[1]。
多尺度架构:分层打包潜在变量
如上图所示,多尺度架构依赖两个关键操作:压缩与变量分解。 压缩:将空间上每个 [math]\displaystyle{ 2\times2 }[/math] 的小块在通道方向上重排,将 [math]\displaystyle{ h\times w\times c }[/math] 重排为 [math]\displaystyle{ h/2 \times w/2 \times 4c }[/math]。空间分辨率减半,通道数四倍——信息被重新排列,但没有丢失。 变量分解:在每个非末尺度结束时,将当前表示的一半维度提取为潜在变量 [math]\displaystyle{ \mathbf{z}^{(i)} }[/math],剩下的一半 [math]\displaystyle{ \mathbf{h}^{(i)} }[/math] 继续进入下一尺度;末尺度则将全部变量提取为 [math]\displaystyle{ \mathbf{z}^{(L)} }[/math]。最终所有 [math]\displaystyle{ \mathbf{z}^{(i)} }[/math] 展平后拼接成完整的潜在变量 [math]\displaystyle{ \mathbf{z} }[/math],其总维度与输入维度相等。 该设计具有以下优势:
- 层次化表示:早期提取的 [math]\displaystyle{ \mathbf{z} }[/math] 对应局部纹理(如边缘、颜色),后期提取的对应全局语义(如物体姿态、场景结构)。
- 梯度分散:损失信号能直接传递到所有尺度,避免梯度在深层消失。
- 降低计算负担:越早提取的维度越多,后续层需要处理的特征图越小。
批量归一化:训练深层的稳定器
批量归一化(Batch Normalization,BN)应用于每个耦合层的输出之后。其映射为
[math]\displaystyle{ \mathbf{x} \mapsto \frac{\mathbf{x} - \tilde{\boldsymbol{\mu}}}{\sqrt{\tilde{\boldsymbol{\sigma}}^2 + \epsilon}} }[/math],
其中 [math]\displaystyle{ \tilde{\boldsymbol{\mu}} }[/math]、[math]\displaystyle{ \tilde{\boldsymbol{\sigma}} }[/math] 为均值与标准差——训练时按当前批次估计,推理时改用滑动平均(原因见下文)。由于 BN 对每个维度独立地缩放和平移,它本身也是一个可逆变换:其雅可比矩阵为对角阵,雅可比行列式即各维度缩放因子的乘积,可直接加入对数似然——这使 BN 可以像耦合层一样作为流中的一层。引入 BN 主要是为了稳定深层训练:它抑制了由缩放参数 [math]\displaystyle{ \exp(\mathbf{s}) }[/math] 引起的数值爆炸,使梯度稳定传播。使用 BN 有一个容易忽略的细节:训练时均值与方差是从当前这一批样本中估计的,因此单个样本的输出会依赖同批次的其他样本——此时的 BN 并不是严格意义上逐样本的可逆变换,其行列式也与整个批量耦合。只有在推理阶段改用训练过程中滑动平均得到的固定统计量,BN 才成为逐样本的、严格可逆的仿射变换,其行列式才能精确计入对数似然。换言之,用训练好的模型评估密度时若忘了做这一切换,同一个输入会因同批样本不同而得到不同的密度值,结果既不可复现也不正确。
数学原理
RealNVP 的数学原理可分为两个层次。其一是标准化流的通用框架:变量替换公式给出精确似然的表达式,最大似然估计提供直接的训练目标,无需近似下界或辅助网络——这些结论对一切流模型成立,并非 RealNVP 特有。其二是 RealNVP 的特有设计:通用框架要求变换可逆且雅可比行列式可算,仿射耦合层的三角雅可比结构正是为满足这一要求而构造的,它使行列式计算开销与维度成线性关系,从而保证可处理性。本节先介绍通用框架,再阐述 RealNVP 的特化实现,进而比较加性耦合与仿射耦合在体积保持上的差异,最后讨论训练目标中两项各自的含义与实际训练配置。
标准化流与变量替换公式
标准化流的核心是概率论中的变量替换公式(change of variables)。设 [math]\displaystyle{ \mathbf{z} \sim p_Z(\mathbf{z}) }[/math] 是已知先验(通常取标准正态 [math]\displaystyle{ \mathcal{N}(0, \mathbf{I}) }[/math]),[math]\displaystyle{ f = g^{-1} }[/math] 是从数据 [math]\displaystyle{ \mathbf{x} }[/math] 到 [math]\displaystyle{ \mathbf{z} }[/math] 的可逆映射(编码器)。则 [math]\displaystyle{ \mathbf{x} }[/math] 的密度为:
-
[math]\displaystyle{ p_X(\mathbf{x}) = p_Z\!\bigl(f(\mathbf{x})\bigr)\;\left|\det\!\left(\frac{\partial f(\mathbf{x})}{\partial \mathbf{x}^\top}\right)\right| }[/math]
(3)
取对数得到训练用的目标:
-
[math]\displaystyle{ \log p_X(\mathbf{x}) = \log p_Z\!\bigl(f(\mathbf{x})\bigr) + \log\left|\det\!\left(\frac{\partial f(\mathbf{x})}{\partial \mathbf{x}^\top}\right)\right| }[/math]
(4)
上式中,第一项 [math]\displaystyle{ \log p_Z(f(\mathbf{x})) }[/math] 衡量编码后分布与标准高斯的接近程度;第二项(对数雅可比行列式)则是体积补偿项——如果 [math]\displaystyle{ f }[/math] 在某个局部使空间扩张([math]\displaystyle{ \det \gt 1 }[/math]),密度必须相应降低,反之亦然。当流由 [math]\displaystyle{ K }[/math] 个变换复合而成 [math]\displaystyle{ f = f_K \circ \cdots \circ f_1 }[/math] 时,总对数密度是各层对数行列式的求和: [math]\displaystyle{ \log p_X(\mathbf{x}) = \log p_Z\!\bigl(f(\mathbf{x})\bigr) + \sum_{k=1}^K \log\left|\det\!\left(\frac{\partial f_k}{\partial \mathbf{x}_k^\top}\right)\right| }[/math] 采样时通过逆向进行:[math]\displaystyle{ \mathbf{z} \sim p_Z }[/math],[math]\displaystyle{ \mathbf{x} = g(\mathbf{z}) = f_1^{-1} \circ \cdots \circ f_K^{-1}(\mathbf{z}) }[/math],每层内部运算可并行,因此整体采样速度高于自回归模型(autoregressive model,如 PixelRNN):这类模型按固定顺序逐维生成数据、每个维度以先前各维为条件,采样必须串行进行——这与流模型对所有维度同时变换形成鲜明对比。
仿射耦合的雅可比行列式与可逆性
上一小节的结论适用于一切标准化流。RealNVP 要回答的问题是:如何构造一个既完全可逆、又满足行列式易算的变换?其答案就是仿射耦合层——它对被动半 [math]\displaystyle{ \mathbf{x}_1 }[/math] 进行恒等映射,对主动半 [math]\displaystyle{ \mathbf{x}_2 }[/math] 进行仿射映射。记被动半的维度为 [math]\displaystyle{ d }[/math](主动半的维度为 [math]\displaystyle{ D-d }[/math]),则雅可比矩阵具有分块三角结构:
[math]\displaystyle{ \frac{\partial \mathbf{y}}{\partial \mathbf{x}^\top} = \begin{bmatrix} \mathbf{I}_d & \mathbf{0} \\[8pt] \dfrac{\partial \mathbf{y}_2}{\partial \mathbf{x}_1^\top} & \operatorname{diag}\!\bigl(\exp[\mathbf{s}(\mathbf{x}_1)]\bigr) \end{bmatrix} }[/math]
由于左上角是单位阵,整个矩阵的行列式等于右下角对角阵的行列式:
-
[math]\displaystyle{ \det\!\left(\frac{\partial \mathbf{y}}{\partial \mathbf{x}^\top}\right) = \exp\!\left(\sum_{j=1}^{D-d} s_j(\mathbf{x}_1)\right) }[/math]
(5)
上述性质是 RealNVP 可处理性的关键:计算行列式不需要对神经网络 [math]\displaystyle{ \mathbf{s} }[/math] 或 [math]\displaystyle{ \mathbf{t} }[/math] 求导。因此 [math]\displaystyle{ \mathbf{s} }[/math] 和 [math]\displaystyle{ \mathbf{t} }[/math] 可以是任意复杂的深度网络(如深层卷积网络、残差网络),而不会破坏密度评估的可处理性。多个耦合层复合时,行列式相乘、逆变换按相反顺序复合,性质仍然保持。 对公式 (1) 做简单的代数移项,即可得到公式 (2) 给出的逆变换。无需对 [math]\displaystyle{ \mathbf{s} }[/math] 或 [math]\displaystyle{ \mathbf{t} }[/math] 求逆,只需按逆变换公式进行减法和除法即可。因此编码(求密度)和解码(采样)同样高效。
加性耦合、仿射耦合与体积保持
下表对比了三种耦合方式:
| 耦合类型 | 变换 | 雅可比行列式 | 体积行为 |
|---|---|---|---|
| 加性(NICE) | [math]\displaystyle{ \mathbf{y}_2 = \mathbf{x}_2 + \mathbf{t}(\mathbf{x}_1) }[/math] | [math]\displaystyle{ 1 }[/math] | 体积保持 |
| 乘法[5] | [math]\displaystyle{ \mathbf{y}_2 = \mathbf{x}_2 \odot \mathbf{s}(\mathbf{x}_1) }[/math] | [math]\displaystyle{ \prod s_j }[/math] | 非体积保持 |
| 仿射(RealNVP) | [math]\displaystyle{ \mathbf{y}_2 = \mathbf{x}_2 \odot \exp(\mathbf{s}(\mathbf{x}_1)) + \mathbf{t}(\mathbf{x}_1) }[/math] | [math]\displaystyle{ \prod \exp(s_j) }[/math] | 非体积保持 |
其中 [math]\displaystyle{ \odot }[/math] 为逐元素乘积,体现为空间的拉伸。NICE 的加性耦合对应于平移变换——只改变位置,不拉伸空间;RealNVP 的仿射耦合则允许缩放与平移的组合,更适合匹配真实数据中不同维度、不同尺度的变化。后续理论表明,缺乏这种"非体积保持"能力的流模型,在表达能力上存在本质限制[6]。
损失函数与最大似然训练
RealNVP 通过最大化数据对数似然来训练(这也是标准化流的通用训练方式),等价于最小化数据分布与模型分布的前向 KL 散度(Kullback-Leibler divergence,KL):
-
[math]\displaystyle{ \mathcal{L}(\theta) = -\mathbb{E}_{p^*(\mathbf{x})}\left[\log p_Z\!\bigl(f(\mathbf{x})\bigr) + \log\left|\det\!\left(\frac{\partial f(\mathbf{x})}{\partial \mathbf{x}^\top}\right)\right|\right] }[/math]
(6)
对于训练集 [math]\displaystyle{ \{\mathbf{x}^{(i)}\}_{i=1}^N }[/math],使用蒙特卡洛近似: [math]\displaystyle{ \mathcal{L}(\theta) \approx -\frac{1}{N}\sum_{i=1}^N \left[\log p_Z\!\bigl(f(\mathbf{x}^{(i)})\bigr) + \sum_{k=1}^K \log\left|\det\!\left(\frac{\partial f_k}{\partial \mathbf{x}_k^\top}\right)\right|\right] }[/math] 损失函数包含两项:
- 先验项 [math]\displaystyle{ \log p_Z(f(\mathbf{x})) }[/math]:促使编码后的分布整体逼近标准高斯,相当于对潜在空间的正则化。
- 行列式项:增大体积膨胀会提高似然,因此该项处处偏好扩张,与先验项的收缩压力相平衡——净效果是数据密集处扩张、稀疏处收缩,这与信息论中将码空间分配给高概率事件的原则一致。
从对偶视角看(Papamakarios et al., 2021)[8]:在数据上最小化前向 KL,等价于在潜在空间中最小化推前分布(即数据分布经 [math]\displaystyle{ f }[/math] 映射到潜在空间后得到的分布)与先验之间的反向 KL。 实际训练配置:自适应矩估计(Adaptive Moment Estimation,Adam)优化器,权重衰减 [math]\displaystyle{ 5\times10^{-5} }[/math],批量大小 64,先验为标准高斯。图像数据通常先经 logit 变换 [math]\displaystyle{ \operatorname{logit}(\alpha + (1-\alpha)\mathbf{x}/256) }[/math](原论文取 [math]\displaystyle{ \alpha = 0.05 }[/math])处理,以缓解像素取值集中在 0 和 255 两个边界带来的问题。
应用
RealNVP 的应用对应着它的两个层面:作为完整的标准化流模型,它直接服务于密度估计、生成、异常检测与重建推断等任务;作为通用可逆部件,其仿射耦合层凭借解析可逆、雅可比行列式易算的性质,也被用于概率建模之外的场合。下文按此顺序介绍。
自然图像密度估计与生成
RealNVP 最直接的应用场景是连续高维数据的密度估计与无条件生成[1]。在图像生成任务中,模型一次性学习整个图像的联合分布,然后直接从标准高斯分布采样并通过逆映射得到新图像。 在 CelebA 人脸生成任务中,RealNVP 能够生成全局连贯、五官清晰的人脸样本(见下图)。与变分自编码器(Variational Autoencoder,VAE)相比,RealNVP 不依赖固定的重建损失(如均方误差(Mean Squared Error,MSE)),因此不会过度平滑高频细节;与 PixelRNN 等自回归模型相比,RealNVP 的采样在维度上完全并行,速度更快。
异常检测
RealNVP 能够给出数据的精确似然,这使它天然适合无监督异常检测:在仅含正常样本的数据上训练流模型,测试时将似然显著低于正常水平的样本判为异常。也可以把流与重建类模型结合起来,综合似然与重建误差构造异常分数——下文的 AE-FLOW 即属此类。医学影像领域尤其典型——正常扫描数量庞大,而病变样本稀少且难以标注。AE-FLOW 将自编码器与以 RealNVP 仿射耦合层为核心的标准化流瓶颈相结合,在胸部 X 光、光学相干断层扫描、皮肤镜与脑部核磁共振等多种医学影像数据上检测异常。模型同时利用重建误差与流模型给出的负对数似然构造异常分数:正常样本能被准确重建并获得较高似然,而病变区域会产生高重建残差与低似然[9]。
下图展示了 AE-FLOW 在胸部 X 光数据上的结果。左列为异常(肺炎)样本,右列为正常样本;每列自上而下依次为原图、重建图与残差图。正常样本的残差整体较暗,而异常样本的肺部区域出现明显亮斑,直观地反映了模型对异常模式的识别能力。
重建设置中的推断
由于编码器 [math]\displaystyle{ f }[/math] 与解码器 [math]\displaystyle{ g }[/math] 互为逆映射([math]\displaystyle{ \mathbf{x} = g(f(\mathbf{x})) }[/math]),RealNVP 天然适合图像修复、去噪、着色等重建设置。以图像修复为例:已知像素 [math]\displaystyle{ \mathbf{x}_O }[/math] 固定,对缺失像素 [math]\displaystyle{ \mathbf{x}_H }[/math] 用梯度上升最大化 [math]\displaystyle{ \log p_X(\mathbf{x}_O, \mathbf{x}_H) }[/math],并加入少量高斯噪声扰动:
[math]\displaystyle{ \mathbf{x}_{H}^{(t+1)} = \mathbf{x}_{H}^{(t)} + \alpha_t \left(\frac{\partial \log p_X(\mathbf{x}_O, \mathbf{x}_{H}^{(t)})}{\partial \mathbf{x}_H} + \boldsymbol{\epsilon}\right) }[/math]
由于模型学习了全数据分布的精确密度,它能对任意形状的掩码产生合理的语义补全,而非仅记忆特定训练掩码[1]。在 LSUN 卧室数据集上,RealNVP 对缺失区域(红框)的补全结果在语义和视觉上均保持合理(见下图)。
在 NIS/NIS+ 因果涌现框架中的应用
在神经信息挤压器(Neural Information Squeezer,NIS)及其扩展 NIS+(Neural Information Squeezer Plus)中,RealNVP 的基础仿射耦合模块被用于构建可逆神经网络(Invertible Neural Networks,INN)编码器——堆叠 3 层,[math]\displaystyle{ \mathbf{s} }[/math]、[math]\displaystyle{ \mathbf{t} }[/math] 为小型前馈网络——实现微观状态到宏观状态的可逆粗粒化。NIS 原文的表述本身即体现了范畴与实例的关系:"There are several ways to realize an invertible neural network ... we select RealNVP module ... to concretely implement the invertible computation."(实现可逆神经网络有多种方式,本文选择 RealNVP 模块来具体实现可逆计算。)[10]需要指出,此处复用的是部件而非完整模型:不含多尺度架构与高斯基分布,也不以似然为训练目标;同类的部件级复用还有 RevNet 的可逆残差结构[11]与 HiNet 的图像隐藏网络[12]。NIS 中的粗粒化策略被建模为两阶段过程:
- 无损变换:由 RealNVP 耦合层堆叠构成的 INN [math]\displaystyle{ \psi_\alpha: \mathbb{R}^p \to \mathbb{R}^p }[/math] 将微观状态 [math]\displaystyle{ \mathbf{x}_t }[/math] 映射到等维潜在空间 [math]\displaystyle{ \mathbf{x}'_t }[/math]。该阶段信息不丢失。
- 信息丢弃:投影算子 [math]\displaystyle{ \chi_{p,q} }[/math] 丢弃 [math]\displaystyle{ p-q }[/math] 个维度,得到宏观状态 [math]\displaystyle{ \mathbf{y}_t = \chi_{p,q}(\psi_\alpha(\mathbf{x}_t)) }[/math]。
解码时,将宏观预测 [math]\displaystyle{ \hat{\mathbf{y}}_{t+1} }[/math] 与 [math]\displaystyle{ p-q }[/math] 维标准高斯噪声拼接,通过逆 INN [math]\displaystyle{ \psi_\alpha^{-1} }[/math] 重构微观状态。由于 RealNVP 的编码器与解码器只是方向反转,参数完全共享——相较于使用独立编码器和解码器的方案,参数量减少一半,如下图所示。
NIS 论文给出选用 RealNVP 的三点依据[10]:
- 精确可逆性:微观到潜在的变换不损失信息,信息丢弃量 [math]\displaystyle{ p-q }[/math] 成为显式可控的超参数。
- 参数共享:[math]\displaystyle{ \psi_\alpha }[/math] 与 [math]\displaystyle{ \psi_\alpha^{-1} }[/math] 共用一套参数,显著压缩模型规模。
- 行列式可解析:三角雅可比结构使有效信息(Effective Information,EI)上界和最优宏观维度 [math]\displaystyle{ q }[/math] 的存在性等核心定理得以严格推导。
RealNVP作为NIS+中可逆表示学习模块,使粗粒化映射与反粗粒化映射共享参数,降低了模型复杂度,同时为有效信息最大化提供了可分析的数学框架。此外,基于可逆粗粒化结构构建的encoder具有通用逼近性质,能够表达不同复杂系统中的粗粒化映射。[13]。
优缺点
优势
- 精确似然:RealNVP 能直接算出任意数据点的精确概率密度——不像 VAE 只能优化似然的一个下界,也不像生成对抗网络(Generative Adversarial Network,GAN)完全不给密度。因此异常检测、贝叶斯推断等需要显式概率值的任务可以直接使用它[8]。
- 并行采样与推断:自回归模型必须逐像素按顺序生成,而 RealNVP 编码和解码时所有维度同时计算,生成一张图像只需一次前向传播,采样速度快得多。
- 语义潜在空间:由于潜在分布是简单的标准高斯、且映射光滑可逆,在潜在空间中做插值等简单操作,就对应图像上有语义的连续变化(如人脸姿态、表情的渐变),适合表示学习与属性操控[1]。
- 架构灵活:耦合层只要求 [math]\displaystyle{ \mathbf{s} }[/math]、[math]\displaystyle{ \mathbf{t} }[/math] 两个网络给出缩放和平移量,对它们的内部结构没有任何限制,可以是任意深度的卷积神经网络(Convolutional Neural Network,CNN)或残差网络(Residual Network,ResNet)。提升表达能力可以直接借用主流的深层网络设计,而无须改动可逆框架本身。
- 综合多种模型特性:RealNVP 同时具备了此前难以兼得的能力——自回归模型的精确似然、VAE 的一步采样,以及不依赖固定重建损失的生成质量。
局限
- 似然性能未达最优:RealNVP 对数据分布的拟合精度仍略逊于 PixelRNN 等自回归模型[1]。在以拟合精度为核心的任务(如密度估计、数据压缩)上,这是它的相对劣势——尽管它以并行采样换来了远快于对手的生成速度。
- 维度保持约束:可逆变换要求输入与输出同维,因此 RealNVP 不能像自编码器那样把高维图像压缩成低维向量——潜在变量的总维度始终与输入相同。多尺度架构中的变量分解并不改变这一点,它只是让后续层处理更小的特征图,并把潜在变量分配到不同尺度。
- 仿射耦合的表达力上限:仿射耦合每层只对一半变量做"缩放加平移",函数形式较为简单,拟合复杂分布往往需要堆叠很多层。后续工作把变换换成更灵活的函数形式——如 Flow++ 的 Logistic 混合 CDF 耦合[14]、Neural Spline Flows 的有理二次样条耦合[15]——用更少的层数即可达到更强的拟合能力。
- 均匀去量化的次优性:像素是 0–255 的离散整数,而 RealNVP 建模的是连续分布。RealNVP 的做法是给像素加上均匀噪声,把离散数据"抹平"成连续的——但这相当于让模型拟合一个被噪声模糊过的分布,与真实分布存在偏差。Flow++ 提出的变分去量化让噪声分布本身也由模型学习,缓解了这一问题[14]。
- 分区固定性:掩码决定哪一半变量被变换,而这一划分在训练前就已固定、训练中不再改变;若某层的划分不理想,只能靠后续层弥补。Glow(Generative Flow with Invertible 1×1 Convolutions)改用可学习的可逆 [math]\displaystyle{ 1\times1 }[/math] 卷积,让模型自己学会如何重排变量,划分更灵活,生成效果也更好[16]。
- 训练显存开销大:标准反向传播需要保存每个耦合层的激活值,层数一多显存占用显著上升;RevNet 的可逆残差结构正是为省去这部分存储而提出的[11]。
- 对分布外样本过度自信:上文的异常检测应用依赖"异常样本的似然更低"这一假设,但后续研究发现这一假设未必成立:例如在 CIFAR-10 上训练的 RealNVP,会给截然不同的 SVHN 街景数字图像赋予更高的似然[17]。似然高低与"是否属于训练分布"并不直接对应,因此基于似然的异常检测需要额外的校准手段。
与相关框架的关系
RealNVP 在标准化流的发展中处于承上启下的位置。下表列举前驱与后继的代表性工作:
前驱工作
| 方法 | 年份 | 核心贡献 | 与 RealNVP 的关系 |
|---|---|---|---|
| NICE[5] | 2014 | 加性耦合层 | RealNVP 的直接前身,将加性耦合扩展为仿射耦合 |
| Rezende & Mohamed NFs[4] | 2015 | 标准化流术语 + 平面流/径向流 | 提供了变分推断中标准化流的理论框架 |
后继发展
| 方法 | 年份 | 核心创新 | 与 RealNVP 的对比 |
|---|---|---|---|
| MAF(Masked Autoregressive Flow)[18] | 2017 | 全自回归条件 | 密度评估更灵活,但采样需顺序进行 |
| Glow[16] | 2018 | 可逆 1×1 卷积 + actnorm | 刷新了图像生成基准 |
| FFJORD(Free-form Jacobian of Reversible Dynamics)[19] | 2019 | 连续时间流 + 常微分方程(Ordinary Differential Equation,ODE)描述 | 消除架构约束,但需数值积分 |
| Flow++[14] | 2019 | 变分去量化 + Logistic 混合 CDF 耦合 | 改进 RealNVP 在密度估计、去量化和耦合函数上的短板 |
| Residual Flows[20] | 2019 | 无偏行列式估计 | 将残差网络引入可逆建模 |
参考文献
- ↑ 1.0 1.1 1.2 1.3 1.4 1.5 Dinh L.; Sohl-Dickstein J.; Bengio S. (2017). "Density Estimation Using Real NVP". International Conference on Learning Representations (ICLR).
- ↑ Chen S. S.; Gopinath R. A. (2000). Gaussianization. Advances in Neural Information Processing Systems (NIPS).
- ↑ Tabak E. G.; Vanden-Eijnden E. (2010). "Density Estimation by Dual Ascent of the Log-Likelihood". Communications in Mathematical Sciences. 8 (1): 217–233.
- ↑ 4.0 4.1 Rezende D. J.; Mohamed S. (2015). "Variational Inference with Normalizing Flows". ICML.
- ↑ 5.0 5.1 5.2 Dinh L.; Krueger D.; Bengio Y. (2015). "NICE: Non-linear Independent Components Estimation". ICLR 2015 Workshop.
- ↑ 6.0 6.1 Draxler F.; Wahl S.; Schnörr C.; Köthe U. (2024). "On the Universality of Volume-Preserving and Coupling-Based Normalizing Flows". Proceedings of the 41st International Conference on Machine Learning (ICML). PMLR 235.
- ↑ Kobyzev I.; Prince S. J. D.; Brubaker M. A. (2021). "Normalizing Flows: An Introduction and Review of Current Methods". IEEE Transactions on Pattern Analysis and Machine Intelligence. 43 (11): 3964–3979.
- ↑ 8.0 8.1 8.2 Papamakarios G.; Nalisnick E.; Rezende D. J.; Mohamed S.; Lakshminarayanan B. (2021). "Normalizing Flows for Probabilistic Modeling and Inference". Journal of Machine Learning Research. 22 (57): 1–64.
- ↑ Zhao Y.; Ding Q.; Zhang X. (2023). "AE-FLOW: Autoencoders with Normalizing Flows for Medical Images Anomaly Detection". International Conference on Learning Representations (ICLR).
- ↑ 10.0 10.1 Zhang J.; Liu K. (2023). "Neural Information Squeezer for Causal Emergence". Entropy. 25 (1): 26.
- ↑ 11.0 11.1 Gomez A. N.; Ren M.; Urtasun R.; Grosse R. B. (2017). "The Reversible Residual Network: Backpropagation Without Storing Activations". NeurIPS.
- ↑ Jing J.; Deng X.; Xu M.; Wang J.; Guan Z. (2021). HiNet: Deep Image Hiding by Invertible Network. International Conference on Computer Vision (ICCV).
- ↑ Yang M.; Wang Z.; Liu K.; Rong Y.; Yuan B.; Zhang J. (2025). "Finding Emergence in Data by Maximizing Effective Information". National Science Review. 12 (1): nwae279.
- ↑ 14.0 14.1 14.2 Ho J.; Chen X.; Srinivas A.; Duan Y.; Abbeel P. (2019). "Flow++: Improving Flow-Based Generative Models with Variational Dequantization and Architecture Design". ICML.
- ↑ Durkan C.; Bekasov A.; Murray I.; Papamakarios G. (2019). "Neural Spline Flows". Advances in Neural Information Processing Systems (NeurIPS). 32.
- ↑ 16.0 16.1 Kingma D. P.; Dhariwal P. (2018). "Glow: Generative Flow with Invertible 1×1 Convolutions". NeurIPS.
- ↑ Nalisnick E.; Matsukawa A.; Teh Y. W.; Gorur D.; Lakshminarayanan B. (2019). "Do Deep Generative Models Know What They Don't Know?". International Conference on Learning Representations (ICLR).
- ↑ Papamakarios G.; Pavlakou T.; Murray I. (2017). "Masked Autoregressive Flow for Density Estimation". NIPS.
- ↑ Grathwohl W.; Chen R. T. Q.; Bettencourt J.; Sutskever I.; Duvenaud D. (2019). "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models". ICLR.
- ↑ Chen R. T. Q.; Behrmann J.; Duvenaud D.; Jacobsen J.-H. (2019). "Residual Flows for Invertible Generative Modeling". NeurIPS.
署名
本词条由 黄一凡 编写。