SSR: SAM is a Strong Regularizer for domain adaptive semantic segmentation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ge, Yanqi, Huang, Ye, Li, Wen, Duan, Lixin
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914653499555840
author Ge, Yanqi
Huang, Ye
Li, Wen
Duan, Lixin
author_facet Ge, Yanqi
Huang, Ye
Li, Wen
Duan, Lixin
contents We introduced SSR, which utilizes SAM (segment-anything) as a strong regularizer during training, to greatly enhance the robustness of the image encoder for handling various domains. Specifically, given the fact that SAM is pre-trained with a large number of images over the internet, which cover a diverse variety of domains, the feature encoding extracted by the SAM is obviously less dependent on specific domains when compared to the traditional ImageNet pre-trained image encoder. Meanwhile, the ImageNet pre-trained image encoder is still a mature choice of backbone for the semantic segmentation task, especially when the SAM is category-irrelevant. As a result, our SSR provides a simple yet highly effective design. It uses the ImageNet pre-trained image encoder as the backbone, and the intermediate feature of each stage (ie there are 4 stages in MiT-B5) is regularized by SAM during training. After extensive experimentation on GTA5$\rightarrow$Cityscapes, our SSR significantly improved performance over the baseline without introducing any extra inference overhead.
format Preprint
id arxiv_https___arxiv_org_abs_2401_14686
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SSR: SAM is a Strong Regularizer for domain adaptive semantic segmentation
Ge, Yanqi
Huang, Ye
Li, Wen
Duan, Lixin
Computer Vision and Pattern Recognition
We introduced SSR, which utilizes SAM (segment-anything) as a strong regularizer during training, to greatly enhance the robustness of the image encoder for handling various domains. Specifically, given the fact that SAM is pre-trained with a large number of images over the internet, which cover a diverse variety of domains, the feature encoding extracted by the SAM is obviously less dependent on specific domains when compared to the traditional ImageNet pre-trained image encoder. Meanwhile, the ImageNet pre-trained image encoder is still a mature choice of backbone for the semantic segmentation task, especially when the SAM is category-irrelevant. As a result, our SSR provides a simple yet highly effective design. It uses the ImageNet pre-trained image encoder as the backbone, and the intermediate feature of each stage (ie there are 4 stages in MiT-B5) is regularized by SAM during training. After extensive experimentation on GTA5$\rightarrow$Cityscapes, our SSR significantly improved performance over the baseline without introducing any extra inference overhead.
title SSR: SAM is a Strong Regularizer for domain adaptive semantic segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2401.14686