Deep co-training for semi-supervised image segmentation

PATTERN RECOGNITION

卷:107

Article

Peng, Jizong¹  Estrada, Guillermo²  Pedersoli, Marco¹  Desrosiers, Christian¹ 

[1] ETS Montreal, 1100 Notre Dame W, Montreal, PQ, Canada

[2] Pontificia Univ Catolica Rio de Janeiro, 225 Marques de Sao Vicente St, Rio de Janeiro, Brazil

关键词: Deep learning;    Semi-supervised learning;    Ensemble learning;    Co-training;    Image segmentation;   

DOI  :  10.1016/j.patcog.2020.107269

来源: Elsevier

【 摘 要 】

In this paper, we aim to improve the performance of semantic image segmentation in a semi-supervised setting where training is performed with a reduced set of annotated images and additional non-annotated images. We present a method based on an ensemble of deep segmentation models. Models are trained on subsets of the annotated data and use non-annotated images to exchange information with each other, similar to co-training. Diversity across models is enforced with the use of adversarial samples. We demonstrate the potential of our method on three challenging image segmentation problems, and illustrate its ability to share information between simultaneously trained models, while preserving their diversity. Results indicate clear advantages in terms of performance compared to recently proposed semi-supervised methods for segmentation. (C) 2020 Elsevier Ltd. All rights reserved.

【 授权许可】

Free