Chameleon: Dual Memory Replay for Online Continual Learning on Edge Devices
Shivam Aggarwal, Kuluhan Binici, Tulika Mitra · IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems · 2023
Once deployed on edge devices, a deep neural network model should dynamically adapt to newly discovered environments and personalize its utility for each user. The system must be capable of continual learning, i.e., learning new information from a temporal stream of data in situ without forgetting previously acquired knowledge. However, creating a personalized continual learning framework poses significant challenges due to limited compute and storage resources on edge devices. Existing methods rely on large memory storage to preserve past data while learning from incoming streams, making them impractical for such devices. In this paper, we propose Chameleon as a hardware-friendly continual learning solution for user-centric continual learning with dual replay buffers. The strategy takes advantage of the hierarchical memory structure commonly found in edge devices, utilizing a short-term replay store in on-chip memory and a long-term replay store in off-chip memory. We also present an FPGA-based analytical model to estimate the compute and communication costs of the dual replay strategy on the hardware, making effective design choices considering various latent layer options. We conduct extensive experiments on four different models, demonstrating our method’s consistent performance across diverse model architectures. Our method achieves up to 7× speedup and improved energy efficiency on popular edge devices, including ZCU102 FPGA, NVIDIA Jetson Nano, and Google’s EdgeTPU. Our code is available at https://github.com/ecolab-nus/Chameleon.