| 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 Publication | NEUROCOMPUTING
 |
| ISSN | 0925-2312 |
| 2022-10-28 | |
| Volume | 511Pages:175-186 |
| Corresponding Author | Li, Guoqi(liguoqi@mail.tsinghua.edu.cn) |
| Abstract | Huge 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. |
| Keyword | Full 8-bit quantization Network without batch normalization Small batch Online training Resource-limited devices |
| DOI | 10.1016/j.neucom.2022.08.045 |
| WOS Keyword | DEEP NEURAL-NETWORKS ; MEMORY |
| Indexed By | SCI |
| Language | 英语 |
| Funding Project | National 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 Organization | National Key RD program ; Beijing Natural Science Fundation ; Beijing Academy of Artificial Intelligence (BAAI) ; Science and Technology Major Project of Guangzhou ; Pengcheng Lab |
| WOS Research Area | Computer Science |
| WOS Subject | Computer Science, Artificial Intelligence |
| WOS ID | WOS:000871948700015 |
| Publisher | ELSEVIER |
| Citation statistics | |
| Document Type | 期刊论文 |
| Identifier | http://ir.ia.ac.cn/handle/173211/50506 |
| Collection | 数字内容技术与服务研究中心_听觉模型与认知计算 |
| Corresponding Author | Li, Guoqi |
| Affiliation | 1.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 Affilication | Institute 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. | |||||
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment