ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
2.5k
Citations
162
Influential Citations
IEEE Transactions on Pattern Analysis and Machine Intelligence
Venue
2021
Year
Artificial neural networks thrive in solving the classification problem for a particular rigid task, acquiring knowledge through generalized learning behaviour from a distinct training phase. The resulting network resembles a static entity of knowledge, with endeavours to extend this knowledge without targeting the original task resulting in a catastrophic forgetting. Continual learning shifts this paradigm towards networks that can continually accumulate knowledge over different tasks without the need to retrain from scratch. We focus on task incremental classification, where tasks arrive sequentially and are delineated by clear boundaries. Our main contributions concern: (1) a taxonomy and extensive overview of the state-of-the-art; (2) a novel framework to continually determine the stability-plasticity trade-off of the continual learner; (3) a comprehensive experimental comparison of 11 state-of-the-art continual learning methods; and (4) baselines. We empirically scrutinize method strengths and weaknesses on three benchmarks, considering Tiny Imagenet and large-scale unbalanced iNaturalist and a sequence of recognition datasets. We study the influence of model capacity, weight decay and dropout regularization, and the order in which the tasks are presented, and qualitatively compare methods in terms of required memory, computation time, and storage.
Continual learning is a critical step towards artificial general intelligence, enabling models to learn sequentially from a stream of tasks without forgetting previous knowledge. This paper, published in IEEE TPAMI and cited over 2500 times, offers a timely and comprehensive survey of the field, addressing the challenge of catastrophic forgetting in classification tasks. By systematically categorizing methods and providing a rigorous experimental comparison, it serves as a cornerstone for researchers and practitioners aiming to build systems that can adapt to new information while retaining old skills.
The significance of this work lies in its dual role: it not only consolidates existing knowledge but also introduces a novel framework to evaluate the stability-plasticity trade-off, a fundamental dilemma in continual learning. This framework allows for a principled assessment of how well a model balances retaining old knowledge (stability) and acquiring new knowledge (plasticity). The extensive experiments on diverse benchmarks, including large-scale and unbalanced datasets, provide actionable insights into the performance of current methods under realistic conditions.
The paper reports extensive experimental results, showing that no single method dominates across all scenarios. For instance, on Tiny Imagenet, replay-based methods like Experience Replay and Gradient Episodic Memory achieve high accuracy but require significant memory, while regularization methods like Elastic Weight Consolidation are more memory-efficient but less accurate. The stability-plasticity framework reveals that methods often favor one aspect over the other, and the optimal trade-off depends on task similarity and resource constraints. The study also finds that increasing model capacity can improve performance but may exacerbate forgetting, and that weight decay and dropout can mitigate forgetting to some extent. Task order significantly affects performance, with some methods being more robust to order than others.
This paper has had a profound impact on the continual learning community by providing a standardized benchmark and evaluation framework. It has guided the development of new methods that better balance stability and plasticity, and has highlighted the importance of considering practical constraints like memory and computation. The taxonomy and experimental insights have become reference points for subsequent research, fostering progress towards more robust and efficient continual learning systems. As AI models are increasingly deployed in dynamic environments, the principles and findings from this survey are essential for building lifelong learning agents.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba