DLCFT: Deep Linear Continual Fine-Tuning for General Incremental Learning

By Hyounguk Shon, Janghyeon Lee, Seungwook Kim, Junmo Kim. 2022

20 citations recorded by the source corpus.

Pre-trained representation is one of the key elements in the success of modern deep learning. However, existing works on continual learning methods have mostly focused on learning models incrementally from scratch. In this paper, we explore an alternative framework to incremental learning where we continually fine-tune the model from a pre-trained representation. Our method takes advantage of linearization technique of a pre-trained neural network for simple and effective continual learning. We show that this allows us to design a linear model where quadratic parameter regularization method is placed as the optimal continual learning policy, and at the same time enjoying the high performance of neural networks. We also show that the proposed algorithm enables parameter regularization methods to be applied to class-incremental problems. Additionally, we provide a theoretical reason why the existing parameter-space regularization algorithms such as EWC underperform on neural networks trained with cross-entropy loss. We show that the proposed method can prevent forgetting while achieving high continual fine-tuning performance on image classification tasks. To show that our method can be applied to general continual learning settings, we evaluate our method in data-incremental, task-incremental, and class-incremental learning problems.

Subfields: Artificial Intelligence, Optimization & Training, Meta & Continual Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Hang Su
  2. Jun Zhu
  3. Liyuan Wang
  4. Xingxing Zhang
  5. Tianyu Pang
  6. Yinpeng Dong
  7. Vamsi Krishna Ithapu
  8. Mirella Lapata

In Optimization & Training

  1. Liyuan Wang
  2. Jun Zhu
  3. Hang Su
  4. Xingxing Zhang

In Meta & Continual Learning

  1. Xingxing Zhang
  2. Liyuan Wang
  3. Hang Su
  4. Jun Zhu
  5. Simone Scardapane
  6. Valerio Marsocci
  7. Liyuan Wang
  8. Hongwei Yan

Citations in the corpus