arXiv · 2111.08651
Diversified Multi-prototype Representation for Semi-supervised Segmentation
Abstract
This work considers semi-supervised segmentation as a dense prediction problem based on prototype vector correlation and proposes a simple way to represent each segmentation class with multiple prototypes. To avoid degenerate solutions, two regularization strategies are applied on unlabeled images. The first one leverages mutual information maximization to ensure that all prototype vectors are considered by the network. The second explicitly enforces prototypes to be orthogonal by minimizing their cosine distance. Experimental results on two benchmark medical segmentation datasets reveal our method's effectiveness in improving segmentation performance when few annotated images are available.
Explore related subjects
Keep this discovery
Jizong Peng, Christian Desrosiers, Marco Pedersoli. 2021-11-16. Diversified Multi-prototype Representation for Semi-supervised Segmentation. https://arxiv.org/abs/2111.08651
Cite the original work for its findings. Save a collection to share your selection of sources.