94% on CIFAR-10 in 3.29 Seconds on a Single GPU

Fuente: arXiv
Saved in:
Bibliographic Details
Main Author: Jordan, Keller
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913300699152384
author Jordan, Keller
author_facet Jordan, Keller
contents CIFAR-10 is among the most widely used datasets in machine learning, facilitating thousands of research projects per year. To accelerate research and reduce the cost of experiments, we introduce training methods for CIFAR-10 which reach 94% accuracy in 3.29 seconds, 95% in 10.4 seconds, and 96% in 46.3 seconds, when run on a single NVIDIA A100 GPU. As one factor contributing to these training speeds, we propose a derandomized variant of horizontal flipping augmentation, which we show improves over the standard method in every case where flipping is beneficial over no flipping at all. Our code is released at https://github.com/KellerJordan/cifar10-airbench.
format Preprint
id arxiv_https___arxiv_org_abs_2404_00498
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle 94% on CIFAR-10 in 3.29 Seconds on a Single GPU
Jordan, Keller
Machine Learning
Computer Vision and Pattern Recognition
CIFAR-10 is among the most widely used datasets in machine learning, facilitating thousands of research projects per year. To accelerate research and reduce the cost of experiments, we introduce training methods for CIFAR-10 which reach 94% accuracy in 3.29 seconds, 95% in 10.4 seconds, and 96% in 46.3 seconds, when run on a single NVIDIA A100 GPU. As one factor contributing to these training speeds, we propose a derandomized variant of horizontal flipping augmentation, which we show improves over the standard method in every case where flipping is beneficial over no flipping at all. Our code is released at https://github.com/KellerJordan/cifar10-airbench.
title 94% on CIFAR-10 in 3.29 Seconds on a Single GPU
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2404.00498