ResNet strikes back: An improved training procedure in timm

The influential Residual Networks designed by He et al. remain the gold-standard architecture in numerous scientific publications. They typically serve as the default architecture in studies, or as baselines when new architectures are proposed. Yet there has been significant progress on best practices for training neural networks since the inception of the ResNet architecture in 2015. Novel optimization & data-augmentation have increased the effectiveness of the training recipes. In this paper, we re-evaluate the performance of the vanilla ResNet-50 when trained with a procedure that integrates such advances. We share competitive training settings and pre-trained models in the timm open-source library, with the hope that they will serve as better baselines for future work. For instance, with our more demanding training setting, a vanilla ResNet-50 reaches 80.4% top-1 accuracy at resolution 224x224 on ImageNet-val without extra data or distillation. We also report the performance achieved with popular models with our training procedure.

PDF Abstract NeurIPS Workshop 2021 PDF NeurIPS Workshop 2021 Abstract
Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Benchmark
Image Classification CIFAR-10 ResNet50 (A1) Percentage correct 98.3 # 41
PARAMS 25M # 210
Image Classification CIFAR-10 cvpr_class Percentage correct 85.28 # 202
Image Classification CIFAR-100 ResNet50 (A1) Percentage correct 86.9 # 49
PARAMS 25M # 192
Image Classification Flowers-102 ResNet50 (A1) Accuracy 97.9 # 29
FLOPS 4.1 # 1
PARAMS 25M # 48
Image Classification ImageNet DeiT-S (T2) Top 1 Accuracy 80.4% # 645
Number of params 22M # 557
Image Classification ImageNet ResNet50 (A3) Top 1 Accuracy 78.1% # 785
Number of params 25M # 587
Image Classification ImageNet ResNet-152 (A2) Top 1 Accuracy 81.8% # 553
Number of params 60.2M # 768
Image Classification ImageNet ResNet-152 (A2 + reg) Top 1 Accuracy 82.4% # 491
Number of params 60.2M # 768
Image Classification ImageNet ResNet50 (A1) Top 1 Accuracy 80.4% # 645
Number of params 25M # 587
Image Classification ImageNet ReaL ResNet50 (A1) Accuracy 85.7% # 39
Params 25M # 40
Image Classification ImageNet V2 ResNet50 (A1) Top 1 Accuracy 68.7 # 27
Image Classification iNaturalist 2019 ResNet50 (A2) Top-1 Accuracy 75.0 # 10
Classification InDL ResNetV2_50 Average Recall 88.08% # 8
Medical Image Classification NCT-CRC-HE-100K ResNeXt-50-32x4d Accuracy (%) 95.46 # 2
F1-Score 97.46 # 2
Precision 99.91 # 4
Specificity 99.43 # 2
Fine-Grained Image Classification Oxford 102 Flowers ResNet50 (A1) Accuracy 97.9% # 16
FLOPS 4.1 # 1
PARAMS 24M # 24
Fine-Grained Image Classification Stanford Cars ResNet50 (A1) Accuracy 92.7% # 63
FLOPS 4.1B # 2
PARAMS 24M # 72
Domain Generalization VizWiz-Classification ResNet-50 (gn) Accuracy - All Images 48.9 # 9
Accuracy - Corrupted Images 39.1 # 14
Accuracy - Clean Images 44.4 # 38

Methods