ICML 2024 | 川大提出外部引导的深度聚类新范式，大幅提升CLIP图像聚类性能

©作者 | 李云帆

单位 | 四川大学博士生

研究方向 | 深度聚类

怎样才能将可爱又迷人的柯基与柴犬的图像进行区分？

如果单从图像识别角度而言，两种犬类在外观上极为相似，拥有相近的色块像素，仅凭数据内蕴信息（即图像自身）可能难以对二者进行区分，但如果借助外部数据和知识，情况可能会大幅改观。

近日，一篇以《Image Clustering with External Guidance》为题的论文，提出了能够大幅提升 CLIP 图像聚类性能，引入外部知识库内容辅助深度聚类的方法。

论文标题：

Image Clustering with External Guidance

论文链接：

https://arxiv.org/abs/2310.11989

代码链接：

https://github.com/XLearning-SCU/2024-ICML-TAC

居然要降低学习率？

作为机器学习的经典任务之一，图像聚类旨在无需依赖样本标注的情况下，将图像依据语义划分到不同的类簇中，其核心在于利用先验知识构建监督信号。从经典基于类簇紧致性的 k-means 到近年来基于增广不变性的对比聚类 [1]，聚类方法的发展本质上对应于监督信号的演进。

▲ 图1. 聚类方法的发展大致可分为三个阶段：1）传统聚类，基于数据分布假设设计聚类策略；2）深度聚类，利用深度神经网络提取有利于聚类的特征；3）自监督聚类，通过数据增广或动量网络等策略构建自监督信号。不同于此前的工作聚焦于从数据内部挖掘监督信号，本文提出利用外部知识来引导聚类，并将新范式归类为4）外部引导聚类。实验结果表明，通过利用文本模态中的语义信息，所提出的方法 (TAC) 显著提升了图像聚类精度。

现有的聚类研究虽然在方法设计上各不相同，但均是从数据内部挖掘监督信号，其性能最终会受限于数据自身所蕴含信息量的固有上限。举例来说，柯基和巴哥犬的图片有明显的差异，但其和柴犬在外观上十分相似，仅依据图像本身难以对二者进行区分。

但值得注意的是，在数据内蕴信息之外，现实世界中还存在着大量有助于聚类的外部知识，而在现有工作中被很大程度地忽略了。

在上述例子中，假设模型具备来自知识库的「柯基腿较短，而柴犬腿较长」等非图像域的外部先验，则能更准确地对二者的图像进行区分。

换而言之，与从数据中竭力地挖掘内部监督信号相比，利用更加丰富且容易获得的外部知识来引导聚类，有望起到事半功倍的效果。

方法

本文提出了一种简单而有效的外部引导聚类方法 TAC（Text-Aided Clustering，文本辅助的聚类），基于预训练 CLIP 模型，通过利用来自文本模态的外部知识辅助图像聚类。

在缺乏类别标注和图像描述等文本信息的情况下，利用文本语义辅助图像聚类面临两个挑战：

• 如何构建图像的文本表征；

• 如何协同图像和文本进行聚类。

▲ 图2. 所提出的TAC方法框架，包括文本表征构建和跨模态互蒸馏两部分

2.1 文本表征构建

针对第一个挑战，由于样本标注、图像描述等文本信息在图像聚类任务中不可用，本文提出将来自 WordNet [2] 的所有名词作为文本模态的候选词，选择其中具有代表性的名词集合来组成文本空间。

具体地，为了使文本表征精确地覆盖图像语义，同时尽可能在不同类别图像之间具有区分度，本文首先使用 k-means 算法来计算图像语义中心。

考虑到过多的语义中心会关注过于细粒度的特征，不利于区分不同类别的图像，而过少的语义中心则会难以准确覆盖位于聚类边界图像的语义，本文提出根据样本点的个数估计 k-means 算法中合适的 k 值（实验中选取 k=N/300，N 为图像个数），并计算图像语义中心如下：‍

其中 当且仅当图像 属于第 l 个聚类， 表示第 i 张图像经过 CLIP 图像编码器后得到的表征。在得到图像语义中心后，为了选取具有代表性的名词集合，与常见的 CLIP Zero-shot 分类相反，本文将所有 WordNet 中的名词划分到 k 个图像语义中心，其中第 i 个名词属于第 l 和语义中心的概率为：

其中 sim 表示余弦相似性， 表示第 i 个名词经过预训练好的文本编码器后得到的表征。保留每个语义中心对应概率最高的名词，作为组成文本空间的候选词。

选取完具有代表性的名词集合后，可通过为每张图像检索其最相关的名词来构建其文本模态的表征：

其中， 表示第 i 张图像对应的文本模态中的表征， 表示组成文本空间的第 j 个候选名词， 控制检索的平滑程度。

至此，作者为每张图像构建出了其在文本模态中的表征。此时可通过在文本和图像的拼接表征 直接使用经典 k-means 聚类方法来实现图像聚类。

由于融入了来自文本模态的紧凑语义，拼接后的表征具有更好的判别性，从而相较于直接在图像表征上使用 k-means 会得到更好的图像聚类结果。

值得注意的是，上述文本模态的构建过程不需要任何的额外训练和模型调优，其中名词选取和检索过程的计算开销几乎可以忽略不计。

2.2 跨模态互蒸馏

尽管直接将文本和图像表征进行拼接已能显著提升图像聚类效果，但是简单的拼接并不能充分协同文本和图像两个模态。因此，本文进一步提出跨模态互蒸馏方法，通过训练额外的聚类网络进一步提升聚类性能。

