Du Ye, Fu Zehua, Liu Qingjie
IEEE Trans Image Process. 2024;33:4654-4669. doi: 10.1109/TIP.2024.3444190. Epub 2024 Aug 28.
Recent attention has been devoted to the pursuit of learning semantic segmentation models exclusively from image tags, a paradigm known as image-level Weakly Supervised Semantic Segmentation (WSSS). Existing attempts adopt the Class Activation Maps (CAMs) as priors to mine object regions yet observe the imbalanced activation issue, where only the most discriminative object parts are located. In this paper, we argue that the distribution discrepancy between the discriminative and the non-discriminative parts of objects prevents the model from producing complete and precise pseudo masks as ground truths. For this purpose, we propose a Pixel-Level Domain Adaptation (PLDA) method to encourage the model in learning pixel-wise domain-invariant features. Specifically, a multi-head domain classifier trained adversarially with the feature extraction is introduced to promote the emergence of pixel features that are invariant with respect to the shift between the source (i.e., the discriminative object parts) and the target (i.e., the non-discriminative object parts) domains. In addition, we come up with a Confident Pseudo-Supervision strategy to guarantee the discriminative ability of each pixel for the segmentation task, which serves as a complement to the intra-image domain adversarial training. Our method is conceptually simple, intuitive and can be easily integrated into existing WSSS methods. Taking several strong baseline models as instances, we experimentally demonstrate the effectiveness of our approach under a wide range of settings.
最近,人们致力于仅从图像标签中学习语义分割模型,这是一种被称为图像级弱监督语义分割(WSSS)的范式。现有的尝试采用类激活映射(CAMs)作为先验来挖掘目标区域,但存在激活不均衡的问题,即仅定位了最具判别力的目标部分。在本文中,我们认为目标的判别性部分和非判别性部分之间的分布差异阻碍了模型生成完整且精确的伪掩码作为真实标签。为此,我们提出了一种像素级域适应(PLDA)方法,以鼓励模型学习逐像素的域不变特征。具体而言,引入了一个与特征提取进行对抗训练的多头域分类器,以促进相对于源域(即判别性目标部分)和目标域(即非判别性目标部分)之间的偏移具有不变性的像素特征的出现。此外,我们提出了一种置信伪监督策略,以确保每个像素对分割任务的判别能力,这作为图像内域对抗训练的补充。我们的方法在概念上简单、直观,并且可以轻松集成到现有的WSSS方法中。以几个强大的基线模型为例,我们通过实验证明了我们的方法在广泛设置下的有效性。