arXiv:2602.00387stat.MLcs.LG2026-02被引 3

用低秩分解压缩贝叶斯神经网络,大幅减少参数量同时保持预测性能。

Singular Bayesian Neural Networks

  • 通过低秩矩阵分解参数化权重,诱导奇异后验以捕捉结构化相关性。
  • 在多个模型上参数减少33倍,且外推检测和校准能力优于均值场方法。
  • 适合追求高效高可靠性模型的开发者,尤其关注泛化与不确定性估计。

贝叶斯神经网络虽能提供校准的不确定性,但标准均值场高斯后验需 $O(mn)$ 参数。我们指出该成本常不必要,尤其当权重矩阵具有快速奇异值衰减时。通过将权重参数化为 $W = AB^{ op}$($A o bR^{m imes r}$, $B o bR^{n imes r}$),后验在勒贝格测度下呈奇异,集中于秩-$r$流形。这通过共享潜在因子捕获结构化权重相关性,几何上区别于均值场的独立假设。我们推导出复杂度项为 $ extstyleigO( extstyle rac{1}{2} extstyleig[ rac{ extstyle rac{}{} extstyleig]})$ 的帕累托-贝叶斯泛化界,其复杂度项缩放为 $ extstyleigO( extstyle rac{1}{2} extstyleig[ rac{ extstyle rac{}{} extstyleig]})$,并利用Eckart-Young-Mirsky定理将误差分解为优化误差与秩诱导偏差。进一步将近期针对低秩确定性网络的高斯复杂度界推广至贝叶斯预测均值。实验表明,在标准基准上的MLP、LSTM与Transformer中,该方法在使用最多比5成员深度集成少33倍参数的同时,实现竞争性预测性能;显著提升分布外检测能力,并通常优于均值场与扰动基线的校准表现,而深度集成在分布内似然指标上仍更优。

原文摘要 · Abstract (English)

Bayesian neural networks promise calibrated uncertainty but require $O(mn)$ parameters for standard mean-field Gaussian posteriors. We argue this cost is often unnecessary, particularly when weight matrices exhibit fast singular value decay. By parameterizing weights as $W = AB^{\top}$ with $A \in \mathbb{R}^{m \times r}$, $B \in \mathbb{R}^{n \times r}$, we induce a posterior that is \emph{singular} with respect to the Lebesgue measure, concentrating on the rank-$r$ manifold. This singularity captures structured weight correlations through shared latent factors, geometrically distinct from mean-field's independence assumption. We derive PAC-Bayes generalization bounds whose complexity term scales as $\sqrt{r(m+n)}$ instead of $\sqrt{m n}$, and prove loss bounds that decompose the error into optimization and rank-induced bias using the Eckart-Young-Mirsky theorem. We further adapt recent Gaussian complexity bounds for low-rank deterministic networks to Bayesian predictive means. Empirically, across MLPs, LSTMs, and Transformers on standard benchmarks, our method achieves competitive predictive performance while using up to $33\times$ fewer parameters than 5-member Deep Ensembles. It substantially improves OOD detection and often improves calibration relative to mean-field and perturbation baselines, while Deep Ensembles can still be stronger on in-distribution likelihood-based metrics.

贝叶斯神经网络低秩分解不确定性估计参数压缩

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。