机器学习 | 贝叶斯神经网络

机器学习 | 贝叶斯神经网络

贝叶斯神经网络通过引入参数的概率分布来描述模型的不确定性。与传统的确定性神经网络不同，BNNs 通过学习参数的后验分布来对模型进行建模，从而在预测中提供不确定性的估计。

让$(x,\theta )\mapsto g_{\mathbf{\theta }}(\mathbf{x})$定义的$g:\mathcal{X}\times \Theta \to \mathcal{Y}$ 作为神经网络（NN），其中 $g_{\theta }(x)$ 是由 $x$ 和 $\theta$ 确定的函数。贝叶斯神经网络（BNNs）的主要假设是通过一个简单的分布来近似参数 $g$ 的后验分布$p(\theta |D)$ ，该分布在参数空间 $\Theta \subseteq {\mathbb{R}}^P$ 上编码了不确定性。

标准的点估计定义为：
$${\theta }_{\ast }=\underset{\mathbf{\theta }\in \Theta }{\mathrm{argmax}}\underset{=\log p(\mathbf{\theta }\mid \mathcal{D})-\mathrm{const}}{\underbrace{\log p(\mathcal{D}\mid \mathbf{\theta })+\log p(\mathbf{\theta })}}$$

其中 $\log p(D|\theta )$ 是对数似然损失，$\log p(\theta )$是对参数 $\theta$ 的正则化项，虽然在参数空间上可以被视为对$\Theta$的狄拉克雷分布，但并不是一个 BNN，因为根据任何标准度量（方差、熵等），它都具有零不确定性。

拉普拉斯近似（LA）

其中最简单的贝叶斯神经网络之一是拉普拉斯近似（LA），并且越来越多地被用于贝叶斯优化（BO）。给定一个（局部）最大值${\theta }^{\ast }$，拉普拉斯近似拟合了一个以 ${\theta }^{\ast }$ 为中心的高斯分布$q(\theta |D):=\mathcal{N}({\theta }^{\ast },{\Sigma }^{\ast })$，其中协方差由逆海森矩阵给出，即

$${\Sigma }^{\ast }=(-{\nabla }_{\theta }^2\log p(\theta |D){|}_{{\theta }^{\ast }}{)}^{-1}$$

线性化拉普拉斯近似（LLA）

拉普拉斯近似的一个流行类别是线性化拉普拉斯近似（LLA），它通过广义高斯-牛顿矩阵近似海森矩阵，并对神经网络在参数空间上进行线性化$g_{\theta }^{\text{lin}}(x)=g_{{\theta }^{\ast }}(x)+J^{\ast }(x)\cdot (\theta -{\theta }^{\ast })$。这里，$J^{\ast }(x)$是网络在 ${\theta }^{\ast }$ 处的雅可比矩阵$\left(\partial g/\partial \theta {|}_{{\theta }^{\ast }}\right)$ 。需要注意的是，网络函数 $x\to g_{\theta }^{\text{lin}}(x)$ 仍然是非线性的。

关键是，由于$g$在$\theta$上的线性性以及$\theta$的高斯性，输出分布 $p({\text{g}}^{\text{lin}}(x)|D)=\int {\text{g}}_{\theta }^{\text{lin}}(x)q(\theta |D)d\theta$ 也是高斯的，给出为

$$p({\text{g}}^{\text{lin}}(x)|D)=\mathcal{N}({\text{g}}_{{\theta }^{\ast }}(x),J^{\ast }(x)\Sigma J^{\ast }(x{)}^{\top })$$

超参数优化与模型校准

BNNs 的性能和稳定性在很大程度上取决于超参数的选择，例如权重衰减强度和数据噪声强度。因此，优化超参数成为了重要的研究方向。

事实上，$p({\text{g}}^{\text{lin}}|D)$是一个具有神经网络$g$作为均值函数和与经验神经切向核相关的协方差函数的高斯过程（GP）。

这些事实使得线性化拉普拉斯近似（LLA）具有解释性：它简单地为原始神经网络预测$g_{{\theta }^{\ast }}(x)$添加了一个不确定性估计。这种不确定性可以通过拉普拉斯近似的边缘似然近似进行进一步校准：

$$Z(\gamma )=\log p({\theta }^{\ast }|D;\gamma )+\frac{P}{2}\log 2\pi +\frac{1}{2}\log |{\Sigma }^{\ast }(\gamma )|$$

其中明确地表明了后验和海森矩阵对超参数$\gamma$的依赖性。例如，$\gamma$可以是一个集合，其中包含了权重衰减强度（对应于$\Theta$上的高斯先验的先验精度）以及$g$的似然中的噪声强度。在这种设置下，优化${\max }_{\gamma }\log Z(\gamma )$ 可以被看作是学习一个合适的先验并估计数据噪声。