arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2308.02487cs.CV

卷积不死:基于单一冻结卷积CLIP的开放词汇分割

Convolutions Die Hard: Open-Vocabulary Segmentation with Single Frozen Convolutional CLIP

ByteDance(字节跳动);The Johns Hopkins University(约翰霍普金斯大学)

Qihang Yu, Ju He, Xueqing Deng, Xiaohui Shen, Liang-Chieh Chen

AI总结:

提出单阶段FC-CLIP框架,利用冻结卷积CLIP骨干同时进行掩码生成与开放词汇分类,简化两阶段流程,在多个数据集上以更少参数和更快速度取得更优分割性能。

摘要:

开放词汇分割是一项具有挑战性的任务,需要从开放类别集合中分割并识别对象。解决该挑战的一种方式是利用多模态模型(如CLIP)在共享嵌入空间中提供图像和文本特征,从而弥合封闭词汇与开放词汇识别之间的差距。因此,现有方法通常采用两阶段框架来处理该问题,输入首先经过掩码生成器,然后与预测掩码一起经过CLIP模型。该过程涉及多次从图像中提取特征,这可能低效且不经济。相比之下,我们提出将所有内容构建到使用共享冻结卷积CLIP骨干网络的单阶段框架中,这不仅显著简化了当前的两阶段流程,而且 remarkably 产生了更好的精度-成本权衡。所提出的FC-CLIP受益于以下观察:冻结的CLIP骨干网络保持了开放词汇分类的能力,同时也可以作为强大的掩码生成器,并且卷积CLIP能够很好地泛化到比对比图像-文本预训练时使用的更大输入分辨率。仅在COCO全景数据上训练并以零样本方式测试时,FC-CLIP在ADE20K上达到26.8 PQ、16.8 AP和34.1 mIoU,在Mapillary Vistas上达到18.2 PQ和27.9 mIoU,在Cityscapes上达到44.0 PQ、26.8 AP和56.2 mIoU,分别比先前最优方法在ADE20K上高出+4.2 PQ、+2.4 AP、+4.2 mIoU,在Mapillary Vistas上高出+4.0 PQ,在Cityscapes上高出+20.1 PQ。此外,FC-CLIP的训练和测试时间分别比先前最优方法显著快7.5倍和6.6倍,同时使用的参数少5.9倍。FC-CLIP还在各种开放词汇语义分割数据集上设置了新的最先进性能。代码见https://github.com/bytedance/fc-clip

英文摘要:

Open-vocabulary segmentation is a challenging task requiring segmenting and recognizing objects from an open set of categories. One way to address this challenge is to leverage multi-modal models, such as CLIP, to provide image and text features in a shared embedding space, which bridges the gap between closed-vocabulary and open-vocabulary recognition. Hence, existing methods often adopt a two-stage framework to tackle the problem, where the inputs first go through a mask generator and then through the CLIP model along with the predicted masks. This process involves extracting features from images multiple times, which can be ineffective and inefficient. By contrast, we propose to build everything into a single-stage framework using a shared Frozen Convolutional CLIP backbone, which not only significantly simplifies the current two-stage pipeline, but also remarkably yields a better accuracy-cost trade-off. The proposed FC-CLIP, benefits from the following observations: the frozen CLIP backbone maintains the ability of open-vocabulary classification and can also serve as a strong mask generator, and the convolutional CLIP generalizes well to a larger input resolution than the one used during contrastive image-text pretraining. When training on COCO panoptic data only and testing in a zero-shot manner, FC-CLIP achieve 26.8 PQ, 16.8 AP, and 34.1 mIoU on ADE20K, 18.2 PQ, 27.9 mIoU on Mapillary Vistas, 44.0 PQ, 26.8 AP, 56.2 mIoU on Cityscapes, outperforming the prior art by +4.2 PQ, +2.4 AP, +4.2 mIoU on ADE20K, +4.0 PQ on Mapillary Vistas and +20.1 PQ on Cityscapes, respectively. Additionally, the training and testing time of FC-CLIP is 7.5x and 6.6x significantly faster than the same prior art, while using 5.9x fewer parameters. FC-CLIP also sets a new state-of-the-art performance across various open-vocabulary semantic segmentation datasets. Code at https://github.com/bytedance/fc-clip

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

↑