Scale-invariant Gaussian derivative residual networks

TL;DR

This paper introduces scale-invariant Gaussian derivative residual networks (GaussDerResNets) that maintain high accuracy and generalize well across different image scales, addressing a key challenge in deep learning for vision tasks.

Contribution

The paper proposes a novel scale-covariant residual network architecture with provable scale-invariance, enhancing scale generalization in deep networks.

Findings

GaussDerResNets exhibit strong scale generalization across multiple datasets.

Architectural variants with depthwise-separable convolutions reduce parameters and computation.

Theoretical proofs confirm scale-covariance and scale-invariance properties.

Abstract

Generalisation across image scales remains a fundamental challenge for deep networks, which often fail to handle images at scales not seen during training (the out-of-distribution problem). In this paper, we present provably scale-invariant Gaussian derivative residual networks (GaussDerResNets), constructed out of scale-covariant Gaussian derivative residual blocks coupled in cascade, aimed at addressing this problem. By adding residual skip connections to the previous notion of Gaussian derivative layers, deeper networks with substantially increased accuracy can be constructed, while preserving very good scale generalisation properties at the higher level of accuracy. Explicit proofs are provided regarding the underlying scale-covariant and scale-invariant properties in arbitrary dimensions. To analyse the ability of GaussDerResNets to generalise to new scales, we apply them on the new rescaled version of the STL-10 dataset, where training is done at a single fixed scale and evaluation is performed on multiple copies of the test set, each rescaled to a single distinct spatial scale, with scale factors extending over a range of 4. We also conduct similar systematic experiments on the rescaled versions of Fashion-MNIST and CIFAR-10 datasets. Experimentally, we demonstrate that the GaussDerResNets have strong scale generalisation and scale selection properties on all the three rescaled datasets. In our ablation studies, we investigate different architectural variants of GaussDerResNets, demonstrating that basing the architecture on depthwise-separable convolutions allows for decreasing both the number of parameters and the amount of computations, with reasonably maintained accuracy and scale generalisation.