Overcoming Domain Limitations in Open-vocabulary Segmentation

• Published in: arXiv.org 2024-10
• Main Authors: Hwang, Dongjun, Oh, Seong Joon, Choe, Junsuk
• Format: Article
• Language: English
• Subjects: Datasets; Decoders; Domains; Performance enhancement; Segmentation; Source code
• ISSN: 2331-8422

Open-vocabulary segmentation (OVS) has gained attention for its ability to recognize a broader range of classes. However, OVS models show significant performance drops when applied to unseen domains beyond the previous training dataset. Fine-tuning these models on new datasets can improve performance, but often leads to the catastrophic forgetting of previously learned knowledge. To address this issue, we propose a method that allows OVS models to learn information from new domains while preserving prior knowledge. Our approach begins by evaluating the input sample's proximity to multiple domains, using precomputed multivariate normal distributions for each domain. Based on this prediction, we dynamically interpolate between the weights of the pre-trained decoder and the fine-tuned decoders. Extensive experiments demonstrate that this approach allows OVS models to adapt to new domains while maintaining performance on the previous training dataset. The source code is available at https://github.com/dongjunhwang/dwi.