Involution: Inverting the Inherence of Convolution for Visual Recognition

By Duo Li, Jie Hu, Changhu Wang, Xiangtai Li, Qi She, Lei Zhu, T. Zhang, Qifeng Chen. 2021

392 citations recorded by the source corpus.

Convolution has been the core ingredient of modern neural networks, triggering the surge of deep learning in vision. In this work, we rethink the inherent principles of standard convolution for vision tasks, specifically spatial-agnostic and channel-specific. Instead, we present a novel atomic operation for deep neural networks by inverting the aforementioned design principles of convolution, coined as involution. We additionally demystify the recent popular self-attention operator and subsume it into our involution family as an over-complicated instantiation. The proposed involution operator could be leveraged as fundamental bricks to build the new generation of neural networks for visual recognition, powering different deep learning models on several prevalent benchmarks, including ImageNet classification, COCO detection and segmentation, together with Cityscapes segmentation. Our involution-based models improve the performance of convolutional baselines using ResNet-50 by up to 1.6% top-1 accuracy, 2.5% and 2.4% bounding box AP, and 4.7% mean IoU absolutely while compressing the computational cost to 66%, 65%, 72%, and 57% on the above benchmarks, respectively. Code and pre-trained models for all the tasks are available at https://github.com/d-li14/involution.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Model Architectures

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Neil Houlsby
  2. Jianfeng Gao
  3. Xiaohua Zhai
  4. Quoc V. Le
  5. Lucas Beyer
  6. Alexander Kolesnikov
  7. Mohit Bansal
  8. Jianwei Yang

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Trevor Darrell
  4. Jitendra Malik
  5. Xiaogang Wang
  6. Ming-Hsuan Yang
  7. Andrew Zisserman
  8. Cordelia Schmid

In Model Architectures

  1. Jianfeng Gao
  2. Jianwei Yang
  3. Xiaohua Zhai
  4. Neil Houlsby
  5. Alexander Kolesnikov
  6. Lucas Beyer
  7. Thomas Unterthiner
  8. Alexey Dosovitskiy

Citations in the corpus