Navigating Memory Construction by Global Pseudo-Task Simulation for Continual Learning

Yejia Liu, Wang Zhu, Shaolei Ren

Keywords: continual learning, experience replay, pseudo-task simulation, global optimization, dynamic memory construction

TL;DR: We have proposed a novel method Global Pseudo-task Simulation (GPS) to solve the dynamic memory construction problem in the online continual learning setting.

Abstract: Continual learning faces a crucial challenge of catastrophic forgetting. To address this challenge, experience replay (ER) that maintains a tiny subset of samples from previous tasks has been commonly used. Existing ER works usually focus on refining the learning objective for each task with a static memory construction policy. In this paper, we formulate the dynamic memory construction in ER as a combinatorial optimization problem, which aims at directly minimizing the global loss across all experienced tasks. We first apply three tactics to solve the problem in the offline setting as a starting point. To provide an approximate solution to this problem under the online continual learning setting, we further propose the Global Pseudo-task Simulation (GPS), which mimics future catastrophic forgetting of the current task by permutation. Our empirical results and analyses suggest that the GPS consistently improves accuracy across four commonly used vision benchmarks. We have also shown that our GPS can serve as the unified framework for integrating various memory construction policies in existing ER works.

Community Implementations: [ 1 code implementation](https://www.catalyzex.com/paper/navigating-memory-construction-by-global/code)