arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.06934cs.CV

尺度-旋转等变李群卷积神经网络(Lie Group-CNNs)

Scale-Rotation-Equivariant Lie Group Convolution Neural Networks (Lie Group-CNNs)

Wei-Dong Qiao, Yang Xu, Hui Li

AI总结:

针对现有CNN缺乏尺度-旋转联合等变性的问题,提出基于SIM(2)李群的Lie Group-CNN,通过李群空间映射与李代数参数化群卷积实现等变特征提取,在多数据集上取得优于同类方法的准确率与鲁棒性。

摘要:

卷积核的权值共享机制保证了卷积神经网络(CNNs)的平移等变性。近年来,旋转等变性已得到研究,但针对尺度等变性或同时具备尺度-旋转等变性的研究仍不充分。本研究提出一种Lie Group-CNN(李群卷积神经网络),可在图像分类任务中保持尺度-旋转等变性。\nLie Group-CNN包含提升模块、一系列群卷积模块、全局池化层和分类层。提升模块将输入图像从欧几里得空间转换到李群空间,群卷积通过以李群元素的李代数为输入的全连接网络进行参数化,从而实现尺度-旋转等变性。研究采用SIM(2)李群构建具备尺度-旋转等变性的Lie Group-CNN。\n实验验证了Lie Group-CNN的尺度-旋转等变性,其在血细胞数据集上取得97.50%的最佳识别准确率,在HAM10000数据集上达到77.90%的准确率,优于李代数卷积网络、空洞卷积、空间变换网络和尺度等变可操纵网络。此外,在rotated-MNIST和rotated-CIFAR10上验证了基于SIM(2)的Lie Group-CNN在旋转等变性方面的泛化能力,并在SO(2)和SE(2)上验证了网络的鲁棒性。因此,Lie Group-CNN能够成功提取几何特征,并对存在旋转和尺度变换的图像执行等变识别。

英文摘要:

The weight-sharing mechanism of convolutional kernels ensures translation-equivariance of convolution neural networks (CNNs). Recently, rotation-equivariance has been investigated. However, research on scale-equivariance or simultaneous scale-rotation-equivariance is insufficient. This study proposes a Lie group-CNN, which can keep scale-rotation-equivariance for image classification tasks. The Lie group-CNN includes a lifting module, a series of group convolution modules, a global pooling layer, and a classification layer. The lifting module transfers the input image from Euclidean space to Lie group space, and the group convolution is parameterized through a fully connected network using Lie-algebra of Lie-group elements as inputs to achieve scale-rotation-equivariance. The Lie group SIM(2) is utilized to establish the Lie group-CNN with scale-rotation-equivariance. Scale-rotation-equivariance of Lie group-CNN is verified and achieves the best recognition accuracy on the blood cell dataset (97.50%) and the HAM10000 dataset (77.90%) superior to Lie algebra convolution network, dilation convolution, spatial transformer network, and scale-equivariant steerable network. In addition, the generalization ability of the Lie group-CNN on SIM(2) on rotation-equivariance is verified on rotated-MNIST and rotated-CIFAR10, and the robustness of the network is verified on SO(2) and SE(2). Therefore, the Lie group-CNN can successfully extract geometric features and performs equivariant recognition on images with rotation and scale transformations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