4、贝叶斯深度学习基础:贝叶斯定理、先验/后验分布、变分推断、蒙特卡洛采样、马尔可夫链蒙特卡洛

贝叶斯深度学习(Bayesian Deep Learning)并非简单地将神经网络与贝叶斯公式叠加,而是从认识论层面重新定义了模型参数的“不确定性”。在PINN(物理信息神经网络)的语境下,我们不仅要拟合偏微分方程的残差,还要回答一个更深刻的问题:在有限、含噪的观测数据下,我们对解场的估计有多大的置信度? 本章作为贝叶斯PINN不确定性量化的基石,将系统梳理贝叶斯推断的核心数学框架、近似计算手段以及采样策略,为后续将不确定性嵌入PINN损失函数奠定方法论基础。

贝叶斯深度学习核心逻辑链 贝叶斯定理 P(θ|D) ∝ P(D|θ)·P(θ) 先验 P(θ) 物理约束/经验 似然 P(D|θ) 数据拟合项 后验 P(θ|D) 不确定性量化核心目标 后验不可解析求解 高维积分无闭式解 变分推断 (VI) ELBO最大化 · 近似分布q(θ) MCMC采样 MH / HMC · 无偏渐近精确 蒙特卡洛采样 重要性采样 · 重参数化 预测分布 P(y*|x*,D) = ∫P(y*|x*,θ)P(θ|D)dθ 不确定性分解 认知 + 偶然不确定性 应用:贝叶斯PINN → 物理场预测 + 置信区间 + 主动学习

4.1 核心原理:从贝叶斯定理到后验分布

贝叶斯深度学习的出发点是将神经网络权重 θ 视为随机变量。给定训练数据集 D = {(xᵢ, yᵢ)},贝叶斯定理给出了参数的后验分布:

P(θ|D) = P(D|θ) · P(θ) / P(D)
       ∝ P(D|θ) · P(θ)

其中 P(θ) 为先验分布,编码了我们在看到数据之前对参数的信念;P(D|θ) 为似然函数,在回归任务中通常假设高斯噪声,即 P(D|θ) = ∏ᵢ N(yᵢ; f_θ(xᵢ), σ²);P(D) 为边际似然(证据),是一个与 θ 无关的归一化常数。在PINN场景中,先验不仅包含参数的正则化约束,还可以通过物理方程残差引入软约束,例如将 PDE 残差作为先验能量项的一部分。

后验分布 P(θ|D) 是贝叶斯推断的完整解。它给出了参数在观测数据下的所有可能取值及其概率密度。有了后验,我们可以计算预测分布:

P(y*|x*, D) = ∫ P(y*|x*, θ) · P(θ|D) dθ

这个积分正是不确定性量化的数学根源——它综合了所有可能的参数配置,而不是像点估计那样只依赖单一最优参数。然而,对于深度神经网络,这个积分没有解析解,且维度极高(动辄百万级参数),因此必须借助近似方法。

关键认知: 贝叶斯深度学习与正则化的本质区别在于,正则化只改变最优参数的位置,而贝叶斯方法同时刻画了参数分布的“宽度”。在PINN中,这意味着我们不仅能给出解场的均值预测,还能给出每个空间点的方差,从而判断哪些区域数据不足或物理约束冲突。

4.2 先验与后验:如何选择与解释

先验分布的选择直接影响后验的形状和推断结果。在贝叶斯PINN中,先验通常包含两类信息:

先验类型 数学形式 物理含义 典型场景
高斯先验(L2正则) θ ~ N(0, λ⁻¹I) 参数能量有限,偏向小范数 标准PINN权重初始化
拉普拉斯先验(L1正则) θ ~ Laplace(0, b) 鼓励稀疏性,特征选择 高维输入特征筛选
物理信息先验 P(θ) ∝ exp(-λ·‖R(u_θ)‖²) PDE残差越小,先验概率越高 流体力学、热传导方程约束
分层先验 θ ~ N(0, α⁻¹), α ~ Gamma 自动调节正则化强度 数据噪声水平未知时

后验分布的解释需要谨慎。后验的均值可以作为点估计的替代,但更重要的是后验的方差——它反映了参数估计的不确定性。在PINN中,如果某个区域的后验方差很大,说明该区域的解场对数据或物理约束不敏感,需要补充观测点或加强物理约束。

