Title
Adversarial Structure Matching Loss for Image Segmentation.
Abstract
The per-pixel cross-entropy loss (CEL) has been widely used in structured output prediction tasks as a spatial extension of generic image classification. However, its i.i.d. assumption neglects the structural regularity present in natural images. Various attempts have been made to incorporate structural reasoning mostly through structure priors in a cooperative way where co-occuring patterns are encouraged. We, on the other hand, approach this problem from an opposing angle and propose a new framework for training such structured prediction networks via an adversarial process, in which we train a structure analyzer that provides the supervisory signals, the adversarial structure matching loss (ASML). The structure analyzer is trained to maximize ASML, or to exaggerate recurring structural mistakes usually among co-occurring patterns. On the contrary, the structured output prediction network is trained to reduce those mistakes and is thus enabled to distinguish fine-grained structures. As a result, training structured output prediction networks using ASML reduces contextual confusion among objects and improves boundary localization. We demonstrate that ASML outperforms its counterpart CEL especially in context and boundary aspects on figure-ground segmentation and semantic segmentation tasks with various base architectures, such as FCN, U-Net, DeepLab, and PSPNet.
Year
Venue
Field
2018
arXiv: Computer Vision and Pattern Recognition
Structure matching,Adversarial process,Pattern recognition,Computer science,Segmentation,Structured prediction,Image segmentation,Artificial intelligence,Contextual image classification,Prior probability,Machine learning,Adversarial system
DocType
Volume
Citations 
Journal
abs/1805.07457
1
PageRank 
References 
Authors
0.34
14
4
Name
Order
Citations
PageRank
Jyh-Jing Hwang1264.56
Tsung-Wei Ke211.02
Jianbo Shi3102071031.66
Yu, Stella X.487786.36