HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SemiVL: Semi-Supervised Semantic Segmentation with Vision-Language Guidance

Lukas Hoyer David Joseph Tan Muhammad Ferjad Naeem Luc Van Gool Federico Tombari

SemiVL: Semi-Supervised Semantic Segmentation with Vision-Language Guidance

Abstract

In semi-supervised semantic segmentation, a model is trained with a limited number of labeled images along with a large corpus of unlabeled images to reduce the high annotation effort. While previous methods are able to learn good segmentation boundaries, they are prone to confuse classes with similar visual appearance due to the limited supervision. On the other hand, vision-language models (VLMs) are able to learn diverse semantic knowledge from image-caption datasets but produce noisy segmentation due to the image-level training. In SemiVL, we propose to integrate rich priors from VLM pre-training into semi-supervised semantic segmentation to learn better semantic decision boundaries. To adapt the VLM from global to local reasoning, we introduce a spatial fine-tuning strategy for label-efficient learning. Further, we design a language-guided decoder to jointly reason over vision and language. Finally, we propose to handle inherent ambiguities in class labels by providing the model with language guidance in the form of class definitions. We evaluate SemiVL on 4 semantic segmentation datasets, where it significantly outperforms previous semi-supervised methods. For instance, SemiVL improves the state-of-the-art by +13.5 mIoU on COCO with 232 annotated images and by +6.1 mIoU on Pascal VOC with 92 labels. Project page: https://github.com/google-research/semivl

Code Repositories

google-research/semivl
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
semi-supervised-semantic-segmentation-on-1SemiVL (ViT-B/16)
Validation mIoU: 80.3%
semi-supervised-semantic-segmentation-on-10SemiVL (ViT-B/16
Validation mIoU: 87.3
semi-supervised-semantic-segmentation-on-10UniMatch (ViT-B/16)
Validation mIoU: 84.0
semi-supervised-semantic-segmentation-on-2SemiVL (ViT-B/16)
Validation mIoU: 79.4%
semi-supervised-semantic-segmentation-on-22SemiVL (ViT-B/16)
Validation mIoU: 77.9
semi-supervised-semantic-segmentation-on-27UniMatch (ViT-B/16)
Validation mIoU: 77.9
semi-supervised-semantic-segmentation-on-27SemiVL (ViT-B/16)
Validation mIoU: 84.0
semi-supervised-semantic-segmentation-on-28UniMatch (ViT-B/16)
Validation mIoU: 80.1
semi-supervised-semantic-segmentation-on-28SemiVL (ViT-B/16)
Validation mIoU: 85.6
semi-supervised-semantic-segmentation-on-29UniMatch (ViT-B/16)
Validation mIoU: 82.0
semi-supervised-semantic-segmentation-on-29SemiVL (ViT-B/16)
Validation mIoU: 86.0
semi-supervised-semantic-segmentation-on-3SemiVL (ViT-B/16)
Validation mIoU: 76.2
semi-supervised-semantic-segmentation-on-30UniMatch (ViT-B/16)
Validation mIoU: 83.3
semi-supervised-semantic-segmentation-on-30SemiVL (ViT-B/16)
Validation mIoU: 86.7
semi-supervised-semantic-segmentation-on-41SemiVL
Validation mIoU: 35.1
semi-supervised-semantic-segmentation-on-42SemiVL
Validation mIoU: 37.2
semi-supervised-semantic-segmentation-on-8SemiVL (ViT-B/16)
Validation mIoU: 80.6%
semi-supervised-semantic-segmentation-on-cocoSemiVL
Validation mIoU: 50.1
semi-supervised-semantic-segmentation-on-coco-1SemiVL
Validation mIoU: 52.8
semi-supervised-semantic-segmentation-on-coco-2SemiVL
Validation mIoU: 53.6
semi-supervised-semantic-segmentation-on-coco-3SemiVL
Validation mIoU: 55.4
semi-supervised-semantic-segmentation-on-coco-4SemiVL
Validation mIoU: 56.5

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp