用大模型生成更丰富的描述词,提升未知物体的语义分割效果
LMSeg: Unleashing the Power of Large-Scale Models for Open-Vocabulary Semantic Segmentation
- 用大语言模型生成带颜色、形状等属性的详细描述词
- 融合SAM与CLIP,增强像素级视觉特征提取能力
- 在多个公开数据集上达到当前最佳性能,适合开放词汇分割研究者
当前开放词汇语义分割方法依赖视觉-语言模型(如CLIP)将图像特征与预训练获得的语义特征对齐,但其使用固定模板的简短文本提示,难以捕捉物体的完整属性。此外,CLIP虽擅长图像级特征,但在像素级表示上表现不足,限制了分割精度。本文提出LMSeg,通过多大规模模型协同改进:利用大语言模型(LLMs)为每类生成包含颜色、形状/大小、纹理/材质等多样属性的丰富语言提示;同时采用SAM作为补充,通过可学习加权融合策略增强视觉编码器的像素级特征提取能力。基于此,LMSeg在所有主流开放词汇分割基准上均取得当前最优结果。
原文摘要 · Abstract (English)
It is widely agreed that open-vocabulary-based approaches outperform classical closed-set training solutions for recognizing unseen objects in images for semantic segmentation. Existing open-vocabulary approaches leverage vision-language models, such as CLIP, to align visual features with rich semantic features acquired through pre-training on large-scale vision-language datasets. However, the text prompts employed in these methods are short phrases based on fixed templates, failing to capture comprehensive object attributes. Moreover, while the CLIP model excels at exploiting image-level features, it is less effective at pixel-level representation, which is crucial for semantic segmentation tasks. In this work, we propose to alleviate the above-mentioned issues by leveraging multiple large-scale models to enhance the alignment between fine-grained visual features and enriched linguistic features. Specifically, our method employs large language models (LLMs) to generate enriched language prompts with diverse visual attributes for each category, including color, shape/size, and texture/material. Additionally, for enhanced visual feature extraction, the SAM model is adopted as a supplement to the CLIP visual encoder through a proposed learnable weighted fusion strategy. Built upon these techniques, our method, termed LMSeg, achieves state-of-the-art performance across all major open-vocabulary segmentation benchmarks. The code will be made available soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。