CIAN: Cross-Image Affinity Net for Weakly Supervised Semantic Segmentation

• DOI: 10.1609/aaai.v34i07.6705
• Publication Date: 2020-06-29T18:45:57Z
• Authors: Junsong Fan; Zhaoxiang Zhang; Tieniu Tan; Chunfeng Song; Jun Xiao
• Keywords: FOS: Computer and information sciences; 03 medical and health sciences; 0302 clinical medicine; Computer Vision and Pattern Recognition (cs.CV); Computer Science - Computer Vision and Pattern Recognition; 0202 electrical engineering, electronic engineering, information engineering; 02 engineering and technology

Weakly supervised semantic segmentation with only image-level labels saves large human effort to annotate pixel-level labels. Cutting-edge approaches rely on various innovative constraints and heuristic rules to generate the masks for every single image. Although great progress has been achieved by these methods, they treat each image independently and do not take account of the relationships across different images. In this paper, however, we argue that the cross-image relationship is vital for weakly supervised segmentation. Because it connects related regions across images, where supplementary representations can be propagated to obtain more consistent and integral regions. To leverage this information, we propose an end-to-end cross-image affinity module, which exploits pixel-level cross-image relationships with only image-level labels. By means of this, our approach achieves 64.3% and 65.3% mIoU on Pascal VOC 2012 validation and test set respectively, which is a new state-of-the-art result by only using image-level labels for weakly supervised semantic segmentation, demonstrating the superiority of our approach.