English

Rethinking Momentum Knowledge Distillation in Online Continual Learning

Machine Learning 2024-06-06 v2 Artificial Intelligence

Abstract

Online Continual Learning (OCL) addresses the problem of training neural networks on a continuous data stream where multiple classification tasks emerge in sequence. In contrast to offline Continual Learning, data can be seen only once in OCL, which is a very severe constraint. In this context, replay-based strategies have achieved impressive results and most state-of-the-art approaches heavily depend on them. While Knowledge Distillation (KD) has been extensively used in offline Continual Learning, it remains under-exploited in OCL, despite its high potential. In this paper, we analyze the challenges in applying KD to OCL and give empirical justifications. We introduce a direct yet effective methodology for applying Momentum Knowledge Distillation (MKD) to many flagship OCL methods and demonstrate its capabilities to enhance existing approaches. In addition to improving existing state-of-the-art accuracy by more than 10%10\% points on ImageNet100, we shed light on MKD internal mechanics and impacts during training in OCL. We argue that similar to replay, MKD should be considered a central component of OCL. The code is available at \url{https://github.com/Nicolas1203/mkd_ocl}.

Keywords

Cite

@article{arxiv.2309.02870,
  title  = {Rethinking Momentum Knowledge Distillation in Online Continual Learning},
  author = {Nicolas Michel and Maorong Wang and Ling Xiao and Toshihiko Yamasaki},
  journal= {arXiv preprint arXiv:2309.02870},
  year   = {2024}
}

Comments

Accepted to ICML 2024

R2 v1 2026-06-28T12:14:05.139Z