arXiv:2606.15479cs.LGcs.AI2026-06

将贝叶斯方法与可旋转卷积结合,同时实现模型不确定性量化和对称性保持。

Bayesian 3D Steerable CNNs: Enabling Equivariance and Uncertainty Quantification Simultaneously

论文配图:Bayesian 3D Steerable CNNs: Enabling Equivariance and Uncertainty Quantification Simultaneously
图 1 · 摘自论文原文
  • 在可旋转卷积核系数上建立后验分布,生成随机核并严格保持SE(3)等变性。
  • 在分布外数据上提升6.17%准确率,预期校准误差低至0.0263。
  • 可分离认知与偶然不确定性,适合高可靠性场景如医疗或自动驾驶。

可旋转卷积神经网络(Steerable-CNN)通过基函数线性组合参数化卷积核,保证了SE(3)等变性,但其确定性特性无法进行不确定性量化,限制了在需置信度估计场景中的应用。本文提出一种贝叶斯可旋转CNN,对基函数系数施加后验分布,生成随机卷积核的同时精确保留等变性。模型损失函数通过变分推断获得,采用贝叶斯反向传播优化。该框架可分解预测不确定性为认知不确定性和偶然不确定性。实验表明,模型在分类任务中达到竞争性准确率,预期校准误差为0.0263,并在添加高斯噪声导致的分布偏移下,相比确定性模型最高提升6.17%。此外,利用不确定性估计显著提升性能,在84%测试样本上实现约4%的准确率增益。统计学上,认知不确定性与预测误差呈显著负相关,验证后验方差具有语义意义。该框架统一了贝叶斯不确定性量化与等变归纳偏置。

原文摘要 · Abstract (English)

Steerable convolutional neural networks (Steerable-CNNs) guarantee SE(3)-equivariance by parameterizing kernels as linear combinations of steerable basis functions, but their deterministic nature precludes uncertainty quantification - limiting their use in settings where confidence estimates are essential. We propose a Bayesian Steerable-CNN that places posterior distributions over the basis coefficients, yielding stochastic kernels while preserving equivariance exactly. The loss function of the model is obtained via variational inference and minimized by Bayes-by-Backpropagation. The framework admits a decomposition of predictive uncertainty into epistemic and aleatoric components. Empirically, the model attains competitive classification accuracy alongside an expected calibration error of 0.0263 and outperforms its deterministic counterpart by up to 6.17% under distributional shift induced by additive Gaussian noise. Furthermore, we leverage the model's uncertainty estimates to enhance its performance significantly, achieving a notable gain - approximately 4% higher accuracy across 84% of the test dataset. A statistically significant negative correlation between epistemic uncertainty and prediction error confirms that the learned posterior variance is semantically meaningful. The framework unifies Bayesian uncertainty quantification with the inductive bias of equivariant CNNs.

贝叶斯学习等变网络不确定性量化

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