SSUL: Semantic Segmentation with Unknown Label for Exemplar-based Class-Incremental Learning

By Sungmin Cha, Beomyoung Kim, Y. Yoo, Taesup Moon. 2021

135 citations recorded by the source corpus.

This paper introduces a solid state-of-the-art baseline for a class-incremental semantic segmentation (CISS) problem. While the recent CISS algorithms utilize variants of the knowledge distillation (KD) technique to tackle the problem, they failed to fully address the critical challenges in CISS causing the catastrophic forgetting; the semantic drift of the background class and the multi-label prediction issue. To better address these challenges, we propose a new method, dubbed SSUL-M (Semantic Segmentation with Unknown Label with Memory), by carefully combining techniques tailored for semantic segmentation. Specifically, we claim three main contributions. (1) defining unknown classes within the background class to help to learn future classes (help plasticity), (2) freezing backbone network and past classifiers with binary cross-entropy loss and pseudo-labeling to overcome catastrophic forgetting (help stability), and (3) utilizing tiny exemplar memory for the first time in CISS to improve both plasticity and stability. The extensively conducted experiments show the effectiveness of our method, achieving significantly better performance than the recent state-of-the-art baselines on the standard benchmark datasets. Furthermore, we justify our contributions with thorough ablation analyses and discuss different natures of the CISS problem compared to the traditional class-incremental learning targeting classification. The official code is available at https://github.com/clovaai/SSUL.

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. Han-Jia Ye
  2. De-chuan Zhan
  3. Da-Wei Zhou
  4. Hang Su
  5. Qiwen Wang
  6. Jun Zhu
  7. Ziwei Liu
  8. Liyuan Wang

In Computer Vision and Pattern Recognition

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

In Radiology, Nuclear Medicine and Imaging

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

In Meta & Continual Learning

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

Citations in the corpus