Boosting Unsupervised Semantic Segmentation with Principal Mask Proposals

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hahn, Oliver, Araslanov, Nikita, Schaub-Meyer, Simone, Roth, Stefan
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917794884354048
author Hahn, Oliver
Araslanov, Nikita
Schaub-Meyer, Simone
Roth, Stefan
author_facet Hahn, Oliver
Araslanov, Nikita
Schaub-Meyer, Simone
Roth, Stefan
contents Unsupervised semantic segmentation aims to automatically partition images into semantically meaningful regions by identifying global semantic categories within an image corpus without any form of annotation. Building upon recent advances in self-supervised representation learning, we focus on how to leverage these large pre-trained models for the downstream task of unsupervised segmentation. We present PriMaPs - Principal Mask Proposals - decomposing images into semantically meaningful masks based on their feature representation. This allows us to realize unsupervised semantic segmentation by fitting class prototypes to PriMaPs with a stochastic expectation-maximization algorithm, PriMaPs-EM. Despite its conceptual simplicity, PriMaPs-EM leads to competitive results across various pre-trained backbone models, including DINO and DINOv2, and across different datasets, such as Cityscapes, COCO-Stuff, and Potsdam-3. Importantly, PriMaPs-EM is able to boost results when applied orthogonally to current state-of-the-art unsupervised semantic segmentation pipelines. Code is available at https://github.com/visinf/primaps.
format Preprint
id arxiv_https___arxiv_org_abs_2404_16818
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Boosting Unsupervised Semantic Segmentation with Principal Mask Proposals
Hahn, Oliver
Araslanov, Nikita
Schaub-Meyer, Simone
Roth, Stefan
Computer Vision and Pattern Recognition
Unsupervised semantic segmentation aims to automatically partition images into semantically meaningful regions by identifying global semantic categories within an image corpus without any form of annotation. Building upon recent advances in self-supervised representation learning, we focus on how to leverage these large pre-trained models for the downstream task of unsupervised segmentation. We present PriMaPs - Principal Mask Proposals - decomposing images into semantically meaningful masks based on their feature representation. This allows us to realize unsupervised semantic segmentation by fitting class prototypes to PriMaPs with a stochastic expectation-maximization algorithm, PriMaPs-EM. Despite its conceptual simplicity, PriMaPs-EM leads to competitive results across various pre-trained backbone models, including DINO and DINOv2, and across different datasets, such as Cityscapes, COCO-Stuff, and Potsdam-3. Importantly, PriMaPs-EM is able to boost results when applied orthogonally to current state-of-the-art unsupervised semantic segmentation pipelines. Code is available at https://github.com/visinf/primaps.
title Boosting Unsupervised Semantic Segmentation with Principal Mask Proposals
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2404.16818