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

基于自监督编码器的未见数据集聚类实证研究

An Empirical Study into Clustering of Unseen Datasets with Self-Supervised Encoders

Vector Institute(向量研究所);Aalborg University(奥尔堡大学);Pioneer Centre for AI(人工智能先锋中心);Dalhousie University(达尔豪斯大学);University of Guelph(圭尔夫大学)

Scott C. Lowe, Joakim Bruslund Haurum, Sageev Oore, Thomas B. Moeslund, Graham W. Taylor

AI总结:

该研究通过基准实验对比监督与自监督编码器在未见图像数据集上的聚类表现,发现微调SSL编码器的域内外性能变化,且UMAP降维下的轮廓系数可作为无标签数据聚类性能的代理指标。

摘要:

预训练模型能否在不进行任何重新训练的情况下泛化到新数据集?我们将预训练的图像模型部署在其未训练过的数据集上,研究它们的嵌入是否形成有意义的聚类。我们的基准实验套件使用仅在ImageNet-1k上预训练的编码器,这些编码器采用监督或自监督(SSL)训练技术,部署在训练过程中未见过的图像数据集上,并使用传统聚类算法进行聚类。该评估为自监督模型的嵌入提供了新见解,它们与监督模型优先考虑不同的特征。我们发现,在训练域内,监督编码器比SSL编码器提供更多效用,而在训练域外则相反。然而,对SSL编码器进行ImageNet-1k分类微调会产生相反的行为,在域内表现优于仅监督模型,而在远域外数据上表现下降——在远OOD上比仅SSL或仅监督模型表现更差。聚类提供了一种评估自监督学习表示效用的方法,该方法与现有特征质量估计方法正交。此外,我们发现在UMAP降维空间中测量的轮廓系数与聚类性能高度相关,因此可作为无真实标签数据的聚类性能的代理指标。我们的代码实现可在该https URL获取。

英文摘要:

Can pretrained models generalize to new datasets without any retraining? We deploy pretrained image models on datasets they were not trained for, and investigate whether their embeddings form meaningful clusters. Our suite of benchmarking experiments uses encoders pretrained solely on ImageNet-1k with either supervised or self-supervised training techniques, deployed on image datasets that were not seen during training, and clustered with conventional clustering algorithms. This evaluation provides new insights into the embeddings of self-supervised models, which prioritize different features to supervised models. We find evidence that supervised encoders offer more utility than SSL encoders within the training domain, and vice-versa far outside of it. However, fine-tuning SSL encoders for ImageNet-1k classification results in the opposite behaviour, with better performance than supervised-only models on in-domain and decreased performance on far out of domain data - worse at far-OOD than either SSL-only or supervised-only models. Clustering provides a way to evaluate the utility of self-supervised learnt representations orthogonal to existing feature quality estimation methods. Additionally, we find the silhouette score when measured in a UMAP-reduced space is highly correlated with clustering performance, and can therefore be used as a proxy for clustering performance on data with no ground truth labels. Our code implementation is available at https://github.com/scottclowe/zs-ssl-clustering/.

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

↑