Masked Image Modeling Boosting Semi-Supervised Semantic Segmentation
This work addresses the challenge of improving semantic segmentation accuracy with limited labeled data, which is crucial for applications like autonomous driving and medical imaging, though it appears incremental by building on existing self-supervised methods.
The paper tackles the problem of semi-supervised semantic segmentation by integrating masked image modeling, a generative self-supervised paradigm, to enhance regularization from unlabeled data. The result is state-of-the-art performance on benchmarks, as demonstrated through extensive experiments.
In view of the fact that semi- and self-supervised learning share a fundamental principle, effectively modeling knowledge from unlabeled data, various semi-supervised semantic segmentation methods have integrated representative self-supervised learning paradigms for further regularization. However, the potential of the state-of-the-art generative self-supervised paradigm, masked image modeling, has been scarcely studied. This paradigm learns the knowledge through establishing connections between the masked and visible parts of masked image, during the pixel reconstruction process. By inheriting and extending this insight, we successfully leverage masked image modeling to boost semi-supervised semantic segmentation. Specifically, we introduce a novel class-wise masked image modeling that independently reconstructs different image regions according to their respective classes. In this way, the mask-induced connections are established within each class, mitigating the semantic confusion that arises from plainly reconstructing images in basic masked image modeling. To strengthen these intra-class connections, we further develop a feature aggregation strategy that minimizes the distances between features corresponding to the masked and visible parts within the same class. Additionally, in semantic space, we explore the application of masked image modeling to enhance regularization. Extensive experiments conducted on well-known benchmarks demonstrate that our approach achieves state-of-the-art performance. The code will be available at https://github.com/haoxt/S4MIM.