具体地，为每张图像构建邻居集合 ，并引入一个聚类网络对每个图像表征做出聚类指派，在每次迭代中，计算所有图像和其邻居集合中随机的一个图像的聚类指派，记为：

其中 和 分别对应图像 其邻居的聚类指派， 和 均为 的矩阵，其中 K 表示目标聚类个数。

相类似的，引入另一个聚类网络 g 来对每个文本表征做出聚类指派，同样为每个文本表征构建邻居集合 ，在每次迭代中，计算所有文本和其邻居集合中随机的一个文本的聚类指派，记为：

其中 和 分别对应文本 及其邻居的聚类指派， 和 同样均为 的矩阵。

为了协同图像和文本两个模态，要求网络对于图像和其对应文本模态的邻居具有类似的聚类指派，同时对于文本和其对应图像模态的邻居也具有类似的聚类指派。为实现该目标，本文设计了如下的损失函数：

其中 分别表示聚类指派矩阵 的第 i 列， 为温度系数。该损失函数一方面能通过跨模态邻居之间的聚类指派一致性实现图文模态的协同，另一方面能扩大不同的类簇之间的差异性。

此外，为了使训练过程更加稳定，本文设计了另外两个正则项损失函数。首先，为了鼓励模型做出更加置信的聚类指派，提出如下损失函数：

该损失函数在 和 均为独热（One-hot）编码时被最小化，因此能提升聚类指派的置信度。另外，为了防止模型将大量图像和文本都分配到个别类簇中，提出了以下损失函数：

其中 和 分别表示图像和文本模态中整体的聚类分布。

综合上述三个损失函数，本文使用如下损失函数来优化图像和文本模态的聚类网络 f 和 g：

其中 为权重参数。需要指出的是，上述损失函数只用来优化额外引入的聚类网络，并不修改 CLIP 预训练好的文本和图像编码器，因此其整体训练开销较小，实验表明所提出的方法在 CIFAR-10 的 6 万张图像上训练仅需使用 1 分钟。

训练完成后，只需将待聚类的图像输入聚类网络 f，即可得到其聚类指派，从而实现准确的图像聚类。

实验

本文在五个经典数据集和三个更具挑战性的图像聚类数据集上对方法进行了验证，部分实验结果如下： 

▲ 表1. 所提出的TAC方法在经典图像聚类数据集上的聚类性能

▲ 表2. 所提出的TAC方法在更具挑战性的图像聚类数据集上的聚类性能

从结果中可以看出，在缺少标注信息的情况下所提出的 TAC 方法通过为每个图像构建文本表征，能够有效地从文本模态中挖掘语义信息。在无需任何额外训练的情况下，TAC (no train) 显著提高了直接在 CLIP 提取的图像表征上使用 k-means 聚类的性能，特别是在更困难的数据集上。

当进一步使用提出的跨模态相互蒸馏策略训练聚类网络时，TAC 取得了最优的聚类性能，甚至超过了依赖类别标签信息的 CLIP Zero-shot 分类性能。

总结与展望

不同于现有的聚类研究聚焦于从数据内部构建监督信号，本文创新性地提出利用此前被忽略的外部知识来引导聚类。

所提出的 TAC 方法通过在无需文本描述的情况下，从预训练 CLIP 模型的文本模态挖掘语义信息，显著提升了图像聚类性能，证明了所提出的外部引导聚类新范式的有效性。

所提出的外部引导聚类范式的挑战在于：

• 如何选择合适的外部知识；

• 如何有效的整合外部知识以辅助聚类。

除了本工作关注的文本语义外，外部知识广泛存在于各类的数据、模型、知识库等，对于不同的数据类型和聚类目标，需要针对性地选择与利用外部知识。

总的来说，在目前大模型、知识库日趋成熟背景下，外部引导的聚类新范式具备良好的发展潜力，希望未来有更多工作进行相关的探索。

参考文献

[1] Li Y, Hu P, Liu Z, et al. Contrastive clustering[C]//Proceedings of the AAAI conference on artificial intelligence. 2021, 35(10): 8547-8555.

[2] Miller G A. WordNet: a lexical database for English[J]. Communications of the ACM, 1995, 38(11): 39-41.

更多阅读

#投 稿 通 道#

 让你的文字被更多人看到 

如何才能让更多的优质内容以更短路径到达读者群体，缩短读者寻找优质内容的成本呢？答案就是：你不认识的人。

总有一些你不认识的人，知道你想知道的东西。PaperWeekly 或许可以成为一座桥梁，促使不同背景、不同方向的学者和学术灵感相互碰撞，迸发出更多的可能性。 

PaperWeekly 鼓励高校实验室或个人，在我们的平台上分享各类优质内容，可以是最新论文解读，也可以是学术热点剖析、科研心得或竞赛经验讲解等。我们的目的只有一个，让知识真正流动起来。

📝 稿件基本要求：

• 文章确系个人原创作品，未曾在公开渠道发表，如为其他平台已发表或待发表的文章，请明确标注 

• 稿件建议以 markdown 格式撰写，文中配图以附件形式发送，要求图片清晰，无版权问题

• PaperWeekly 尊重原作者署名权，并将为每篇被采纳的原创首发稿件，提供业内具有竞争力稿酬，具体依据文章阅读量和文章质量阶梯制结算

📬 投稿通道：

• 投稿邮箱：[e-mail] 

• 来稿请备注即时联系方式（微信），以便我们在稿件选用的第一时间联系作者

• 您也可以直接添加小编微信（pwbot02）快速投稿，备注：姓名-投稿

△长按添加PaperWeekly小编

🔍

现在，在「知乎」也能找到我们了

进入知乎首页搜索「PaperWeekly」

点击「关注」订阅我们的专栏吧

·

·

·