Making Convolutional Networks Shift-Invariant Again

Code

• github.com/adobe/antialiased-cnns
  Officialpytorch★ 0
• github.com/rwightman/pytorch-image-models
  pytorch★ 36,538
• github.com/mauriceweiler/MobiusCNNs
  pytorch★ 80
• github.com/mnikitin/Shift-Invariant-CNNs
  mxnet★ 0
• github.com/kornia/kornia/blob/e4a54c2d4fd9b8eaa88dec50061e33f7aee5d959/kornia/filters/blur_pool.py
  pytorch★ 0
• github.com/ricky40403/BlurPool
  none★ 0
• github.com/tattaka/Antialiased-CNNs-Converter-PyTorch
  pytorch★ 0

Abstract

Modern convolutional networks are not shift-invariant, as small input shifts or translations can cause drastic changes in the output. Commonly used downsampling methods, such as max-pooling, strided-convolution, and average-pooling, ignore the sampling theorem. The well-known signal processing fix is anti-aliasing by low-pass filtering before downsampling. However, simply inserting this module into deep networks degrades performance; as a result, it is seldomly used today. We show that when integrated correctly, it is compatible with existing architectural components, such as max-pooling and strided-convolution. We observe increased accuracy in ImageNet classification, across several commonly-used architectures, such as ResNet, DenseNet, and MobileNet, indicating effective regularization. Furthermore, we observe better generalization, in terms of stability and robustness to input corruptions. Our results demonstrate that this classical signal processing technique has been undeservingly overlooked in modern deep networks. Code and anti-aliased versions of popular networks are available at https://richzhang.github.io/antialiased-cnns/ .

Tasks

Benchmark Results

Reproductions