Semi-Supervised Semantic Segmentation via Marginal Contextual Information

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kimhi, Moshe, Kimhi, Shai, Zheltonozhskii, Evgenii, Litany, Or, Baskin, Chaim
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929406920884224
author Kimhi, Moshe
Kimhi, Shai
Zheltonozhskii, Evgenii
Litany, Or
Baskin, Chaim
author_facet Kimhi, Moshe
Kimhi, Shai
Zheltonozhskii, Evgenii
Litany, Or
Baskin, Chaim
contents We present a novel confidence refinement scheme that enhances pseudo labels in semi-supervised semantic segmentation. Unlike existing methods, which filter pixels with low-confidence predictions in isolation, our approach leverages the spatial correlation of labels in segmentation maps by grouping neighboring pixels and considering their pseudo labels collectively. With this contextual information, our method, named S4MC, increases the amount of unlabeled data used during training while maintaining the quality of the pseudo labels, all with negligible computational overhead. Through extensive experiments on standard benchmarks, we demonstrate that S4MC outperforms existing state-of-the-art semi-supervised learning approaches, offering a promising solution for reducing the cost of acquiring dense annotations. For example, S4MC achieves a 1.39 mIoU improvement over the prior art on PASCAL VOC 12 with 366 annotated images. The code to reproduce our experiments is available at https://s4mcontext.github.io/
format Preprint
id arxiv_https___arxiv_org_abs_2308_13900
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Semi-Supervised Semantic Segmentation via Marginal Contextual Information
Kimhi, Moshe
Kimhi, Shai
Zheltonozhskii, Evgenii
Litany, Or
Baskin, Chaim
Computer Vision and Pattern Recognition
Machine Learning
We present a novel confidence refinement scheme that enhances pseudo labels in semi-supervised semantic segmentation. Unlike existing methods, which filter pixels with low-confidence predictions in isolation, our approach leverages the spatial correlation of labels in segmentation maps by grouping neighboring pixels and considering their pseudo labels collectively. With this contextual information, our method, named S4MC, increases the amount of unlabeled data used during training while maintaining the quality of the pseudo labels, all with negligible computational overhead. Through extensive experiments on standard benchmarks, we demonstrate that S4MC outperforms existing state-of-the-art semi-supervised learning approaches, offering a promising solution for reducing the cost of acquiring dense annotations. For example, S4MC achieves a 1.39 mIoU improvement over the prior art on PASCAL VOC 12 with 366 annotated images. The code to reproduce our experiments is available at https://s4mcontext.github.io/
title Semi-Supervised Semantic Segmentation via Marginal Contextual Information
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2308.13900