常见误区: 将先验视为“惩罚项”是片面的。先验不仅影响最优解的位置,还影响后验的曲率(即不确定性)。一个过于强的高斯先验(λ很大)会人为压缩后验方差,导致不确定性被低估。在贝叶斯PINN中,先验的强度应与数据量、物理约束的可信度相匹配。

4.3 变分推断:用优化替代积分

变分推断(Variational Inference, VI)的核心思想是:用一个参数化的分布族 q_φ(θ) 去逼近真实后验 P(θ|D),通过最小化两个分布之间的 KL 散度来实现。由于 KL 散度直接计算需要后验归一化常数,实际优化目标是证据下界(ELBO):

ELBO(φ) = E_{q_φ(θ)}[log P(D|θ)] - KL[q_φ(θ) || P(θ)]

ELBO 的第一项是期望对数似然,鼓励模型拟合数据;第二项是负的 KL 散度,鼓励近似分布接近先验。最大化 ELBO 等价于最小化 KL(q_φ || P(θ|D))。在深度学习中,最常用的变分族是均值场高斯分布:q_φ(θ) = ∏ᵢ N(θᵢ; μᵢ, σᵢ²),其中 φ = {μᵢ, σᵢ²} 为变分参数。

实际操作中,ELBO 的期望项通过蒙特卡洛采样估计,并使用重参数化技巧(Reparameterization Trick)保持梯度传播:

θ = μ + σ ⊙ ε,  ε ~ N(0, I)
∇_φ ELBO ≈ (1/M) Σ_{m=1}^M ∇_φ log P(D|θ^(m)) - ∇_φ KL[q_φ || P]

这种“贝叶斯神经网络”的训练方式称为 Bayes by Backprop。在PINN中,变分推断的额外优势在于:我们可以将 PDE 残差作为额外的似然项或先验项纳入 ELBO,从而在优化过程中同时考虑数据拟合和物理一致性。

工程建议: 对于大规模PINN,全连接层的变分参数数量是普通权重的两倍(均值和方差)。建议先训练一个确定性PINN作为初始化,再切换到变分训练,可以显著加速收敛并避免方差参数过早坍缩。

4.4 蒙特卡洛采样与马尔可夫链蒙特卡洛

蒙特卡洛(Monte Carlo, MC)方法通过从目标分布中抽取大量样本,用样本均值近似期望。在贝叶斯推断中,MC 方法直接用于估计后验统计量:

E_{P(θ|D)}[f(θ)] ≈ (1/N) Σ_{n=1}^N f(θ^(n)),  θ^(n) ~ P(θ|D)

