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.
翻译:可转向卷积神经网络(Steerable-CNNs)通过将核参数化为可转向基函数的线性组合来保证SE(3)-等变性,但其确定性本质阻碍了不确定性量化——这限制了其在需要置信度估计的场景中的应用。我们提出了一种贝叶斯可转向CNN,该方法在基系数上放置后验分布,从而在严格保持等变性的同时产生随机核。该模型的损失函数通过变分推断获得,并通过贝叶斯反向传播最小化。该框架可将预测不确定性分解为认知不确定性和偶然不确定性。实验表明,该模型在达到竞争性分类准确率的同时,期望校准误差仅为0.0263,且在加性高斯噪声引起的分布偏移下,其性能比确定性对应模型高出最多6.17%。此外,我们利用模型的不确定性估计显著提升了性能——在84%的测试数据集上准确率提升了约4%。认知不确定性与预测误差之间统计显著的负相关性证实了学习到的后验方差具有语义意义。该框架将贝叶斯不确定性量化与等变CNN的归纳偏置统一起来。