Fast and robust analog in-memory deep neural network training

Analog in-memory computing is a promising future technology for efficiently accelerating deep learning networks. While using in-memory computing to accelerate the inference phase has been studied extensively, accelerating the training phase has received less attention, despite its arguably much larger compute demand to accelerate. While some analog in-memory training algorithms have been suggested, they either invoke significant amount of auxiliary digital compute-accumulating the gradient in digital floating point precision, limiting the potential speed-up-or suffer from the need for near perfectly programming reference conductance values to establish an algorithmic zero point. Here, we propose two improved algorithms for in-memory training, that retain the same fast runtime complexity while resolving the requirement of a precise zero point. We further investigate the limits of the algorithms in terms of conductance noise, symmetry, retention, and endurance which narrow down possible device material choices adequate for fast and robust in-memory deep neural network training.

Authors

Institutions

Publication Details

Journal
Nature Communications
Published
2024-08-20
DOI
https://doi.org/10.1038/s41467-024-51221-z
Citations
20
Primary Topic
Advanced Memory and Neural Computing
Type
article
Field-Weighted Citation Impact
4.17
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Fast and robust analog in-memory deep neural network training

Malte J. Rasch, Omobayode I. Fagbohungbe, Fabio Carta, Tayfun Gokmen
20 citations
Nature Communications
Advanced Memory and Neural Computing
4.17
article

Fast and robust analog in-memory deep neural network training

Malte J. Rasch, Omobayode I. Fagbohungbe, Fabio Carta, Tayfun Gokmen
article en
20 citations

Abstract

Analog in-memory computing is a promising future technology for efficiently accelerating deep learning networks. While using in-memory computing to accelerate the inference phase has been studied extensively, accelerating the training phase has received less attention, despite its arguably much larger compute demand to accelerate. While some analog in-memory training algorithms have been suggested, they either invoke significant amount of auxiliary digital compute-accumulating the gradient in digital floating point precision, limiting the potential speed-up-or suffer from the need for near perfectly programming reference conductance values to establish an algorithmic zero point. Here, we propose two improved algorithms for in-memory training, that retain the same fast runtime complexity while resolving the requirement of a precise zero point. We further investigate the limits of the algorithms in terms of conductance noise, symmetry, retention, and endurance which narrow down possible device material choices adequate for fast and robust in-memory deep neural network training.

Nature CommunicationsVol. 15(1)
IBM (United States) (US), HES-SO University of Applied Sciences and Arts Western Switzerland (CH), IBM Research - Thomas J. Watson Research Center (US)
Openalex Percentile: Top 5%
Advanced Memory and Neural Computing
4.17
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.