HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

ResNet strikes back: An improved training procedure in timm

Ross Wightman Hugo Touvron Hervé Jégou

ResNet strikes back: An improved training procedure in timm

Abstract

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.

Code Repositories

jayroxis/ordering-in-graph-generation
pytorch
Mentioned in GitHub
rwightman/pytorch-image-models
Official
pytorch
Mentioned in GitHub
samsungsailmontreal/ghn3
pytorch
Mentioned in GitHub
open-mmlab/mmdetection
pytorch
Mentioned in GitHub
open-mmlab/mmclassification
pytorch
Mentioned in GitHub
hunto/dist_kd
pytorch
Mentioned in GitHub
eshoyuan/taop2021-3rd-solution
pytorch
Mentioned in GitHub
sithu31296/semantic-segmentation
pytorch
Mentioned in GitHub
shinya7y/UniverseNet
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
classification-on-indlResNetV2_50
Average Recall: 88.08%
domain-generalization-on-vizwizResNet-50 (gn)
Accuracy - All Images: 48.9
Accuracy - Clean Images: 44.4
Accuracy - Corrupted Images: 39.1
fine-grained-image-classification-on-oxfordResNet50 (A1)
Accuracy: 97.9%
FLOPS: 4.1
PARAMS: 24M
fine-grained-image-classification-on-stanfordResNet50 (A1)
Accuracy: 92.7%
FLOPS: 4.1B
PARAMS: 24M
image-classification-on-cifar-10cvpr_class
Percentage correct: 85.28
image-classification-on-cifar-10ResNet50 (A1)
Percentage correct: 98.3
image-classification-on-cifar-100ResNet50 (A1)
PARAMS: 25M
Percentage correct: 86.9
image-classification-on-flowers-102ResNet50 (A1)
Accuracy: 97.9
FLOPS: 4.1
PARAMS: 25M
image-classification-on-imagenetDeiT-S (T2)
Number of params: 22M
Top 1 Accuracy: 80.4%
image-classification-on-imagenetResNet-152 (A2 + reg)
Number of params: 60.2M
Top 1 Accuracy: 82.4%
image-classification-on-imagenetResNet-152 (A2)
Number of params: 60.2M
Top 1 Accuracy: 81.8%
image-classification-on-imagenetResNet50 (A3)
Number of params: 25M
Top 1 Accuracy: 78.1%
image-classification-on-imagenetResNet50 (A1)
Number of params: 25M
Top 1 Accuracy: 80.4%
image-classification-on-imagenet-realResNet50 (A1)
Accuracy: 85.7%
Params: 25M
image-classification-on-imagenet-v2ResNet50 (A1)
Top 1 Accuracy: 68.7
image-classification-on-inaturalist-2019ResNet50 (A2)
Top-1 Accuracy: 75.0
medical-image-classification-on-nct-crc-heResNeXt-50-32x4d
Accuracy (%): 95.46
F1-Score: 97.46
Precision: 99.91
Specificity: 99.43

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp