RepVGG:让 VGG 风格的卷积网络再次伟大

RepVGG: Making VGG-style ConvNets Great Again

张祥雨 Xiangyu Zhang · MEGVII (Face++) / Tsinghua University · 2021-01-11 · arXiv:2101.03697 ↗ · 被引 2366

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了一种简单而强大的卷积神经网络架构,其推理时主体类似于 VGG,仅由 3x3 卷积和 ReLU 堆叠而成,而训练时模型则具有多分支拓扑结构。这种训练时与推理时架构的解耦通过结构重参数化技术实现,因此该模型被命名为 RepVGG。在 ImageNet 上,RepVGG 达到了超过 80%的 top-1 准确率,据我们所知,这是普通模型首次达到这一水平。在 NVIDIA 1080Ti GPU 上,RepVGG 模型比 ResNet-50 快 83%,比 ResNet-101 快 101%,且准确率更高,与 EfficientNet 和 RegNet 等最先进模型相比,显示出良好的准确率-速度权衡。代码和训练好的模型可在 https://github.com/megvii-model/RepVGG 获取。

We present a simple but powerful architecture of convolutional neural network, which has a VGG-like inference-time body composed of nothing but a stack of 3x3 convolution and ReLU, while the training-time model has a multi-branch topology. Such decoupling of the training-time and inference-time architecture is realized by a structural re-parameterization technique so that the model is named RepVGG. On ImageNet, RepVGG reaches over 80% top-1 accuracy, which is the first time for a plain model, to the best of our knowledge. On NVIDIA 1080Ti GPU, RepVGG models run 83% faster than ResNet-50 or 101% faster than ResNet-101 with higher accuracy and show favorable accuracy-speed trade-off compared to the state-of-the-art models like EfficientNet and RegNet. The code and trained models are available at https://github.com/megvii-model/RepVGG.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 16)

阅读逐段中英对照全文 →