Large-scale Unsupervised Semantic Segmentation

6 Jun 2021  ยท  ShangHua Gao, Zhong-Yu Li, Ming-Hsuan Yang, Ming-Ming Cheng, Junwei Han, Philip Torr ยท

Empowered by large datasets, e.g., ImageNet, unsupervised learning on large-scale data has enabled significant advances for classification tasks. However, whether the large-scale unsupervised semantic segmentation can be achieved remains unknown. There are two major challenges: i) we need a large-scale benchmark for assessing algorithms; ii) we need to develop methods to simultaneously learn category and shape representation in an unsupervised manner. In this work, we propose a new problem of large-scale unsupervised semantic segmentation (LUSS) with a newly created benchmark dataset to help the research progress. Building on the ImageNet dataset, we propose the ImageNet-S dataset with 1.2 million training images and 50k high-quality semantic segmentation annotations for evaluation. Our benchmark has a high data diversity and a clear task objective. We also present a simple yet effective method that works surprisingly well for LUSS. In addition, we benchmark related un/weakly/fully supervised methods accordingly, identifying the challenges and possible directions of LUSS. The benchmark and source code is publicly available at https://github.com/LUSSeg.

PDF Abstract

Datasets


Introduced in the Paper:

ImageNet-S

Used in the Paper:

MS COCO Cityscapes ImageNet-1K
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Unsupervised Semantic Segmentation ImageNet-S PASS mIoU (val) 11.5 # 1
mIoU (test) 11.0 # 1
Semantic Segmentation ImageNet-S PASS (ResNet-50 D32, 224x224, LUSS) mIoU (val) 21.0 # 20
mIoU (test) 20.3 # 18
Semantic Segmentation ImageNet-S PASS (ResNet-50 D16, 224x224, LUSS) mIoU (val) 21.6 # 19
mIoU (test) 20.8 # 17
Unsupervised Semantic Segmentation ImageNet-S-300 PASS mIoU (val) 18 # 1
mIoU (test) 18.1 # 1
Unsupervised Semantic Segmentation ImageNet-S-50 PASS (+Saliency map) mIoU (val) 43.3 # 1
mIoU (test) 42.3 # 1
Unsupervised Semantic Segmentation ImageNet-S-50 PASS mIoU (val) 32.4 # 2
mIoU (test) 32.0 # 2

Methods


No methods listed for this paper. Add relevant methods here