TransFGU: A Top-down Approach to Fine-Grained Unsupervised Semantic Segmentation

Abstract

Unsupervised semantic segmentation aims to obtain high-level semantic representation on low-level visual features without manual annotations. Most existing methods are bottom-up approaches that try to group pixels into regions based on their visual cues or certain predefined rules. As a result, it is difficult for these bottom-up approaches to generate fine-grained semantic segmentation when coming to complicated scenes with multiple objects and some objects sharing similar visual appearance. In contrast, we propose the first top-down unsupervised semantic segmentation framework for fine-grained segmentation in extremely complicated scenarios. Specifically, we first obtain rich high-level structured semantic concept information from large-scale vision data in a self-supervised learning manner, and use such information as a prior to discover potential semantic categories presented in target datasets. Secondly, the discovered high-level semantic categories are mapped to low-level pixel features by calculating the class activate map (CAM) with respect to certain discovered semantic representation. Lastly, the obtained CAMs serve as pseudo labels to train the segmentation module and produce final semantic segmentation. Experimental results on multiple semantic segmentation benchmarks show that our topdown unsupervised segmentation is robust to both objectcentric and scene-centric datasets under different semantic granularity levels, and outperforms all the current stateof-the-art bottom-up methods. Our code is available at https://github.com/damo-cv/TransFGU .

Rethinking Alignment and Uniformity in Unsupervised Image Semantic Segmentation

“…(3) Pretrained model methods(Cho et al 2021) (Yin et al 2021) utilize pretrained models to generate better representation and introduce high-level semantic information to the model while they still focus on imagewise information and use additional information compared to our method. Our proposed method is more robust and flexible for multiple class segmenting regions.…”