CASIA OpenIR  > 复杂系统认知与决策实验室  > 听觉模型与认知计算
Towards efficient full 8-bit integer DNN online training on resource-limited devices without batch normalization
Yang, Yukuan1; Chi, Xiaowei2; Deng, Lei1; Yan, Tianyi3; Gao, Feng4; Li, Guoqi1,5
Source PublicationNEUROCOMPUTING
ISSN0925-2312
2022-10-28
Volume511Pages:175-186
Corresponding AuthorLi, Guoqi(liguoqi@mail.tsinghua.edu.cn)
AbstractHuge computational costs brought by convolution and batch normalization (BN) have caused great challenges for the online training and corresponding applications of deep neural networks (DNNs), especially in resource-limited devices. Existing works only focus on the convolution or BN acceleration and no solution can alleviate both problems with satisfactory performance. Online training has gradually become a trend in resource-limited devices like mobile phones while there is still no complete technical scheme with acceptable model performance, processing speed, and computational cost. In this research, an efficient online-training quantization framework termed EOQ for abbreviation is proposed by combining Fixup initialization and a novel quantization scheme for the online training in resource-limited devices. Based on the proposed framework, we have successfully realized full 8-bit integer network training and removed BN in large-scale DNNs. Especially, weight updates are quantized to 8-bit integers for the first time. Theoretical analyses of EOQ utilizing Fixup initialization for removing BN have been further given using a novel Block Dynamical Isometry theory with weaker assumptions. Benefiting from rational quantization strategies and the absence of BN, the full 8-bit networks based on EOQ can achieve state-of-the-art accuracy and immense advantages in computational cost and processing speed. Experiments show that the 8-bit EOQ networks achieve 2.78%, 3.85%, and 4.31% accuracy improvements compared with existing full 8-bit integer networks in ResNet-18/34/50. At the same time, the 8-bit EOQ networks can improve the computing speed greatly, and decrease the power consumption and circuit area by about an order of magnitude compared with 32-bit floating-point vanilla networks. In addition to the huge advantages brought by quantization in convolution operations, 8-bit networks based on EOQ without BN can realize >66x lower in power, >13 x faster in the processing speed compared with the traditional 32-bit floating-point BN in the inference process. What's more, the design of deep learning chips can be profoundly simplified in the absence of unfriendly square root operations in BN. Beyond this, EOQ has been evidenced to be more advantageous in small-batch online training with fewer batch samples. In summary, the EOQ framework is specially designed for reducing the high cost of convolution and BN in network training, demonstrating a broad application prospect of online training in resource-limited devices. (C) 2022 Published by Elsevier B.V.
KeywordFull 8-bit quantization Network without batch normalization Small batch Online training Resource-limited devices
DOI10.1016/j.neucom.2022.08.045
WOS KeywordDEEP NEURAL-NETWORKS ; MEMORY
Indexed BySCI
Language英语
Funding ProjectNational Key RD program[2018YFE0200200] ; National Key RD program[2018AAA0102604] ; Beijing Natural Science Fundation[JQ21015] ; Beijing Academy of Artificial Intelligence (BAAI) ; Science and Technology Major Project of Guangzhou[202007030006] ; Pengcheng Lab
Funding OrganizationNational Key RD program ; Beijing Natural Science Fundation ; Beijing Academy of Artificial Intelligence (BAAI) ; Science and Technology Major Project of Guangzhou ; Pengcheng Lab
WOS Research AreaComputer Science
WOS SubjectComputer Science, Artificial Intelligence
WOS IDWOS:000871948700015
PublisherELSEVIER
Citation statistics
Document Type期刊论文
Identifierhttp://ir.ia.ac.cn/handle/173211/50506
Collection复杂系统认知与决策实验室_听觉模型与认知计算
Corresponding AuthorLi, Guoqi
Affiliation1.Tsinghua Univ, Ctr Brain Inspired Comp Res, Dept Precis Instrument, Beijing 100084, Peoples R China
2.Beijing Univ Posts & Telecommun, Int Sch, Beijing 100876, Peoples R China
3.Beijing Inst Technol, Sch Life Sci, Beijing 100081, Peoples R China
4.Capital Med Univ, Beijing Tiantan Hosp, Dept Intervent Neuroradiol, Beijing 100070, Peoples R China
5.Chinese Acad Sci, Inst Automat, Beijing 100190, Peoples R China
Corresponding Author AffilicationInstitute of Automation, Chinese Academy of Sciences
Recommended Citation
GB/T 7714
Yang, Yukuan,Chi, Xiaowei,Deng, Lei,et al. Towards efficient full 8-bit integer DNN online training on resource-limited devices without batch normalization[J]. NEUROCOMPUTING,2022,511:175-186.
APA Yang, Yukuan,Chi, Xiaowei,Deng, Lei,Yan, Tianyi,Gao, Feng,&Li, Guoqi.(2022).Towards efficient full 8-bit integer DNN online training on resource-limited devices without batch normalization.NEUROCOMPUTING,511,175-186.
MLA Yang, Yukuan,et al."Towards efficient full 8-bit integer DNN online training on resource-limited devices without batch normalization".NEUROCOMPUTING 511(2022):175-186.
Files in This Item:
There are no files associated with this item.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Yang, Yukuan]'s Articles
[Chi, Xiaowei]'s Articles
[Deng, Lei]'s Articles
Baidu academic
Similar articles in Baidu academic
[Yang, Yukuan]'s Articles
[Chi, Xiaowei]'s Articles
[Deng, Lei]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Yang, Yukuan]'s Articles
[Chi, Xiaowei]'s Articles
[Deng, Lei]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.