Few-Shot Lifelong Learning

By Pratik Mazumder, Pravendra Singh, Piyush Rai. 2021

159 citations recorded by the source corpus.

Many real-world classification problems often have classes with very few labeled training samples. Moreover, all possible classes may not be initially available for training, and may be given incrementally. Deep learning models need to deal with this two-fold problem in order to perform well in real-life situations. In this paper, we propose a novel Few-Shot Lifelong Learning (FSLL) method that enables deep learning models to perform lifelong/continual learning on few-shot data. Our method selects very few parameters from the model for training every new set of classes instead of training the full model. This helps in preventing overfitting. We choose the few parameters from the model in such a way that only the currently unimportant parameters get selected. By keeping the important parameters in the model intact, our approach minimizes catastrophic forgetting. Furthermore, we minimize the cosine similarity between the new and the old class prototypes in order to maximize their separation, thereby improving the classification performance. We also show that integrating our method with self-supervision improves the model performance significantly. We experimentally show that our method significantly outperforms existing methods on the miniImageNet, CIFAR-100, and CUB-200 datasets. Specifically, we outperform the state-of-the-art method by an absolute margin of 19.27% for the CUB dataset.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Radiology, Nuclear Medicine and Imaging, 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. De-chuan Zhan
  2. Han-Jia Ye
  3. Da-Wei Zhou
  4. Hang Su
  5. Jun Zhu
  6. Liyuan Wang
  7. Xingxing Zhang
  8. Xinchao Wang

In Computer Vision and Pattern Recognition

  1. Han-Jia Ye
  2. De-chuan Zhan
  3. Dacheng Tao
  4. Da-Wei Zhou
  5. Jun Zhu
  6. Hang Su
  7. Xinchao Wang
  8. Liyuan Wang

In Radiology, Nuclear Medicine and Imaging

  1. Da-Wei Zhou
  2. De-chuan Zhan
  3. Han-Jia Ye
  4. Bahram Zonooz
  5. Elahe Arani
  6. Xinchao Wang
  7. Mingli Song
  8. Sung Ju Hwang

In Meta & Continual Learning

  1. Xingxing Zhang
  2. Liyuan Wang
  3. Hang Su
  4. Jun Zhu
  5. Han-Jia Ye
  6. De-chuan Zhan
  7. Da-Wei Zhou
  8. Gyeong-Moon Park

Citations in the corpus