arrow
返回

SMIT: Static mixed integer training quantization

delete2026-09-18
delete0
PRE
AI
A
Anoosh Mahjabin
H
Hakem Beitollahi *
A
Abdollah Amirkhani
P
Pejman Lotfi-Kamran
DOI:10.1007/s11227-026-08857-zdelete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
深度神经网络(DNNs)在视觉任务中取得了显著成功,但其在训练过程中的大规模部署引入了巨大的计算和能耗需求。低精度量化为提高效率提供了有前景的途径;然而,现有的亚8位训练技术常遇到不稳定性和显著的精度下降,限制了其实用性。本研究提出了一种静态混合整数训练(SMIT)方法,这是一种用于端到端DNN训练的经验框架,采用固定的、异构位宽对权重、激活值和梯度进行量化。SMIT的动机源于经验观察,即不同的训练组件对量化噪声的敏感性不同。通过系统性的设计空间探索,我们确定了一种经验上有效的配置W4/A6/E8(4位权重、6位激活值和8位梯度),该配置在数值稳定性和降低精度之间提供了有利的权衡。在包括ImageNet分类、目标检测(Cityscapes)和语义分割(Pascal VOC)在内的多个视觉任务上的实验结果表明,SMIT始终能在FP32性能的1.0%~2%范围内实现精度。这些结果证明了静态异构亚8位训练的实用性,同时避免了运行时精度调度或特定架构的调整。除了数值稳定性之外,所提出的静态精度分配可能通过避免运行时精度自适应,简化未来低精度高性能计算的实现,并为未来的低精度加速器和分布式AI系统提供硬件友好的框架。
Keyword:
Quantization
Deep neural networks (DNN)
Mixed-Integer

期刊

Journal of Supercomputing 封面图
Journal of Supercomputing
IF:
2.7
论文数:
1.1K
被引数:
1.0W

机构

暂无机构信息
引用论文

引用论文

Distribution Adaptive INT8 Quantization for Training CNNs
err2021-05-18
err0
errOAAI
errKang Zhao; Sida Huang; Pan Pan; Yinghan Li; Yingya Zhang; Zhenyu Gu; Yinghui Xu
err分享
err收藏
ImageNet Large Scale Visual Recognition ChallengeImageNet大规模视觉识别挑战
err2015-04-11
err2.7W
PREAI
errRussakovsky, Olga; Deng, Jia; Su, Hao; Krause, Jonathan; Satheesh, Sanjeev; Ma, Sean; Huang, Zhiheng; Karpathy, Andrej; Khosla, Aditya; Bernstein, Michael; Berg, Alexander C.; Fei-Fei, Li
err分享
err收藏
The Pascal Visual Object Classes (VOC) ChallengePascal视觉对象课程 (VOC) 挑战
err2009-09-09
err9.0K
PREAI
errEveringham, Mark; Van Gool, Luc; Williams, Christopher K. I.; Winn, John; Zisserman, Andrew
err分享
err收藏
Is Integer Arithmetic Enough for Deep Learning Training?整数算术是否足以进行深度学习训练?
err2022-01-01
err0
PREAI
errGhaffari,Alireza; Tahaei,Marzieh S.; Tayaranian,Mohammadreza; Asgharian,Masoud; Nia,Vahid Partovi
err分享
err收藏
没有更多内容