Boosting Unsupervised Semantic Segmentation with Principal Mask Proposals

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hahn, Oliver, Araslanov, Nikita, Schaub-Meyer, Simone, Roth, Stefan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_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