Deconfound Semantic Shift and Incompleteness in Incremental Few-shot Semantic Segmentation
Incremental few-shot semantic segmentation (IFSS) expands segmentation capacity of the trained model to segment new-class images with few samples. However, semantic meanings may shift from background to object class or vice versa during incremental learning. Moreover, new-class samples often lack representative attribute features when the new class greatly differs from the pre-learned old class. In this paper, we propose a causal framework to discuss the cause of semantic shift and incompleteness in IFSS, and we deconfound the revealed causal effects from two aspects. First, we propose a Causal Intervention Module (CIM) to resist semantic shift. CIM progressively and adaptively updates prototypes of old class, and removes the confounder in an intervention manner. Second, a Prototype Refinement Module (PRM) is proposed to complete the missing semantics. In PRM, knowledge gained from the episode learning scheme assists in fusing features of new-class and old-class prototypes. Experiments on both PASCAL-VOC 2012 and ADE20k benchmarks demonstrate the outstanding performance of our method.
Authors
- Shaohua Wan (ORCID: https://orcid.org/0000-0001-7013-9081)
- Junyang Chen (ORCID: https://orcid.org/0000-0002-1139-8654)
- Yirui Wu (ORCID: https://orcid.org/0000-0003-3022-3718)
- Hao Li (ORCID: https://orcid.org/0000-0003-2476-0664)
- Jun Liu (ORCID: https://orcid.org/0000-0002-6004-0675)
- Tong Lu
- Yuhang Xia (ORCID: https://orcid.org/0009-0006-1675-1919)
- Lixin Yuan (ORCID: https://orcid.org/0009-0002-7490-1437)
Institutions
- Hohai University (CN)
- Shenzhen University (CN)
- Institute for Advanced Study (DE)
- Ministry of Water Resources of the People's Republic of China (CN)
- Lancaster University (GB)
- Nanjing University (CN)
Publication Details
- Journal
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2025-04-11
- DOI
- https://doi.org/10.1609/aaai.v39i8.32915
- Primary Topic
- Multimodal Machine Learning Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- Government of Jiangsu Province
- Natural Science Foundation of Guangdong Province
- Natural Science Foundation of Jiangsu Province
- Hohai University