arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.08960cs.CVcs.AI

基于非配对掩码-文本监督的开放词汇分割

Open-Vocabulary Segmentation with Unpaired Mask-Text Supervision

Sydney AI Centre, The University of Sydney(悉尼大学悉尼人工智能中心);Melbourne Centre for Data Science, The University of Melbourne(墨尔本大学墨尔本数据科学中心);AI2Robotics

Zhaoqing Wang, Xiaobo Xia, Ziye Chen, Xiao He, Yandong Guo, Mingming Gong, Tongliang Liu

AI总结:

本文提出Unpair-Seg弱监督开放词汇分割框架,利用非配对图像-掩码和图像-文本数据,结合视觉-语言大模型与多尺度匹配策略缓解噪声配对,在多个数据集上显著缩小与全监督方法的性能差距。

摘要:

当前最先进的开放词汇分割方法通常依赖图像-掩码-文本三元组标注进行监督。然而,获取如此详细的标注劳动密集,并且在复杂的现实场景中面临可扩展性挑战。虽然现有的弱监督方法利用图像-文本对来降低高昂的标注成本,但缺乏掩码监督使得模型难以定位多个实例并准确分组具有相似语义的像素,显著限制了通用性和性能。在本文中,我们提出了Unpair-Seg,一种新颖的弱监督开放词汇分割框架,该框架从非配对的图像-掩码和图像-文本对中学习,这些数据可以独立且高效地收集。Unpair-Seg首先预测一组二值掩码,并通过识别置信的掩码与文本实体对来生成伪标签。然后,我们训练一个特征适配器,基于这些伪标签将区域嵌入与文本嵌入对齐,从而实现开放词汇分割。然而,掩码-实体对应关系中固有的噪声给获得可靠配对带来了挑战。为了解决这个问题,我们采用视觉-语言大模型对输入图像进行重新描述并提取精确实体,同时设计了一种多尺度匹配策略以减少噪声掩码-实体对。我们的Unpair-Seg框架展示了出色的性能,在ADE-847和PASCAL Context-459数据集上分别达到了14.6%和19.5%的mIoU,显著缩小了全监督方法与弱监督方法之间的差距。

英文摘要:

Current state-of-the-art open-vocabulary segmentation methods typically rely on image-mask-text triplet annotations for supervision. However, acquiring such detailed annotations is labour-intensive and poses scalability challenges in complex real-world scenarios. While existing weakly-supervised approaches leverage image-text pairs to reduce the expansive annotation cost, the lack of mask supervision makes it difficult for the model to locate multiple instances and accurately group pixels with similar semantics, significantly hampering versatility and performance. In this paper, we introduce Unpair-Seg, a novel weakly-supervised open-vocabulary segmentation framework that learns from unpaired image-mask and image-text pairs, which can be independently and efficiently collected. Unpair-Seg initially predicts a set of binary masks and generates pseudo labels by identifying confident pairs of masks and text entities. We then train a feature adapter to align region embeddings with text embeddings based on these pseudo labels, achieving open-vocabulary segmentation. However, the inherent noise in the mask-entity correspondence poses a challenge to obtaining reliable pairs. To address this, we employ a vision-language large model to re-caption the input images and extract precise entities, and we design a multi-scale matching strategy to reduce noisy mask-entity pairs. Our Unpair-Seg framework demonstrates impressive performance, achieving 14.6\% and 19.5\% mIoU on the ADE-847 and PASCAL Context-459 datasets, significantly narrowing the gap between fully-supervised and weakly-supervised methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