[15a-B414-1~9] FS.1 Focused Session "AI Electronics"

△ [15a-B414-6] Device Error Compensation for Computation-in-Memory 1
Compensation of Write Variation of Non-volatile Memory by Neural Network Algorithm Aware Layer-wise Retraining in Computation-in-Memory

〇Shinsei Yoshikiyo¹, Naoko Misawa¹, Chihiro Matsui¹, Ken Takeuchi¹ (1.Univ. Tokyo)

Keywords:Computation-in-Memory, error compensation, edge retraining

本研究では、Computation-in-Memoryにおいて、不揮発性メモリの書き込みばらつきにより低下した推論精度を回復するニューラルネットワーク（NN）のエッジ再学習手法を提案する。提案する再学習はNNモデルのshortcut connectionに起因する層間のエラー耐性の違いを利用し、エラー耐性が高い層のみを更新することで、再書き込み時の書き込みばらつきによる精度低下を抑制する。提案手法はResnet-32でのCIFAR-10の分類において再書き込み後の推論精度を従来手法よりも23.4%改善する。