Abstract
Convolutional neural networks (CNNs) are essential tools for computer visiontasks, but they lack traditionally desired properties of extracted featuresthat could further improve model performance, e.g., rotational equivariance.Such properties are ubiquitous in biomedical images, which often lack explicitorientation. While current work largely relies on data augmentation or explicitmodules to capture orientation information, this comes at the expense ofincreased training costs or ineffective approximations of the desiredequivariance. To overcome these challenges, we propose a novel and efficientimplementation of the Symmetric Rotation-Equivariant (SRE) Convolution(SRE-Conv) kernel, designed to learn rotation-invariant features whilesimultaneously compressing the model size. The SRE-Conv kernel can easily beincorporated into any CNN backbone. We validate the ability of a deep SRE-CNNto capture equivariance to rotation using the public MedMNISTv2 dataset (16total tasks). SRE-Conv-CNN demonstrated improved rotated image classificationperformance accuracy on all 16 test datasets in both 2D and 3D images, allwhile increasing efficiency with fewer parameters and reduced memory footprint.The code is available at https://github.com/XYPB/SRE-Conv.