Mitigating Background Shift in Class-Incremental Semantic Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Park, Gilhan, Moon, WonJun, Lee, SuBeen, Kim, Tae-Young, Heo, Jae-Pil
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914873217122304
author Park, Gilhan
Moon, WonJun
Lee, SuBeen
Kim, Tae-Young
Heo, Jae-Pil
author_facet Park, Gilhan
Moon, WonJun
Lee, SuBeen
Kim, Tae-Young
Heo, Jae-Pil
contents Class-Incremental Semantic Segmentation(CISS) aims to learn new classes without forgetting the old ones, using only the labels of the new classes. To achieve this, two popular strategies are employed: 1) pseudo-labeling and knowledge distillation to preserve prior knowledge; and 2) background weight transfer, which leverages the broad coverage of background in learning new classes by transferring background weight to the new class classifier. However, the first strategy heavily relies on the old model in detecting old classes while undetected pixels are regarded as the background, thereby leading to the background shift towards the old classes(i.e., misclassification of old class as background). Additionally, in the case of the second approach, initializing the new class classifier with background knowledge triggers a similar background shift issue, but towards the new classes. To address these issues, we propose a background-class separation framework for CISS. To begin with, selective pseudo-labeling and adaptive feature distillation are to distill only trustworthy past knowledge. On the other hand, we encourage the separation between the background and new classes with a novel orthogonal objective along with label-guided output distillation. Our state-of-the-art results validate the effectiveness of these proposed methods.
format Preprint
id arxiv_https___arxiv_org_abs_2407_11859
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Mitigating Background Shift in Class-Incremental Semantic Segmentation
Park, Gilhan
Moon, WonJun
Lee, SuBeen
Kim, Tae-Young
Heo, Jae-Pil
Computer Vision and Pattern Recognition
Class-Incremental Semantic Segmentation(CISS) aims to learn new classes without forgetting the old ones, using only the labels of the new classes. To achieve this, two popular strategies are employed: 1) pseudo-labeling and knowledge distillation to preserve prior knowledge; and 2) background weight transfer, which leverages the broad coverage of background in learning new classes by transferring background weight to the new class classifier. However, the first strategy heavily relies on the old model in detecting old classes while undetected pixels are regarded as the background, thereby leading to the background shift towards the old classes(i.e., misclassification of old class as background). Additionally, in the case of the second approach, initializing the new class classifier with background knowledge triggers a similar background shift issue, but towards the new classes. To address these issues, we propose a background-class separation framework for CISS. To begin with, selective pseudo-labeling and adaptive feature distillation are to distill only trustworthy past knowledge. On the other hand, we encourage the separation between the background and new classes with a novel orthogonal objective along with label-guided output distillation. Our state-of-the-art results validate the effectiveness of these proposed methods.
title Mitigating Background Shift in Class-Incremental Semantic Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.11859