Deep Rotation Equivariant Network

Junying Li,Zichen Yang,Haifeng Liu,Deng Cai
DOI: https://doi.org/10.1016/j.neucom.2018.02.029
IF: 6
2018-01-01
Neurocomputing
Abstract:Recently, learning equivariant representations has attracted considerable research attention. Dieleman et al. introduce four operations which can be inserted into convolutional neural network to learn deep representations equivariant to rotation. However, feature maps should be copied and rotated four times in each layer in their approach, which causes much running time and memory overhead. In order to address this problem, we propose Deep Rotation Equivariant Network consisting of cycle layers, isotonic layers and decycle layers. Our proposed layers apply rotation transformation on filters rather than feature maps, achieving a speed up of more than 2 times with even less memory overhead. We evaluate DRENs on Rotated MNIST and CIFAR-10 datasets and demonstrate that it can improve the performance of state-of-the-art architectures.
What problem does this paper attempt to address?