Abstract
Many deep neural networks are built by using stacked convolutional layers of fixed and single size (often 3 × 3) kernels. This paper describes a method for learning the size of convolutional kernels to provide varying size kernels in a single layer. The method utilizes a differentiable, and therefore backpropagation-trainable Gaussian envelope which can grow or shrink in a base grid. Our experiments compared the proposed adaptive layers to ordinary convolution layers in a simple two-layer network, a deeper residual network, and a U-Net architecture. The results in the popular image classification datasets such as MNIST, MNIST-CLUTTERED, CIFAR-10, Fashion, and “Faces in the Wild” showed that the adaptive kernels can provide statistically significant improvements on ordinary convolution kernels. A segmentation experiment in the Oxford-Pets dataset demonstrated that replacing ordinary convolution layers in a U-shaped network with 7 × 7 adaptive layers can improve its learning performance and ability to generalize.
Original language | English |
---|---|
Article number | 103015 |
Journal | Journal of Visual Communication and Image Representation |
Volume | 75 |
DOIs | |
Publication status | Published - Feb 2021 |
Externally published | Yes |
Bibliographical note
Publisher Copyright:© 2021 Elsevier Inc.
Keywords
- Adaptive convolution
- Image classification
- Multi-scale convolution
- Residual networks