Training-Free Open-Vocabulary Segmentation with Offline Diffusion-Augmented Prototype Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Barsellotti, Luca, Amoroso, Roberto, Cornia, Marcella, Baraldi, Lorenzo, Cucchiara, Rita
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929308308602880
author Barsellotti, Luca
Amoroso, Roberto
Cornia, Marcella
Baraldi, Lorenzo
Cucchiara, Rita
author_facet Barsellotti, Luca
Amoroso, Roberto
Cornia, Marcella
Baraldi, Lorenzo
Cucchiara, Rita
contents Open-vocabulary semantic segmentation aims at segmenting arbitrary categories expressed in textual form. Previous works have trained over large amounts of image-caption pairs to enforce pixel-level multimodal alignments. However, captions provide global information about the semantics of a given image but lack direct localization of individual concepts. Further, training on large-scale datasets inevitably brings significant computational costs. In this paper, we propose FreeDA, a training-free diffusion-augmented method for open-vocabulary semantic segmentation, which leverages the ability of diffusion models to visually localize generated concepts and local-global similarities to match class-agnostic regions with semantic classes. Our approach involves an offline stage in which textual-visual reference embeddings are collected, starting from a large set of captions and leveraging visual and semantic contexts. At test time, these are queried to support the visual matching process, which is carried out by jointly considering class-agnostic regions and global semantic similarities. Extensive analyses demonstrate that FreeDA achieves state-of-the-art performance on five datasets, surpassing previous methods by more than 7.0 average points in terms of mIoU and without requiring any training.
format Preprint
id arxiv_https___arxiv_org_abs_2404_06542
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Training-Free Open-Vocabulary Segmentation with Offline Diffusion-Augmented Prototype Generation
Barsellotti, Luca
Amoroso, Roberto
Cornia, Marcella
Baraldi, Lorenzo
Cucchiara, Rita
Computer Vision and Pattern Recognition
Open-vocabulary semantic segmentation aims at segmenting arbitrary categories expressed in textual form. Previous works have trained over large amounts of image-caption pairs to enforce pixel-level multimodal alignments. However, captions provide global information about the semantics of a given image but lack direct localization of individual concepts. Further, training on large-scale datasets inevitably brings significant computational costs. In this paper, we propose FreeDA, a training-free diffusion-augmented method for open-vocabulary semantic segmentation, which leverages the ability of diffusion models to visually localize generated concepts and local-global similarities to match class-agnostic regions with semantic classes. Our approach involves an offline stage in which textual-visual reference embeddings are collected, starting from a large set of captions and leveraging visual and semantic contexts. At test time, these are queried to support the visual matching process, which is carried out by jointly considering class-agnostic regions and global semantic similarities. Extensive analyses demonstrate that FreeDA achieves state-of-the-art performance on five datasets, surpassing previous methods by more than 7.0 average points in terms of mIoU and without requiring any training.
title Training-Free Open-Vocabulary Segmentation with Offline Diffusion-Augmented Prototype Generation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2404.06542