然而,直接从后验采样通常不可行,因为后验的归一化常数未知且高维分布复杂。马尔可夫链蒙特卡洛(MCMC)方法通过构造一条马尔可夫链,使其平稳分布恰好为目标后验分布,从而间接实现采样。最基础的 Metropolis-Hastings (MH) 算法流程如下:

  1. 初始化:选择初始参数 θ⁰,设定提议分布 q(θ'|θ)(如高斯随机游走)。
  2. 提议:从 q(θ'|θᵗ) 中抽取候选样本 θ'。
  3. 接受/拒绝:计算接受概率 α = min(1, [P(θ'|D)·q(θᵗ|θ')] / [P(θᵗ|D)·q(θ'|θᵗ)]),以概率 α 接受 θ',否则保持 θᵗ。
  4. 迭代:重复步骤 2-3 直到链收敛,丢弃前 B 个样本(burn-in),用剩余样本估计后验统计量。

对于高维参数空间,MH 算法的随机游走效率极低。Hamiltonian Monte Carlo (HMC) 利用梯度信息模拟哈密顿动力学,可以大幅提高采样效率。No-U-Turn Sampler (NUTS) 是 HMC 的自动调参版本,在 PyMC、Stan 等概率编程框架中广泛使用。在贝叶斯PINN中,由于参数维度通常较高(数万至数百万),全量 MCMC 的计算代价往往不可接受,因此实践中常采用以下折中策略:

  • 子采样 MCMC:每次迭代只使用小批量数据计算似然,引入随机梯度噪声。
  • 随机梯度 Langevin 动力学 (SGLD):在 SGD 更新中加入高斯噪声,使得参数轨迹收敛到后验分布。
  • 变分 + MCMC 混合:先用 VI 得到近似后验作为提议分布,再用 MCMC 做精修。
对比总结: VI 与 MCMC 各有优劣。VI 收敛快、可扩展性强,但近似误差有偏(均值场假设限制了分布族);MCMC 理论无偏、渐近精确,但计算成本高、收敛诊断困难。在贝叶斯PINN中,推荐先用 VI 快速获得不确定性估计,再用少量 MCMC 样本验证关键区域的后验形状。

4.5 案例拆解:一维热传导方程的贝叶斯PINN

考虑一维稳态热传导方程:d²u/dx² = f(x),边界条件 u(0)=0, u(1)=1。我们假设真实解为 u(x) = x³,源项 f(x) = 6x。在 x = [0.2, 0.4, 0.6, 0.8] 四个位置添加含噪观测数据(σ=0.05),目标是使用贝叶斯PINN估计解场并量化不确定性。

模型设置: 使用一个 3 层全连接网络(隐藏层 20 个神经元,tanh 激活)。先验选择标准高斯 N(0, 1)。似然假设高斯噪声,噪声方差 σ² 作为未知超参数,赋予 Inverse-Gamma 先验。物理约束通过将 PDE 残差 r(x) = d²u_θ/dx² - 6x 作为额外的似然项加入,残差方差 τ² 也作为未知量。

训练过程: 采用变分推断,ELBO 包含三项:数据似然期望、PDE 残差似然期望、KL 散度。使用 Adam 优化器,学习率 0.01,训练 5000 步。变分参数初始化为确定性训练结果附近。

结果对比:

评估指标 确定性PINN 贝叶斯PINN (VI) 贝叶斯PINN (HMC)
均方误差 (MSE) 3.2e-3 2.8e-3 2.5e-3
边界处90%置信区间宽度 — 0.12 0.15
数据稀疏区域(0.5-0.7)区间宽度 — 0.28 0.31
训练时间 2分钟 8分钟 45分钟

分析: 贝叶斯PINN在数据稀疏区域给出了更宽的置信区间,正确反映了信息不足带来的不确定性。HMC 的区间略宽于 VI,说明均值场假设略微低估了后验方差。确定性PINN无法提供任何不确定性信息,在安全关键应用中存在风险。

风险提示: 贝叶斯PINN的不确定性估计可能被“虚假自信”误导。如果物理约束的残差方差 τ² 被过度优化(趋向于零),模型会认为物理方程完全可信,从而压缩所有不确定性。建议对 τ² 设置合理的下界,或使用分层先验让数据自适应调整。

4.6 常见误区与总结复盘

误区一:将 Dropout 视为真正的贝叶斯推断。 MC Dropout 虽然可以近似不确定性,但它等价于一种特殊的变分推断,其近似分布族受限且没有显式的先验控制。在PINN中,MC Dropout 无法自然地融入 PDE 残差约束,因此不推荐作为主要的不确定性量化手段。

误区二:忽视先验敏感性分析。 后验分布对先验的选择并非总是不敏感。当数据量少或信噪比低时,先验的强度会显著影响后验的宽度。建议对先验的超参数(如高斯先验的方差)进行敏感性分析,绘制后验均值/方差随先验强度变化的轨迹图。

误区三:混淆认知不确定性与偶然不确定性。 认知不确定性(Epistemic)来自参数估计的不确定性,可以通过增加数据或物理约束来降低;偶然不确定性(Aleatoric)来自数据本身的噪声,无法通过增加数据消除。在贝叶斯PINN中,两者通过不同的方差项体现:参数后验方差对应认知不确定性,似然噪声方差对应偶然不确定性。混淆两者会导致错误的主动学习策略。

总结: 贝叶斯深度学习为PINN提供了严谨的不确定性量化框架。核心链路为:贝叶斯定理定义后验 → 变分推断或MCMC近似后验 → 预测分布整合参数不确定性 → 分解认知/偶然不确定性。在实际工程中,应根据问题规模、计算预算和精度要求选择合适的近似方法。对于中小规模PINN,HMC/NUTS 可以提供黄金标准的不确定性估计;对于大规模问题,VI 及其变体(如 Bayes by Backprop、MC Dropout)是更务实的选择。下一阶段,我们将把这些方法嵌入PINN的损失函数设计中,实现物理场预测与置信区间的联合输出。