Computer Science > Computer Vision and Pattern Recognition
[Submitted on 7 Dec 2016 (v1), revised 23 Dec 2016 (this version, v2), latest version 9 Apr 2017 (v3)]
Title:Mining Pixels: Weakly Supervised Semantic Segmentation Using Image Labels
View PDFAbstract:We consider the task of learning a classifier for semantic segmentation using weak supervision, in this case, image labels specifying the objects within the image. Our method uses deep convolutional neural networks (CNNs) and adopts an Expectation-Maximization (EM) based approach maintaining the uncertainty on pixel labels. We focus on the following three crucial aspects of the EM based approach: (i) initialization; (ii) latent posterior estimation (E step) and (iii) the parameter update (M step). We show that {\em saliency} and {\em attention} maps provide good cues to learn an initialization model and allows us to skip the bad local minimum to which EM methods are otherwise traditionally prone. In order to update the parameters, we propose minimizing the combination of the standard \textit{softmax} loss and the KL divergence between the true latent posterior and the likelihood given by the CNN. We argue that this combination is more robust to wrong predictions made by the expectation step of the EM method. We support this argument with empirical and visual results. We additionally incorporate an approximate intersection-over-union (IoU) term into the loss function for better parameter estimation. Extensive experiments and discussions show that: (i) our method is very simple and intuitive; (ii) requires only image-level labels; and (iii) consistently outperforms other weakly supervised state-of-the-art methods with a very high margin on the PASCAL VOC 2012 dataset.
Submission history
From: Ming-Ming Cheng Dr [view email][v1] Wed, 7 Dec 2016 02:37:51 UTC (4,417 KB)
[v2] Fri, 23 Dec 2016 03:51:04 UTC (4,373 KB)
[v3] Sun, 9 Apr 2017 08:32:03 UTC (4,856 KB)
Bibliographic and Citation Tools
Bibliographic Explorer (What is the Explorer?)
Litmaps (What is Litmaps?)
scite Smart Citations (What are Smart Citations?)
Code, Data and Media Associated with this Article
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub (What is DagsHub?)
Gotit.pub (What is GotitPub?)
Papers with Code (What is Papers with Code?)
ScienceCast (What is ScienceCast?)
Demos
Recommenders and Search Tools
Influence Flower (What are Influence Flowers?)
Connected Papers (What is Connected Papers?)
CORE Recommender (What is CORE?)
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.