The precise programming of crossbar arrays of unit-cells is crucial for obtaining high matrix-vector-multiplication (MVM) accuracy in analog in-memory computing (AIMC) cores. We propose a radically different approach based on directly minimizing the MVM error using gradient descent with synthetic random input data. Our method significantly reduces the MVM error compared with conventional unit-cell by unit-cell iterative programming. It also eliminates the need for high-resolution analog-to-digital converters (ADCs) to read the small unit-cell conductance during programming. Our method improves the experimental inference accuracy of ResNet-9 implemented on two phase-change memory (PCM)-based AIMC cores by 1.26%.
@article{arxiv.2305.16647,
title = {Gradient descent-based programming of analog in-memory computing cores},
author = {Julian Büchel and Athanasios Vasilopoulos and Benedikt Kersting and Frederic Odermatt and Kevin Brew and Injo Ok and Sam Choi and Iqbal Saraf and Victor Chan and Timothy Philip and Nicole Saulnier and Vijay Narayanan and Manuel Le Gallo and Abu Sebastian},
journal= {arXiv preprint arXiv:2305.16647},
year = {2023}
}