Towards a General Framework for Continual Learning with Pre-training

Published: 20 Oct 2023, Last Modified: 30 Nov 2023IMOL@NeurIPS2023EveryoneRevisionsBibTeX
Keywords: Continual Learning, Catastrophic Forgetting, Pre-training
TL;DR: We present a general framework for continual learning with pre-training, including theoretical, methodological and interdisciplinary insights.
Abstract: In this work, we present a general framework for continual learning of sequentially arrived tasks with the use of pre-training, which has emerged as a promising direction for artificial intelligence systems to accommodate real-world dynamics. From a theoretical perspective, we decompose its objective into three hierarchical components, including within-task prediction, task-identity inference, and task-adaptive prediction. Then we propose an innovative approach to explicitly optimize these components with parameter-efficient fine-tuning (PEFT) techniques and representation statistics. We empirically demonstrate the superiority and generality of our approach in downstream continual learning, and further explore the applicability of PEFT techniques in upstream continual learning. We expect this to provide an important technical foundation for intrinsically motivated open-ended learning.
Submission Number: 4