Expand-and-Quantize: Unsupervised Semantic Segmentation Using High-Dimensional Space and Product Quantization

12 Dec 2023  ·  Jiyoung Kim, Kyuhong Shim, Insu Lee, Byonghyo Shim ·

Unsupervised semantic segmentation (USS) aims to discover and recognize meaningful categories without any labels. For a successful USS, two key abilities are required: 1) information compression and 2) clustering capability. Previous methods have relied on feature dimension reduction for information compression, however, this approach may hinder the process of clustering. In this paper, we propose a novel USS framework called Expand-and-Quantize Unsupervised Semantic Segmentation (EQUSS), which combines the benefits of high-dimensional spaces for better clustering and product quantization for effective information compression. Our extensive experiments demonstrate that EQUSS achieves state-of-the-art results on three standard benchmarks. In addition, we analyze the entropy of USS features, which is the first step towards understanding USS from the perspective of information theory.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Unsupervised Semantic Segmentation Cityscapes test EQUSS mIoU 22.0 # 5
Accuracy 79.9 # 6
Unsupervised Semantic Segmentation COCO-Stuff-27 EQUSS Accuracy 53.8 # 10
mIoU 25.8 # 7
Unsupervised Semantic Segmentation COCO-Stuff-27 EQUSS (ViT-S) Accuracy 53.8 # 10
mIoU 25.8 # 7
Unsupervised Semantic Segmentation Potsdam-3 EQUSS Accuracy 82.0 # 3

Methods


No methods listed for this paper. Add relevant methods here