Circular-Symmetric Correlation Layer based on FFT

Abstract

Despite the vast success of standard planar convolutional neural networks, they are not the most efficient choice for analyzing signals that lie on an arbitrarily curved manifold, such as a cylinder. The problem arises when one performs a planar projection of these signals and inevitably causes them to be distorted or broken where there is valuable information. We propose a Circular-symmetric Correlation Layer (CCL) based on the formalism of roto-translation equivariant correlation on the continuous group $S^1\times \mathbb{R}$, and implement it efficiently using the well-known Fast Fourier Transform (FFT) algorithm. We showcase the performance analysis of a general network equipped with CCL on various recognition and classification tasks and datasets. The PyTorch package implementation of CCL is provided online.

References in corpus (7)

• Rotation-invariant convolutional neural networks for galaxy morphology prediction
• Group Equivariant Convolutional Networks
• Exploiting Cyclic Symmetry in Convolutional Neural Networks
• Equivariance Through Parameter-Sharing
• Invariant and Equivariant Graph Networks
• Permutation-equivariant neural networks applied to dynamics prediction
• CyCNN: A Rotation Invariant CNN using Polar Mapping and Cylindrical Convolution Layers

Cited by in corpus (1)

• Equivariant Deep Dynamical Model for Motion Prediction