Representation Compensation Networks for Continual Semantic Segmentation

• Published in: Proceedings (IEEE Computer Society Conference on Computer Vision and Pattern Recognition. Online) pp. 7043 - 7054
• Main Authors: Zhang, Chang-Bin, Xiao, Jia-Wen, Liu, Xialei, Chen, Ying-Cong, Cheng, Ming-Ming
• Format: Conference Proceeding
• Language: English
• Published: IEEE 01.06.2022
• Subjects: Codes; Computational modeling; Computer vision; Deep learning; grouping and shape analysis; Neural networks; Representation learning; Semantics; Transfer/low-shot/long-tail learning; Segmentation
• DOI: 10.1109/CVPR52688.2022.00692

In this work, we study the continual semantic segmentation problem, where the deep neural networks are required to incorporate new classes continually without catastrophic forgetting. We propose to use a structural re-parameterization mechanism, named representation compensation (RC) module, to decouple the representation learning of both old and new knowledge. The RC module consists of two dynamically evolved branches with one frozen and one trainable. Besides, we design a pooled cube knowledge distillation strategy on both spatial and channel dimensions to further enhance the plasticity and stability of the model. We conduct experiments on two challenging continual semantic segmentation scenarios, continual class segmentation and continual domain segmentation. Without any extra computational overhead and parameters during inference, our method outperforms state-of-the-art performance. The code is available at https://github.com/zhangchbin/RCIL.