Group-Wise Semantic Mining for Weakly Supervised Semantic Segmentation

Authors

  • Xueyi Li School of Computer Science and Technology, Beijing Institute of Technology, China
  • Tianfei Zhou Computer Vision Laboratory, ETH Zurich, Switzerland
  • Jianwu Li School of Computer Science and Technology, Beijing Institute of Technology, China
  • Yi Zhou School of Computer Science and Engineering, Southeast University, China
  • Zhaoxiang Zhang Center for Research on Intelligent Perception and Computing, Chinese Academy of Sciences, China

Keywords:

Segmentation, Scene Analysis & Understanding, Applications, Other Foundations of Computer Vision

Abstract

Acquiring sufficient ground-truth supervision to train deep vi- sual models has been a bottleneck over the years due to the data-hungry nature of deep learning. This is exacerbated in some structured prediction tasks, such as semantic segmen- tation, which requires pixel-level annotations. This work ad- dresses weakly supervised semantic segmentation (WSSS), with the goal of bridging the gap between image-level anno- tations and pixel-level segmentation. We formulate WSSS as a novel group-wise learning task that explicitly models se- mantic dependencies in a group of images to estimate more reliable pseudo ground-truths, which can be used for training more accurate segmentation models. In particular, we devise a graph neural network (GNN) for group-wise semantic min- ing, wherein input images are represented as graph nodes, and the underlying relations between a pair of images are char- acterized by an efficient co-attention mechanism. Moreover, in order to prevent the model from paying excessive atten- tion to common semantics only, we further propose a graph dropout layer, encouraging the model to learn more accurate and complete object responses. The whole network is end-to- end trainable by iterative message passing, which propagates interaction cues over the images to progressively improve the performance. We conduct experiments on the popular PAS- CAL VOC 2012 and COCO benchmarks, and our model yields state-of-the-art performance. Our code is available at: https://github.com/Lixy1997/Group-WSSS.

Downloads

Published

2021-05-18

How to Cite

Li, X., Zhou, T., Li, J., Zhou, Y., & Zhang, Z. (2021). Group-Wise Semantic Mining for Weakly Supervised Semantic Segmentation. Proceedings of the AAAI Conference on Artificial Intelligence, 35(3), 1984-1992. Retrieved from https://ojs.aaai.org/index.php/AAAI/article/view/16294

Issue

Section

AAAI Technical Track on Computer Vision II