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 AbstractCode
Results from the Paper
Task | Dataset | Model | Metric Name | Metric Value | Global Rank | Uses Extra Training Data |
Benchmark |
---|---|---|---|---|---|---|---|
Image Classification | CIFAR-10 | cvpr_class | Percentage correct | 85.28 | # 205 | ||
Image Classification | CIFAR-10 | ResNet50 (A1) | Percentage correct | 98.3 | # 43 | ||
PARAMS | 25M | # 208 | |||||
Image Classification | CIFAR-100 | ResNet50 (A1) | Percentage correct | 86.9 | # 49 | ||
PARAMS | 25M | # 189 | |||||
Image Classification | Flowers-102 | ResNet50 (A1) | Accuracy | 97.9 | # 30 | ||
FLOPS | 4.1 | # 1 | |||||
PARAMS | 25M | # 49 | |||||
Image Classification | ImageNet | ResNet50 (A3) | Top 1 Accuracy | 78.1% | # 753 | ||
Number of params | 25M | # 563 | |||||
Image Classification | ImageNet | ResNet-152 (A2) | Top 1 Accuracy | 81.8% | # 523 | ||
Number of params | 60.2M | # 741 | |||||
Image Classification | ImageNet | ResNet-152 (A2 + reg) | Top 1 Accuracy | 82.4% | # 465 | ||
Number of params | 60.2M | # 741 | |||||
Image Classification | ImageNet | DeiT-S (T2) | Top 1 Accuracy | 80.4% | # 614 | ||
Number of params | 22M | # 534 | |||||
Image Classification | ImageNet | ResNet50 (A1) | Top 1 Accuracy | 80.4% | # 614 | ||
Number of params | 25M | # 563 | |||||
Image Classification | ImageNet ReaL | ResNet50 (A1) | Accuracy | 85.7% | # 39 | ||
Params | 25M | # 38 | |||||
Image Classification | ImageNet V2 | ResNet50 (A1) | Top 1 Accuracy | 68.7 | # 25 | ||
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% | # 15 | ||
FLOPS | 4.1 | # 1 | |||||
PARAMS | 24M | # 22 | |||||
Fine-Grained Image Classification | Stanford Cars | ResNet50 (A1) | Accuracy | 92.7% | # 61 | ||
FLOPS | 4.1B | # 1 | |||||
PARAMS | 24M | # 70 | |||||
Domain Generalization | VizWiz-Classification | ResNet-50 (gn) | Accuracy - All Images | 48.9 | # 11 | ||
Accuracy - Corrupted Images | 39.1 | # 20 | |||||
Accuracy - Clean Images | 44.4 | # 44 |