SED: A Simple Encoder-Decoder for Open-Vocabulary Semantic SegmentationShow others and affiliations
2024 (English)In: 2024 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2024, IEEE COMPUTER SOC , 2024, p. 3426-3436Conference paper, Published paper (Refereed)
Abstract [en]
Open-vocabulary semantic segmentation strives to distinguish pixels into different semantic groups from an open set of categories. Most existing methods explore utilizing pre-trained vision-language models, in which the key is to adapt the image-level model for pixel-level segmentation task. In this paper, we propose a simple encoder-decoder, named SED, for open-vocabulary semantic segmentation, which comprises a hierarchical encoder-based cost map generation and a gradual fusion decoder with category early rejection. The hierarchical encoder-based cost map generation employs hierarchical backbone, instead of plain transformer, to predict pixel-level image-text cost map. Compared to plain transformer, hierarchical backbone better captures local spatial information and has linear computational complexity with respect to input size. Our gradual fusion decoder employs a top-down structure to combine cost map and the feature maps of different backbone levels for segmentation. To accelerate inference speed, we introduce a category early rejection scheme in the decoder that rejects many no-existing categories at the early layer of decoder, resulting in at most 4.7 times acceleration without accuracy degradation. Experiments are performed on multiple open-vocabulary semantic segmentation datasets, which demonstrates the efficacy of our SED method. When using ConvNeXt-B, our SED method achieves mIoU score of 31.6% on ADE20K with 150 categories at 82 millisecond (ms) per image on a single A6000. Our source code is available at https://github.com/xb534/SED.
Place, publisher, year, edition, pages
IEEE COMPUTER SOC , 2024. p. 3426-3436
Series
IEEE Conference on Computer Vision and Pattern Recognition, ISSN 1063-6919
National Category
Computer graphics and computer vision
Identifiers
URN: urn:nbn:se:liu:diva-210824DOI: 10.1109/CVPR52733.2024.00329ISI: 001322555903077ISBN: 9798350353013 (print)ISBN: 9798350353006 (electronic)OAI: oai:DiVA.org:liu-210824DiVA, id: diva2:1927377
Conference
IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, jun 16-22, 2024
Note
Funding Agencies|National Key Research and Development Program of China [2022ZD0160400]; Natural Science Foundation of China [62271346, 62206031]
2025-01-142025-01-142025-02-07