arXiv:2502.04226cs.CVcs.LG2025-02被引 1

不用文本编码器,仅用视觉模型+小头实现高性能图像聚类

Keep It Light! Simplifying Image Clustering Via Text-Free Adapters

论文配图:Keep It Light! Simplifying Image Clustering Via Text-Free Adapters
图 1 · 摘自论文原文
  • 只训练小型聚类头,利用预训练视觉特征和正样本对
  • 在多个数据集上达到与复杂方法相当的聚类效果
  • 适合无文本数据、追求轻量部署的场景

在预训练模型时代,简单线性探测或轻量读出层即可实现有效分类。然而,当前许多先进聚类方法采用多模态设计,依赖大语言模型或文本编码器,以及图文配对数据,这些在真实下游应用中往往不可用。此外,这类框架通常训练复杂、计算开销大,难以广泛推广。本文提出仅需训练小型聚类头的简化方法SCP(Simple Clustering via Pre-trained models),利用预训练视觉模型特征和正样本对,在CIFAR-10、CIFAR-20、CIFAR-100、STL-10、ImageNet-10和ImageNet-Dogs等基准数据集上实现了与现有先进方法相媲美的聚类性能。同时,我们给出一个理论结果,说明在理想条件下,额外的文本嵌入可能并非实现强聚类性能所必需。

原文摘要 · Abstract (English)

In the era of pre-trained models, effective classification can often be achieved using simple linear probing or lightweight readout layers. In contrast, many competitive clustering pipelines have a multi-modal design, leveraging large language models (LLMs) or other text encoders, and text-image pairs, which are often unavailable in real-world downstream applications. Additionally, such frameworks are generally complicated to train and require substantial computational resources, making widespread adoption challenging. In this work, we show that in deep clustering, competitive performance with more complex state-of-the-art methods can be achieved using a text-free and highly simplified training pipeline. In particular, our approach, Simple Clustering via Pre-trained models (SCP), trains only a small cluster head while leveraging pre-trained vision model feature representations and positive data pairs. Experiments on benchmark datasets, including CIFAR-10, CIFAR-20, CIFAR-100, STL-10, ImageNet-10, and ImageNet-Dogs, demonstrate that SCP achieves highly competitive performance. Furthermore, we provide a theoretical result explaining why, at least under ideal conditions, additional text-based embeddings may not be necessary to achieve strong clustering performance in vision.

图像聚类轻量模型视觉预训练无文本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。