Dynamic Convolution: Attention over Convolution Kernels

Code

• github.com/xmu-xiaoma666/External-Attention-pytorch
  pytorch★ 12,169
• github.com/mindspore-courses/External-Attention-MindSpore/blob/main/model/conv/DynamicConv.py
  mindspore★ 0
• github.com/prstrive/CondConv-tensorflow
  tf★ 0
• github.com/kaijieshi7/Dynamic-convolution-Pytorch
  pytorch★ 0
• github.com/TArdelean/DynamicConvolution
  pytorch★ 0

Abstract

Light-weight convolutional neural networks (CNNs) suffer performance degradation as their low computational budgets constrain both the depth (number of convolution layers) and the width (number of channels) of CNNs, resulting in limited representation capability. To address this issue, we present Dynamic Convolution, a new design that increases model complexity without increasing the network depth or width. Instead of using a single convolution kernel per layer, dynamic convolution aggregates multiple parallel convolution kernels dynamically based upon their attentions, which are input dependent. Assembling multiple kernels is not only computationally efficient due to the small kernel size, but also has more representation power since these kernels are aggregated in a non-linear way via attention. By simply using dynamic convolution for the state-of-the-art architecture MobileNetV3-Small, the top-1 accuracy of ImageNet classification is boosted by 2.9% with only 4% additional FLOPs and 2.9 AP gain is achieved on COCO keypoint detection.

Tasks

Benchmark Results

Reproductions