[20a-M206-1~10] FS.1 Focused Session "AI Electronics"

[20a-M206-7] Class-incremental Learning with Knowledge Distillation for Edge Device Computation-in-Memory (CiM)

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

Keywords:Computation-in-Memory, class-incremental learning, knowledge distillation

本研究では、エッジデバイス上でクラス増分学習を行うことができるComputation-in-Memory (CiM)を提案する。提案手法は最終層のみを更新するため、逆伝播演算の必要がなく、必要なメモリが少ない。加えて、知識の蒸留を用いることで、既存クラスの学習データなしでも再学習を行うことができる。CIFAR-10を用いた検証タスクで、提案手法は全層を再学習する既存手法と同程度の精度を達成する。提案するCiMのデバイスエラーに対する耐性の調査も行う。