引言
概率深度学习已成为提升实际决策场景下模型可靠性、可解释性和鲁棒性的关键方向。与产生单点预测的确定性神经网络不同,概率模型显式地考虑不确定性,使其能够表示参数和观测中的歧义(Gal, Ghahramani, 2016, Kendall, Gal, 2017)。然而,许多现有的贝叶斯深度学习框架通过简单的各向同性噪声模型或平均场假设来近似不确定性(Abdar et al., 2021),无法捕获高维参数空间中存在的结构化和几何感知的变异性。因此,不确定性估计往往校准不佳且在分布偏移下不稳定,限制了它们在安全关键应用中的实用性(Krishnan, Tickoo, 2020, Taori, Dave, Shankar, Carlini, Recht, Schmidt, 2020)。
为解决这些局限,本文提出了量子启发的变分卷积(QiVC)框架,直接在卷积层中引入几何和变分的不确定性处理方式。该框架的核心创新是量子启发的旋转集成(QiRE)机制,它对卷积核应用可微的低维子空间旋转。该操作受量子系统中幺正演化的启发,保持了权重空间的内在几何和范数(Sakurai and Napolitano, 2020)。通过在随机扰动中强制施加相干结构,QiVC层实现了更丰富的不确定性表示,同时避免了各向同性或非结构化高斯扰动的不稳定性和表达力局限性。重要的是,QiVC不引入额外的可学习参数;相反,它对卷积滤波器进行结构化重参数化,以最小计算开销生成富有表现力且基于概率的集成。
在量子启发的神经架构(Beer, Bondarenko, Farrelly, Osborne, Salzmann, Scheiermann, Wolf, 2020, Li, Salek, Roy, Wang, & Chowdhury)和函数变分推断(Sun et al., 2019)的先验发展基础上,QiVC提供了一种有原则的方法,将不确定性直接嵌入到模型的功能表示中。通过统一变分学习与几何保持变换,它架起了贝叶斯推断与结构化集成建模之间的桥梁,产生了适用于决策关键领域的稳定和可解释的特征表示。
在本工作中,QiVC框架被实例化在一个卷积网络中,称为QiVC-Net,其有效性在心音图(PCG)分类背景下进行了评估。PCG信号分析在计算机辅助听诊和心血管疾病早期检测中发挥着至关重要的作用。心音记录提供了具有临床意义的诊断线索,但其解释受到低信噪比、高患者间变异性以及实际场景中典型的环境伪影的复杂化影响(Sanei et al., 2020)。深度学习领域的最新进展通过CNN、RNN、Transformer和混合模型提高了PCG分类性能(Ameen, Fattoh, Abd El-Hafeez, Ahmed, 2024, Zhao, Geng, Wang, et al., 2024)。然而,在鲁棒性、可解释性、数据集不平衡以及缺乏可靠的不确定性量化方面仍存在持续的挑战(Azam, Ansari, Nuhash, McLane, Hasan, 2022, Dwivedi, Imtiaz, Rodriguez-Villegas, 2018, Fernando, Gammulle, Denman, Sridharan, Fookes, 2021)。
为了增强时间连贯性和对生理变异性的鲁棒性,QiVC-Net进一步引入了反转融合残差(RFR)模块,用于对PCG记录中的双向时间依赖关系进行建模。这种几何感知的概率卷积与时间对称性的结合,使QiVC-Net非常适合在噪声和异质条件下进行生物信号分析。
本工作的主要贡献总结如下:
- •
QiVConv层:一种概率卷积层,引入保持范数的子空间旋转,在增强不确定性校准的同时保持富有表现力的核结构。
- •
QiRE采样:一种结构化随机扰动机制,对卷积权重执行可微的受幺正启发的旋转,无需复数算术即可实现类贝叶斯推断。
- •
不确定性感知双路径架构:QiVConv与RFR模块的集成,用于捕获PCG信号中的双向时间动力学,同时不添加额外的可学习参数。
- •
临床对齐设计:一个计算高效且可解释的框架,统一了不确定性量化、对信号变化的鲁棒性和时间连贯性,以提供可靠的基于PCG的决策支持。