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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | PASCAL VOC 2012 | mIoU76.49 | 187 | |
| Semantic segmentation | Pascal VOC 15-1 setting 2012 (val) | mIoU (all)72.58 | 88 | |
| Semantic segmentation | Pascal VOC 15-5 setting 2012 (val) | mIoU (All)74.41 | 82 | |
| Semantic segmentation | ADE20k (100-5) | mIoU (All Classes)3.46e+3 | 54 | |
| Semantic segmentation | Pascal VOC 10-1 protocol 2012 (val) | mIoU (0-10)74.79 | 46 | |
| Semantic segmentation | Pascal VOC overlapped setting (15-1 (6 steps)) | mIoU (Classes 1-15)7.26e+3 | 41 | |
| Continual Semantic Segmentation | Pascal-VOC 15-1 scenario 2012 | mIoU (classes 0-15)0.784 | 32 | |
| Semantic segmentation | Pascal-VOC Disjoint 15-5 2012 | mIoU (0-15)75 | 31 | |
| Incremental Semantic Segmentation | Pascal VOC disjoint setup 2012 (VOC 10-1) | mIoU (0-10)74 | 30 | |
| Semantic segmentation | Pascal VOC 5-3 protocol 2012 (val) | mIoU (Classes 0-5)76.04 | 29 |