SOTAVerified

Selective Kernel Networks

2019-03-15CVPR 2019Code Available0· sign in to hype

Xiang Li, Wenhai Wang, Xiaolin Hu, Jian Yang

arXivPDF

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

In standard Convolutional Neural Networks (CNNs), the receptive fields of artificial neurons in each layer are designed to share the same size. It is well-known in the neuroscience community that the receptive field size of visual cortical neurons are modulated by the stimulus, which has been rarely considered in constructing CNNs. We propose a dynamic selection mechanism in CNNs that allows each neuron to adaptively adjust its receptive field size based on multiple scales of input information. A building block called Selective Kernel (SK) unit is designed, in which multiple branches with different kernel sizes are fused using softmax attention that is guided by the information in these branches. Different attentions on these branches yield different sizes of the effective receptive fields of neurons in the fusion layer. Multiple SK units are stacked to a deep network termed Selective Kernel Networks (SKNets). On the ImageNet and CIFAR benchmarks, we empirically show that SKNet outperforms the existing state-of-the-art architectures with lower model complexity. Detailed analyses show that the neurons in SKNet can capture target objects with different scales, which verifies the capability of neurons for adaptively adjusting their receptive field sizes according to the input. The code and models are available at https://github.com/implus/SKNet.

Tasks

Image Classification

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
CIFAR-10SKNet-29 (ResNeXt-29, 16×32d)Percentage correct96.53—Unverified
CIFAR-100SKNet-29 (ResNeXt-29, 16×32d)Percentage correct82.67—Unverified
ImageNetSKNet-101Top 1 Accuracy79.81—Unverified

Reproductions