Gradient-Aligned convolution neural network

作者:

Highlights:

• We propose a general Convolution operation, called GAConv, which can replace conventional operations in CNN to help it achieve rotation invariance.

• With GAConv, Gradient-Aligned CNN (GACNN) can achieve rotation invariance without any data augmentation, feature-map augmentation, and filter enrichment.

• In GACNN, rotation invariance does not learn from the training set, but bases on the network model. Different from the vanilla CNN, GACNN will output invariant results for all rotated versions of an object, no matter whether the network is trained or not.

• We conduct classification experiments on designed dataset and realistic datasets. The results show that with the same computation cost, GACNN achieved better results than conventional CNN and some rotational invariant CNN.

摘要

•We propose a general Convolution operation, called GAConv, which can replace conventional operations in CNN to help it achieve rotation invariance.•With GAConv, Gradient-Aligned CNN (GACNN) can achieve rotation invariance without any data augmentation, feature-map augmentation, and filter enrichment.•In GACNN, rotation invariance does not learn from the training set, but bases on the network model. Different from the vanilla CNN, GACNN will output invariant results for all rotated versions of an object, no matter whether the network is trained or not.•We conduct classification experiments on designed dataset and realistic datasets. The results show that with the same computation cost, GACNN achieved better results than conventional CNN and some rotational invariant CNN.

论文关键词:Gradient alignment,Rotation equivariant convolution,Rotation invariant neural network

论文评审过程:Received 17 October 2019, Revised 18 September 2021, Accepted 25 September 2021, Available online 26 September 2021, Version of Record 4 October 2021.

论文官网地址:https://doi.org/10.1016/j.patcog.2021.108354