Adversarial learning for semi-supervised semantic segmentation

Wei Chih Hung, Yi Hsuan Tsai, Yan Ting Liou, Yen Yu Lin, Ming Hsuan Yang

Research output: Contribution to conferencePaper

Abstract

We propose a method for semi-supervised semantic segmentation using an adversarial network. While most existing discriminators are trained to classify input images as real or fake on the image level, we design a discriminator in a fully convolutional manner to differentiate the predicted probability maps from the ground truth segmentation distribution with the consideration of the spatial resolution. We show that the proposed discriminator can be used to improve semantic segmentation accuracy by coupling the adversarial loss with the standard cross entropy loss of the proposed model. In addition, the fully convolutional discriminator enables semi-supervised learning through discovering the trustworthy regions in predicted results of unlabeled images, thereby providing additional supervisory signals. In contrast to existing methods that utilize weakly-labeled images, our method leverages unlabeled images to enhance the segmentation model. Experimental results on the PASCAL VOC 2012 and Cityscapes datasets demonstrate the effectiveness of the proposed algorithm.

Original languageEnglish
Publication statusPublished - 2019 Jan 1
Event29th British Machine Vision Conference, BMVC 2018 - Newcastle, United Kingdom
Duration: 2018 Sep 32018 Sep 6

Conference

Conference29th British Machine Vision Conference, BMVC 2018
CountryUnited Kingdom
CityNewcastle
Period18/9/318/9/6

All Science Journal Classification (ASJC) codes

  • Computer Vision and Pattern Recognition

Fingerprint Dive into the research topics of 'Adversarial learning for semi-supervised semantic segmentation'. Together they form a unique fingerprint.

  • Cite this

    Hung, W. C., Tsai, Y. H., Liou, Y. T., Lin, Y. Y., & Yang, M. H. (2019). Adversarial learning for semi-supervised semantic segmentation. Paper presented at 29th British Machine Vision Conference, BMVC 2018, Newcastle, United Kingdom.