研究人员开发了一种新颖的深度神经网络训练算法,该算法完全使用整数运算,消除了训练过程中任何阶段对浮点运算的需求。该方法将所有张量(包括权重、梯度和优化器状态)存储为INT8,与标准的FP32基线相比,训练内存大约减小了4倍。该算法已成功在各种卷积神经网络和视觉Transformer上得到验证,在CIFAR-10、CIFAR-100和ImageNet等基准测试中取得了具有竞争力的准确率。 AI
影响 降低了训练深度学习模型的内存需求,有可能在不太强大的硬件上进行训练。
排序理由 详细介绍深度神经网络新颖训练算法的学术论文。[lever_c_demoted from research: ic=1 ai=1.0]
- CIFAR-10
- CIFAR-100
- Hazem Hesham Yousef Shalby
- ImageNet
- Int8
- MobileNetV2
- ResNet-101
- ResNet-18
- ResNet-34
- ResNet-50
- single-precision floating-point format
- Vision Transformer Base
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →