Online Convolutional Re-parameterization
Not on the map. Why?
Abstract (arXiv)
Structural re-parameterization has drawn increasing attention in various computer vision tasks. It aims at improving the performance of deep models without introducing any inference-time cost. Though efficient during inference, such models rely heavily on the complicated training-time blocks to achieve high accuracy, leading to large extra training cost. In this paper, we present online convolutional re-parameterization (OREPA), a two-stage pipeline, aiming to reduce the huge training overhead by squeezing the complex training-time block into a single convolution. To achieve this goal, we introduce a linear scaling layer for better optimizing the online blocks. Assisted with the reduced training cost, we also explore some more effective re-param components. Compared with the state-of-the-art re-param models, OREPA is able to save the training-time memory cost by about 70% and accelerate the training speed by around 2x. Meanwhile, equipped with OREPA, the models outperform previous methods on ImageNet by up to +0.6%.We also conduct experiments on object detection and semantic segmentation and show consistent improvements on the downstream tasks. Codes are available at https://github.com/JUGGHM/OREPA_CVPR2022 .
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 36 of its 51 references that are in the map. First posted on arXiv 2 Apr 2022.
Builds on
36 of 51 references.
References · 36
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,617
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
- 2015You Only Look Once: Unified, Real-Time Object Detectioncited by 4,829
- 2017Squeeze-and-Excitation Networkscited by 4,763
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,427
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,827
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
24 more
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,239
- 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,210
- 2016Pyramid Scene Parsing Networkcited by 2,472
- 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,320
- 2018Group Normalizationcited by 1,514
- 2019MMDetection: Open MMLab Detection Toolbox and Benchmarkcited by 1,379
- 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 541
- 2021RepVGG: Making VGG-style ConvNets Great Againcited by 520
- 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 430
- 2021High-Performance Large-Scale Image Recognition Without Normalizationcited by 359
- 2018How Does Batch Normalization Help Optimization?cited by 355
- 2018On the Optimization of Deep Networks: Implicit Acceleration by Overparameterizationcited by 280
- 2020Do Wide and Deep Networks Learn the Same Things? Uncovering How Neural Network Representations Vary with Width and Depthcited by 241
- 2019Fixup Initialization: Residual Learning Without Normalizationcited by 194
- 2019ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blockscited by 178
- 2020FcaNet: Frequency Channel Attention Networkscited by 177
- 2017The Shattered Gradients Problem: If resnets are the answer, then what is the question?cited by 156
- 2021Characterizing signal propagation to close the performance gap in unnormalized ResNetscited by 87
- 2021Diverse Branch Block: Building a Convolution as an Inception-like Unitcited by 87
- 2018ExpandNets: Linear Over-parameterization to Train Compact Convolutional Networkscited by 53
- 2020DO-Conv: Depthwise Over-parameterized Convolutional Layercited by 28
- 2021RepNAS: Searching for Efficient Re-parameterizing Blockscited by 11
- 2019Efficient Differentiable Neural Architecture Search with Meta Kernelscited by 6
- 2021RMNet: Equivalently Removing Residual Connection from Networkscited by 5
Built on by
Citing papers per year: 2022: 3; 2023: 2.
5 papers.
- 2022YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectorscited by 978
- 20223D UX-Net: A Large Kernel Volumetric ConvNet Modernizing Hierarchical Transformer for Medical Image Segmentationcited by 94
- 2023Scaling Up 3D Kernels with Bayesian Frequency Re-parameterization for Medical Image Segmentationcited by 12
- 2022RepMode: Learning to Re-parameterize Diverse Experts for Subcellular Structure Predictioncited by 3
- 2023Make Explicit Calibration Implicit: Calibrate Denoiser Instead of the Noise Modelcited by 2
- 2023Make Explicit Calibration Implicit: Calibrate Denoiser Instead of the Noise Model7 Aug 2023
- 2023Scaling Up 3D Kernels with Bayesian Frequency Re-parameterization for Medical Image Segmentation10 Mar 2023
- 2022RepMode: Learning to Re-parameterize Diverse Experts for Subcellular Structure Prediction20 Dec 2022
- 20223D UX-Net: A Large Kernel Volumetric ConvNet Modernizing Hierarchical Transformer for Medical Image Segmentation29 Sep 2022
- 2022YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors6 Jul 2022