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

用于无监督物体发现的弱监督对比学习

Weakly-supervised Contrastive Learning for Unsupervised Object Discovery

Northwestern Polytechnical University(西北工业大学);Australian National University(澳大利亚国立大学)

Yunqiu Lv, Jing Zhang, Nick Barnes, Yuchao Dai

AI总结:

本文针对无监督物体发现任务,提出通过弱监督对比学习微调DINO特征编码器以增强语义特征,并结合主成分分析定位物体区域,在基准数据集上验证了有效性。

摘要:

无监督物体发现(UOD)是指在不需要依赖标注数据集的情况下,从场景背景中区分出物体的完整区域的任务,这有助于边界框级定位和像素级分割任务。由于该任务能够以通用方式发现物体,因此具有广阔前景。我们将现有技术大致分为两个主要方向,即基于图像重合成的生成式解决方案,以及基于自监督模型的聚类方法。我们观察到,前者在很大程度上依赖于图像重建的质量,而后者在有效建模语义相关性方面存在局限性。为了直接针对物体发现任务,我们聚焦于后一种方法,并提出一种新颖的解决方案,通过引入弱监督对比学习(WCL)来增强语义信息探索。我们设计了一个语义引导的自监督学习模型,从图像中提取高层语义特征,该模型通过WCL对自监督模型DINO的特征编码器进行微调来实现。随后,我们引入主成分分析(PCA)来定位物体区域。与最大特征值对应的主投影方向可作为物体区域的指示器。在基准无监督物体发现数据集上的大量实验证明了我们所提解决方案的有效性。源代码和实验结果可通过我们的项目主页 https://github.com/npucvr/WSCUOD.git 公开获取。

英文摘要:

Unsupervised object discovery (UOD) refers to the task of discriminating the whole region of objects from the background within a scene without relying on labeled datasets, which benefits the task of bounding-box-level localization and pixel-level segmentation. This task is promising due to its ability to discover objects in a generic manner. We roughly categorise existing techniques into two main directions, namely the generative solutions based on image resynthesis, and the clustering methods based on self-supervised models. We have observed that the former heavily relies on the quality of image reconstruction, while the latter shows limitations in effectively modeling semantic correlations. To directly target at object discovery, we focus on the latter approach and propose a novel solution by incorporating weakly-supervised contrastive learning (WCL) to enhance semantic information exploration. We design a semantic-guided self-supervised learning model to extract high-level semantic features from images, which is achieved by fine-tuning the feature encoder of a self-supervised model, namely DINO, via WCL. Subsequently, we introduce Principal Component Analysis (PCA) to localize object regions. The principal projection direction, corresponding to the maximal eigenvalue, serves as an indicator of the object region(s). Extensive experiments on benchmark unsupervised object discovery datasets demonstrate the effectiveness of our proposed solution. The source code and experimental results are publicly available via our project page at https://github.com/npucvr/WSCUOD.git.

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

↑