1. ACAE-REMIND for online continual learning with compressed feature replay.
- Author
-
Wang, Kai, van de Weijer, Joost, and Herranz, Luis
- Subjects
- *
ONLINE education , *IMAGE representation , *VECTOR quantization - Abstract
• We study feature replay with compressed exemplars for online continual learning. • We show the importance of replaying features from intermediate layers of the network. • We propose ACAE-REMIND enabling better feature compression based on auto-encoders. • The method achieves good performance in several online continual learning settings. • Our online method even surpasses several offline continual learning methods. Online continual learning aims to learn from a non-IID stream of data from a number of different tasks, where the learner is only allowed to consider data once. Methods are typically allowed to use a limited buffer to store some of the images in the stream. Recently, it was found that feature replay, where an intermediate layer representation of the image is stored (or generated) leads to superior results than image replay, while requiring less memory. Quantized exemplars can further reduce the memory usage. However, a drawback of these methods is that they use a fixed (or very intransigent) backbone network. This significantly limits the learning of representations that can discriminate between all tasks. To address this problem, we propose an auxiliary classifier auto-encoder (ACAE) module for feature replay at intermediate layers with high compression rates. The reduced memory footprint per image allows us to save more exemplars for replay. In our experiments, we conduct task-agnostic evaluation under online continual learning setting and get state-of-the-art performance on ImageNet-Subset, CIFAR100 and CIFAR10 dataset. [ABSTRACT FROM AUTHOR]
- Published
- 2021
- Full Text
- View/download PDF