Truly Shift-Invariant Convolutional Neural Networks
Anadi Chaman, Ivan Dokmanic
Abstract
Thanks to the use of convolution and pooling layers, convolutional neural networks were for a long time thought to be shift-invariant. However, recent works have shown that the output of a CNN can change significantly with small shifts in input-a problem caused by the presence of downsampling (stride) layers. The existing solutions rely either on data augmentation or on anti-aliasing, both of which have limitations and neither of which enables perfect shift invariance. Additionally, the gains obtained from these methods do not extend to image patterns not seen during training. To address these challenges, we propose adaptive polyphase sampling (APS), a simple sub-sampling scheme that allows convolutional neural networks to achieve 100% consistency in classification performance under shifts, without any loss in accuracy. With APS, the networks exhibit perfect consistency to shifts even before training, making it the first approach that makes convolutional neural networks truly shift-invariant.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers22
- Alias-Free Generative Adversarial NetworksTero Karras, Miika Aittala, Samuli Laine, Erik Härkönen et al.NeurIPS 2021 · 2,126 citations
- Approximation-Generalization Trade-offs under (Approximate) Group EquivarianceMircea Petrache, Shubhendu TrivediNeurIPS 2023 · 54 citations
- Shift Invariance Can Reduce Adversarial RobustnessVasu Singla, Songwei Ge, Ronen Basri, David W. JacobsNeurIPS 2021 · 29 citations
- Learning Layer-wise Equivariances Automatically using GradientsTycho F. A. van der Ouderaa, Alexander Immer, Mark van der WilkNeurIPS 2023 · 28 citations
- Group Equivariant SubsamplingJin Xu, Hyunjik Kim, Thomas Rainforth, Yee Whye TehNeurIPS 2021 · 26 citations
Builds on3
- How much Position Information Do Convolutional Neural Networks Encode?Md. Amirul Islam, Sen Jia, Neil D. B. BruceICLR 2020 · 392 citations
- Mind the Pad - CNNs Can Develop Blind SpotsBilal Alsallakh, Narine Kokhlikyan, Vivek Miglani, Jun Yuan et al.ICLR 2021 · 32 citations
- On Translation Invariance in CNNs: Convolutional Layers Can Exploit Absolute Spatial LocationOsman Semih Kayhan, Jan C. van GemertCVPR 2020
Related papers
- Learnable Polyphase Sampling for Shift Invariant and Equivariant Convolutional NetworksRenan A. Rojas-Gomez, Teck-Yian Lim, Alexander G. Schwing, Minh N. Do et al.NeurIPS 2022 · 21 citations
- Alias-Free Convnets: Fractional Shift Invariance via Polynomial ActivationsHagay Michaeli, Tomer Michaeli, Daniel SoudryCVPR 2023
- Group Downsampling with Equivariant Anti-aliasingMd Ashiqur Rahman, Raymond A. YehICLR 2025
- Mind the Pool: Convolutional Neural Networks Can Overfit Input SizeBilal Alsallakh, David Yan, Narine Kokhlikyan, Vivek Miglani et al.ICLR 2023
- Truly Scale-Equivariant Deep Nets with Fourier LayersMd Ashiqur Rahman, Raymond A. YehNeurIPS 2023 · 17 citations
