跳到主要导航 跳到搜索 跳到主要内容

面向高能效加速器的二值化神经网络设计和训练方法

  • Jiajun Li
  • , Haobo Xu*
  • , Yujie Wang
  • , Hang Xiao
  • , Ying Wang
  • , Yinhe Han
  • , Xiaowei Li
  • *此作品的通讯作者
  • CAS - Institute of Computing Technology
  • University of Chinese Academy of Sciences

科研成果: 期刊稿件文章同行评审

摘要

Aiming at the problem of computation overflow and multiplier dependence on the binarized neural network accelerator, a set of design and training methods of binarized neural networks (BNN) are proposed. Firstly, an accurate simulator is designed to ensure that BNN does not lose accuracy after deployment. Secondly, the convolutional layer and activation functions of the BNN are optimized to alleviate the total amount of overflow. Thirdly, an operator named Shift-based Batch Normalization is proposed to make the BNN get rid of the dependence on multiplication and reduce memory access. Finally, for the improved BNN, a collaborative training framework based on overflow heuristics is proposed to ensure that the model training converges. The experimental results show that, compared with 10 keyword spotting methods, the accelerator reduces the amount of on-chip computation by more than 49.1% and increases the speed at least 21.0% without significant loss of accuracy.

投稿的翻译标题Design and Training of Binarized Neural Networks for Highly Efficient Accelerators
源语言繁体中文
页(从-至)961-969
页数9
期刊Jisuanji Fuzhu Sheji Yu Tuxingxue Xuebao/Journal of Computer-Aided Design and Computer Graphics
35
6
DOI
出版状态已出版 - 6月 2023
已对外发布

关键词

  • binarized neural networks
  • deep learning
  • model training
  • neural network accelerators

学术指纹

探究 '面向高能效加速器的二值化神经网络设计和训练方法' 的科研主题。它们共同构成独一无二的学术指纹。

引用此