提出可证明尺度不变的残差网络,解决模型对未见缩放图像泛化差的问题。
Scale-invariant Gaussian derivative residual networks
- 基于高斯导数残差块级联构建,具备尺度协变与尺度不变性
- 在三类数据集上验证,对4倍缩放范围的图像保持强泛化能力
- 采用深度可分离卷积减少参数量和计算量,适合实际部署
跨图像尺度的泛化仍是深度网络的根本挑战,通常无法处理训练中未见尺度的图像(分布外问题)。本文提出可证明尺度不变的高斯导数残差网络(GaussDerResNets),由级联的尺度协变高斯导数残差块构成,旨在解决该问题。通过在原有高斯导数层基础上添加残差跳跃连接,可在保持良好尺度泛化性的同时,构建更深、精度更高的网络。文中给出了任意维度下尺度协变与尺度不变性质的严格证明。为评估GaussDerResNets在新尺度上的泛化能力,我们在新构建的STL-10重缩放版本上进行实验:训练仅在一个固定尺度下进行,测试则在多个分别缩放到不同空间尺度的测试集副本上完成,缩放因子覆盖4倍范围。类似实验也在重缩放后的Fashion-MNIST和CIFAR-10数据集上展开。实验表明,GaussDerResNets在三类数据集上均表现出优异的尺度泛化与尺度选择能力。消融研究进一步表明,基于深度可分离卷积的架构设计可显著降低参数量和计算量,同时维持合理的精度与尺度泛化性能。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。