AWF: Adaptive Weight Fusion for Enhanced Class Incremental Semantic Segmentation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Sun, Zechao, Piao, Shuying, Jin, Haolin, Dong, Chang, Yue, Lin, Chen, Weitong, Zhou, Luping
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866918075462320128
author Sun, Zechao
Piao, Shuying
Jin, Haolin
Dong, Chang
Yue, Lin
Chen, Weitong
Zhou, Luping
author_facet Sun, Zechao
Piao, Shuying
Jin, Haolin
Dong, Chang
Yue, Lin
Chen, Weitong
Zhou, Luping
contents Class Incremental Semantic Segmentation (CISS) aims to mitigate catastrophic forgetting by maintaining a balance between previously learned and newly introduced knowledge. Existing methods, primarily based on regularization techniques like knowledge distillation, help preserve old knowledge but often face challenges in effectively integrating new knowledge, resulting in limited overall improvement. Endpoints Weight Fusion (EWF) method, while simple, effectively addresses some of these limitations by dynamically fusing the model weights from previous steps with those from the current step, using a fusion parameter alpha determined by the relative number of previously known classes and newly introduced classes. However, the simplicity of the alpha calculation may limit its ability to fully capture the complexities of different task scenarios, potentially leading to suboptimal fusion outcomes. In this paper, we propose an enhanced approach called Adaptive Weight Fusion (AWF), which introduces an alternating training strategy for the fusion parameter, allowing for more flexible and adaptive weight integration. AWF achieves superior performance by better balancing the retention of old knowledge with the learning of new classes, significantly improving results on benchmark CISS tasks compared to the original EWF. And our experiment code will be released on Github.
format Preprint
id arxiv_https___arxiv_org_abs_2409_08516
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle AWF: Adaptive Weight Fusion for Enhanced Class Incremental Semantic Segmentation
Sun, Zechao
Piao, Shuying
Jin, Haolin
Dong, Chang
Yue, Lin
Chen, Weitong
Zhou, Luping
Computer Vision and Pattern Recognition
Class Incremental Semantic Segmentation (CISS) aims to mitigate catastrophic forgetting by maintaining a balance between previously learned and newly introduced knowledge. Existing methods, primarily based on regularization techniques like knowledge distillation, help preserve old knowledge but often face challenges in effectively integrating new knowledge, resulting in limited overall improvement. Endpoints Weight Fusion (EWF) method, while simple, effectively addresses some of these limitations by dynamically fusing the model weights from previous steps with those from the current step, using a fusion parameter alpha determined by the relative number of previously known classes and newly introduced classes. However, the simplicity of the alpha calculation may limit its ability to fully capture the complexities of different task scenarios, potentially leading to suboptimal fusion outcomes. In this paper, we propose an enhanced approach called Adaptive Weight Fusion (AWF), which introduces an alternating training strategy for the fusion parameter, allowing for more flexible and adaptive weight integration. AWF achieves superior performance by better balancing the retention of old knowledge with the learning of new classes, significantly improving results on benchmark CISS tasks compared to the original EWF. And our experiment code will be released on Github.
title AWF: Adaptive Weight Fusion for Enhanced Class Incremental Semantic Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.08516