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

多标签聚类判别用于视觉表示学习

Multi-label Cluster Discrimination for Visual Representation Learning

DeepGlint(旷视科技);Huawei Noah’s Ark Lab(华为诺亚方舟实验室)

Xiang An, Kaicheng Yang, Xiangzi Dai, Ziyong Feng, Jiankang Deng

AI总结:

提出多标签聚类判别方法MLCD,通过多中心伪标签和分离正负类损失,增强视觉表示学习,在多个下游任务上达到最先进性能。

摘要:

对比语言-图像预训练(CLIP)近年来在各种任务上取得了成功,这得益于图像-文本对比学习所赋予的优越特征表示。然而,CLIP 使用的实例判别方法难以编码训练数据的语义结构。为了解决这一局限,研究者提出了通过迭代聚类分配和分类的聚类判别方法。然而,大多数聚类判别方法仅为每张图像定义一个单一伪标签,忽略了图像中的多标签信号。在本文中,我们提出了一种新颖的多标签聚类判别方法,命名为 MLCD,以增强表示学习。在聚类步骤中,我们首先基于现成的嵌入特征将大规模 LAION-400M 数据集聚类为一百万个中心。考虑到自然图像经常包含多个视觉对象或属性,我们选择多个最近的中心作为辅助类别标签。在判别步骤中,我们设计了一种新颖的多标签分类损失,该损失优雅地将正类别和负类别的损失分离,并减轻了决策边界上的模糊性。我们通过在不同规模的模型和预训练数据集上的实验验证了所提出的多标签聚类判别方法。实验结果表明,我们的方法在多个下游任务上取得了最先进的性能,包括线性探针、零样本分类和图像-文本检索。代码和模型已在 https://github.com/deepglint/unicom 发布。

英文摘要:

Contrastive Language Image Pre-training (CLIP) has recently demonstrated success across various tasks due to superior feature representation empowered by image-text contrastive learning. However, the instance discrimination method used by CLIP can hardly encode the semantic structure of training data. To handle this limitation, cluster discrimination has been proposed through iterative cluster assignment and classification. Nevertheless, most cluster discrimination approaches only define a single pseudo-label for each image, neglecting multi-label signals in the image. In this paper, we propose a novel Multi-Label Cluster Discrimination method named MLCD to enhance representation learning. In the clustering step, we first cluster the large-scale LAION-400M dataset into one million centers based on off-the-shelf embedding features. Considering that natural images frequently contain multiple visual objects or attributes, we select the multiple closest centers as auxiliary class labels. In the discrimination step, we design a novel multi-label classification loss, which elegantly separates losses from positive classes and negative classes, and alleviates ambiguity on decision boundary. We validate the proposed multi-label cluster discrimination method with experiments on different scales of models and pre-training datasets. Experimental results show that our method achieves state-of-the-art performance on multiple downstream tasks including linear probe, zero-shot classification, and image-text retrieval. Code and models have been released at https://github.com/deepglint/unicom .

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

↑