Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

PLOP: Learning without Forgetting for Continual Semantic Segmentation

About

Deep learning approaches are nowadays ubiquitously used to tackle computer vision tasks such as semantic segmentation, requiring large datasets and substantial computational power. Continual learning for semantic segmentation (CSS) is an emerging trend that consists in updating an old model by sequentially adding new classes. However, continual learning methods are usually prone to catastrophic forgetting. This issue is further aggravated in CSS where, at each step, old classes from previous iterations are collapsed into the background. In this paper, we propose Local POD, a multi-scale pooling distillation scheme that preserves long- and short-range spatial relationships at feature level. Furthermore, we design an entropy-based pseudo-labelling of the background w.r.t. classes predicted by the old model to deal with background shift and avoid catastrophic forgetting of the old classes. Our approach, called PLOP, significantly outperforms state-of-the-art methods in existing CSS scenarios, as well as in newly proposed challenging benchmarks.

Arthur Douillard, Yifu Chen, Arnaud Dapogny, Matthieu Cord• 2020

Related benchmarks

TaskDatasetResultRank
Semantic segmentationPASCAL VOC 2012 (test)
mIoU30.45
1415
Semantic segmentationPASCAL VOC 2012
mIoU73.54
218
Image-to-Brain RetrievalNatural Scenes Dataset (NSD)
Average Performance53.08
100
Semantic segmentationPascal VOC 15-1 setting 2012 (val)
mIoU (all)66.7
88
Semantic segmentationPascal VOC 15-5 setting 2012 (val)
mIoU (All)75.44
82
Semantic segmentationADE20k (100-5)
mIoU (All Classes)2.88e+3
54
Semantic segmentationPascal VOC 10-1 protocol 2012 (val)
mIoU (0-10)57.94
46
Semantic segmentationPascal VOC overlapped setting (15-1 (6 steps))
mIoU (Classes 1-15)6.51e+3
41
Class Incremental Panoptic SegmentationADE20K (val)
PQ (Initial Classes)45.8
32
Continual Semantic SegmentationPascal-VOC 15-1 scenario 2012
mIoU (classes 0-15)0.651
32
Showing 10 of 114 rows
...

Other info

Code

Follow for update