Neural Network Training on In-Memory-Computing Hardware With Radix-4 Gradients

Christopher Grimm, Naveen Verma

Research output: Contribution to journalArticlepeer-review

Abstract

Deep learning training involves a large number of operations, which are dominated by high dimensionality Matrix-Vector Multiplies (MVMs). This has motivated hardware accelerators to enhance compute efficiency, but where data movement and accessing are proving to be key bottlenecks. In-Memory Computing (IMC) is an approach with the potential to overcome this, whereby computations are performed in-place within dense 2-D memory. However, IMC fundamentally trades efficiency and throughput gains for dynamic-range limitations, raising distinct challenges for training, where compute precision requirements are seen to be substantially higher than for inferencing. This paper explores training on IMC hardware by leveraging two recent developments: (1) a training algorithm enabling aggressive quantization through a radix-4 number representation; (2) IMC leveraging compute based on precision capacitors, whereby analog noise effects can be made well below quantization effects. Energy modeling calibrated to a measured silicon prototype implemented in 16 nm CMOS shows that energy savings of over 400<inline-formula> <tex-math notation="LaTeX">$\times$</tex-math> </inline-formula> can be achieved with full quantizer adaptability, where all training MVMs can be mapped to IMC, and 3<inline-formula> <tex-math notation="LaTeX">$\times$</tex-math> </inline-formula> can be achieved for two-level quantizer adaptability, where two of the three training MVMs can be mapped to IMC.

Original languageEnglish (US)
Pages (from-to)1-13
Number of pages13
JournalIEEE Transactions on Circuits and Systems I: Regular Papers
DOIs
StateAccepted/In press - 2022

All Science Journal Classification (ASJC) codes

  • Hardware and Architecture
  • Electrical and Electronic Engineering

Keywords

  • Charge-domain compute
  • deep learning
  • Deep learning
  • Dynamic range
  • Hardware
  • hardware accelerators
  • in-memory computing (IMC)
  • neural networks (NNs)
  • Quantization (signal)
  • Throughput
  • training
  • Training
  • Voltage

Fingerprint

Dive into the research topics of 'Neural Network Training on In-Memory-Computing Hardware With Radix-4 Gradients'. Together they form a unique fingerprint.

Cite this