LLM2D
GMR-Conv:一种基于高斯混合环的高效旋转和反射不变卷积核
GMR-Conv: An Efficient Rotation and Reflection Equivariant Convolution Kernel Using Gaussian Mixture Rings
作者: Yuexi Du, Jiazhen Zhang, Nicha C. Dvornek, John A. Onofrey
发布日期: 4/4/2025
arXiv ID: oai:arXiv.org:2504.02819v1

摘要

arXiv:2504.02819v1 交叉类型 摘要:对称性,即在几何变换下某些特征保持不变,常常在设计卷积神经网络(CNNs)时作为强大的先验知识。尽管传统CNN固有支持平移不变性,但将其性质延伸到旋转和反射一直颇具挑战,通常要求在不变性、效率和信息损失之间做出妥协。在本文中,我们引入了一种高效的卷积核——高斯混合环卷积(GMR-Conv),该卷积核通过混合加权环平滑径向对称性。此设计减轻了环形核的离散化误差,从而在不增加计算开销的情况下保持鲁棒的旋转和反射不变性。我们进一步通过一种新颖的参数化和计算策略优化了GMR-Conv的空间和速度效率,使其能够在可接受的成本下使用更大的核。在八个分类和一个分割数据集上的广泛实验表明,GMR-Conv不仅能够与传统CNNs的性能相匹配,还在没有方向信息的数据应用中可以超越它们。此外,GMR-Conv也被证明比最先进的不变性学习方法更稳健和高效。我们的工作提供了启发性的实验证据,证明精心应用径向对称性可以缓解信息损失的问题,标志着在不变性网络架构方面的有希望的进步。代码可在https://github.com/XYPB/GMR-Conv获取。