在单 GPU 上 3.29 秒内达到 CIFAR-10 94% 准确率
机器学习
2024-04-08 v2 计算机视觉与模式识别
摘要
CIFAR-10 是机器学习中最广泛使用的数据集之一，每年促进数千个研究项目。为了加速研究并降低实验成本，我们引入了针对 CIFAR-10 的训练方法，在单块 NVIDIA A100 GPU 上运行时，可在 3.29 秒内达到 94% 的准确率，10.4 秒内达到 95%，46.3 秒内达到 96%。作为促成这些训练速度的因素之一，我们提出了一种去随机化的水平翻转增强变体，我们表明在所有翻转优于不翻转的情况下，该方法均优于标准方法。我们的代码已在 https://github.com/KellerJordan/cifar10-airbench 发布
引用
@article{arxiv.2404.00498,
title = {94% on CIFAR-10 in 3.29 Seconds on a Single GPU},
author = {Keller Jordan},
journal= {arXiv preprint arXiv:2404.00498},
year = {2024}
}