Skip to main navigation Skip to search Skip to main content

You Already Have It: A Generator-Free Low-Precision DNN Training Framework Using Stochastic Rounding

  • Geng Yuan
  • , Sung En Chang
  • , Qing Jin
  • , Alec Lu
  • , Yanyu Li
  • , Yushu Wu
  • , Zhenglun Kong
  • , Yanyue Xie
  • , Peiyan Dong
  • , Minghai Qin
  • , Xiaolong Ma
  • , Xulong Tang
  • , Zhenman Fang
  • , Yanzhi Wang

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Stochastic rounding is a critical technique used in low-precision deep neural networks (DNNs) training to ensure good model accuracy. However, it requires a large number of random numbers generated on the fly. This is not a trivial task on the hardware platforms such as FPGA and ASIC. The widely used solution is to introduce random number generators with extra hardware costs. In this paper, we innovatively propose to employ the stochastic property of DNN training process itself and directly extract random numbers from DNNs in a self-sufficient manner. We propose different methods to obtain random numbers from different sources in neural networks and a generator-free framework is proposed for low-precision DNN training on a variety of deep learning tasks. Moreover, we evaluate the quality of the extracted random numbers and find that high-quality random numbers widely exist in DNNs, while their quality can even pass the NIST test suite.

Original languageEnglish (US)
Title of host publicationComputer Vision – ECCV 2022 - 17th European Conference, Proceedings
EditorsShai Avidan, Gabriel Brostow, Moustapha Cissé, Giovanni Maria Farinella, Tal Hassner
PublisherSpringer Science and Business Media Deutschland GmbH
Pages34-51
Number of pages18
ISBN (Print)9783031197741
DOIs
StatePublished - 2022
Externally publishedYes
Event17th European Conference on Computer Vision, ECCV 2022 - Tel Aviv, Israel
Duration: Oct 23 2022Oct 27 2022

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13672 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference17th European Conference on Computer Vision, ECCV 2022
Country/TerritoryIsrael
CityTel Aviv
Period10/23/2210/27/22

Bibliographical note

Publisher Copyright:
© 2022, The Author(s), under exclusive license to Springer Nature Switzerland AG.

Keywords

  • Efficient training
  • Quantization
  • Stochastic rounding

Fingerprint

Dive into the research topics of 'You Already Have It: A Generator-Free Low-Precision DNN Training Framework Using Stochastic Rounding'. Together they form a unique fingerprint.

Cite this