Resolving Inconsistent Semantics in Multi-Dataset Image Segmentation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhangli, Qilong, Liu, Di, Aich, Abhishek, Metaxas, Dimitris, Schulter, Samuel
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914949876416512
author Zhangli, Qilong
Liu, Di
Aich, Abhishek
Metaxas, Dimitris
Schulter, Samuel
author_facet Zhangli, Qilong
Liu, Di
Aich, Abhishek
Metaxas, Dimitris
Schulter, Samuel
contents Leveraging multiple training datasets to scale up image segmentation models is beneficial for increasing robustness and semantic understanding. Individual datasets have well-defined ground truth with non-overlapping mask layouts and mutually exclusive semantics. However, merging them for multi-dataset training disrupts this harmony and leads to semantic inconsistencies; for example, the class "person" in one dataset and class "face" in another will require multilabel handling for certain pixels. Existing methods struggle with this setting, particularly when evaluated on label spaces mixed from the individual training sets. To overcome these issues, we introduce a simple yet effective multi-dataset training approach by integrating language-based embeddings of class names and label space-specific query embeddings. Our method maintains high performance regardless of the underlying inconsistencies between training datasets. Notably, on four benchmark datasets with label space inconsistencies during inference, we outperform previous methods by 1.6% mIoU for semantic segmentation, 9.1% PQ for panoptic segmentation, 12.1% AP for instance segmentation, and 3.0% in the newly proposed PIQ metric.
format Preprint
id arxiv_https___arxiv_org_abs_2409_09893
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Resolving Inconsistent Semantics in Multi-Dataset Image Segmentation
Zhangli, Qilong
Liu, Di
Aich, Abhishek
Metaxas, Dimitris
Schulter, Samuel
Computer Vision and Pattern Recognition
Leveraging multiple training datasets to scale up image segmentation models is beneficial for increasing robustness and semantic understanding. Individual datasets have well-defined ground truth with non-overlapping mask layouts and mutually exclusive semantics. However, merging them for multi-dataset training disrupts this harmony and leads to semantic inconsistencies; for example, the class "person" in one dataset and class "face" in another will require multilabel handling for certain pixels. Existing methods struggle with this setting, particularly when evaluated on label spaces mixed from the individual training sets. To overcome these issues, we introduce a simple yet effective multi-dataset training approach by integrating language-based embeddings of class names and label space-specific query embeddings. Our method maintains high performance regardless of the underlying inconsistencies between training datasets. Notably, on four benchmark datasets with label space inconsistencies during inference, we outperform previous methods by 1.6% mIoU for semantic segmentation, 9.1% PQ for panoptic segmentation, 12.1% AP for instance segmentation, and 3.0% in the newly proposed PIQ metric.
title Resolving Inconsistent Semantics in Multi-Dataset Image Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.09893