Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Stronger, Fewer, & Superior: Harnessing Vision Foundation Models for Domain Generalized Semantic Segmentation

About

In this paper, we first assess and harness various Vision Foundation Models (VFMs) in the context of Domain Generalized Semantic Segmentation (DGSS). Driven by the motivation that Leveraging Stronger pre-trained models and Fewer trainable parameters for Superior generalizability, we introduce a robust fine-tuning approach, namely Rein, to parameter-efficiently harness VFMs for DGSS. Built upon a set of trainable tokens, each linked to distinct instances, Rein precisely refines and forwards the feature maps from each layer to the next layer within the backbone. This process produces diverse refinements for different categories within a single image. With fewer trainable parameters, Rein efficiently fine-tunes VFMs for DGSS tasks, surprisingly surpassing full parameter fine-tuning. Extensive experiments across various settings demonstrate that Rein significantly outperforms state-of-the-art methods. Remarkably, with just an extra 1% of trainable parameters within the frozen backbone, Rein achieves a mIoU of 78.4% on the Cityscapes, without accessing any real urban-scene datasets.Code is available at https://github.com/w1oves/Rein.git.

Zhixiang Wei, Lin Chen, Yi Jin, Xiaoxiao Ma, Tianle Liu, Pengyang Ling, Ben Wang, Huaian Chen, Jinjin Zheng• 2023

Related benchmarks

TaskDatasetResultRank
Semantic segmentationCityscapes (val)
mIoU66.4
572
Semantic segmentationMapillary (val)
mIoU74.03
153
Semantic segmentationBDD-100K (val)
mIoU63.54
102
Semantic segmentationGTA5 to {Cityscapes, Mapillary, BDD} (test)
mIoU (Cityscapes)70.68
94
Semantic segmentationCityScapes, BDD, and Mapillary (val)
Mean mIoU68.13
85
Skin lesion classificationHAM10000 (test)
Accuracy78.6
83
Video Semantic SegmentationCityscapes-C (test)
mIoU42.53
78
Semantic segmentationMapillary
mIoU75.18
75
Semantic segmentationBDD100K (val)
mIoU60.5
72
Medical Image ClassificationMedMnist BloodMnist (test)
Accuracy95.9
65
Showing 10 of 42 rows

Other info

Code

Follow for update