A Novel Multi-Loss-Based Deep Adversarial Network For Handling Challenging Cases In Semi-Supervised Image Semantic Segmentation
Wei Huang,Zhanfei Shao,Mingyuan Luo,Peng Zhang,Yufei Zha
DOI: https://doi.org/10.1016/j.patrec.2021.03.017
IF: 4.757
2021-01-01
Pattern Recognition Letters
Abstract:Image semantic segmentation is popular in computer vision and pattern recognition, since the highlevel semantic understanding of images can be effectively realized. Based on whether and to what extent the training data should be labeled, most image semantic segmentation methods can be categorized into fully-supervised learning-based methods, weakly-supervised learning-based methods, and semi-supervised learning-based methods. Among them, semi-supervised image semantic segmentation receives increasing popularity recently, because of its flexibility and convenience in requiring partial training data to be labeled. Although semi-supervised image semantic segmentation is promising, its state-of the-arts haven't obtained satisfactory performance when handling challenging cases, including poor illumination, small-sized targets, multi-targets with the same semantics, etc. To tackle the above dilemmas, a novel multi-loss-based deep adversarial network is proposed in this paper. Technically, the more robust WGAN-GP model is utilized as the backbone of the novel network, instead of the conventional GAN model. Moreover, multiple losses including the cross entropy loss, the edge detection loss, the adversarial loss, and the semi-supervised loss, are all incorporated during the novel network's training. Experimental analyses based on challenging cases shortlisted from the Pascal VOC 2012 dataset and the Cityscapes dataset suggest that, the novel network is capable to outperform state-of-the-arts in semi-supervised image semantic segmentation.(c) 2021 Elsevier B.V. All rights reserved.