Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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.

Sungmin Cha, Beomyoung Kim, Youngjoon Yoo, Taesup Moon• 2021

Related benchmarks

TaskDatasetResultRank
Semantic segmentationPASCAL VOC 2012
mIoU76.49
187
Semantic segmentationPascal VOC 15-1 setting 2012 (val)
mIoU (all)72.58
88
Semantic segmentationPascal VOC 15-5 setting 2012 (val)
mIoU (All)74.41
82
Semantic segmentationADE20k (100-5)
mIoU (All Classes)3.46e+3
54
Semantic segmentationPascal VOC 10-1 protocol 2012 (val)
mIoU (0-10)74.79
46
Semantic segmentationPascal VOC overlapped setting (15-1 (6 steps))
mIoU (Classes 1-15)7.26e+3
41
Continual Semantic SegmentationPascal-VOC 15-1 scenario 2012
mIoU (classes 0-15)0.784
32
Semantic segmentationPascal-VOC Disjoint 15-5 2012
mIoU (0-15)75
31
Incremental Semantic SegmentationPascal VOC disjoint setup 2012 (VOC 10-1)
mIoU (0-10)74
30
Semantic segmentationPascal VOC 5-3 protocol 2012 (val)
mIoU (Classes 0-5)76.04
29
Showing 10 of 40 rows

Other info

Code

Follow for update