当前位置:首页 > 文章列表 > 科技周边 > 人工智能 > ICML 2024 Oral|外部引导的深度聚类新范式

ICML 2024 Oral|外部引导的深度聚类新范式

来源:机器之心 2024-06-06 22:36:15 0浏览 收藏

编程并不是一个机械性的工作,而是需要有思考,有创新的工作,语法是固定的,但解决问题的思路则是依靠人的思维,这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《ICML 2024 Oral|外部引导的深度聚类新范式》,文章讲解的知识点主要包括,如果你对科技周边方面的知识点感兴趣,就不要错过golang学习网,在这可以对大家的知识积累有所帮助,助力开发能力的提升。

ICML 2024 Oral|外部引导的深度聚类新范式
AIxiv专栏是本站发布学术、技术内容的栏目。过去数年,本站AIxiv专栏接收报道了2000多篇内容,覆盖全球各大高校与企业的顶级实验室,有效促进了学术交流与传播。如果您有优秀的工作想要分享,欢迎投稿或者联系报道。投稿邮箱:liyazhou@jiqizhixin.com;zhaoyunfeng@jiqizhixin.com

本文作者李云帆,四川大学计算机学院 2020级直博研究生。在导师彭玺教授的指导下,博士期间主要围绕深度聚类开展理论、方法和应用的研究。目前已在国际权威刊物Nature Communications/JMLR/TPAMI/IJCV/ICML/CVPR等上发表学术论文13篇,谷歌学术引用共954次;发表于AAAI2021的Contrastive Clustering被引516次,是2021年以来聚类领域引用最高的论文;获首批国家自然科学基金青年学生基础研究项目(博士研究生)资助。

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

ICML 2024 Oral|外部引导的深度聚类新范式

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

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

ICML 2024 Oral|外部引导的深度聚类新范式

  • 论文题目:Image Clustering with External Guidance
  • 论文地址:https://arxiv.org/abs/2310.11989
  • 代码地址:https://github.com/XLearning-SCU/2024-ICML-TAC

背景

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

ICML 2024 Oral|外部引导的深度聚类新范式

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

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

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

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

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

方法

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

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

  • 如何构建图像的文本表征;
  • 如何协同图像和文本进行聚类。

ICML 2024 Oral|外部引导的深度聚类新范式

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

一、文本表征构建

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

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

考虑到过多的语义中心会关注过于细粒度的特征,不利于区分不同类别的图像,而过少的语义中心则会难以准确覆盖位于聚类边界图像的语义,本文提出根据样本点的个数估计k-means算法中合适的k值(实验中选取k=N/300,N为图像个数),并计算图像语义中心如下:ICML 2024 Oral|外部引导的深度聚类新范式
其中ICML 2024 Oral|外部引导的深度聚类新范式当且仅当图像ICML 2024 Oral|外部引导的深度聚类新范式属于第l个聚类,ICML 2024 Oral|外部引导的深度聚类新范式表示第i张图像经过CLIP图像编码器后得到的表征。在得到图像语义中心后,为了选取具有代表性的名词集合,与常见的CLIP Zero-shot分类相反,本文将所有WordNet中的名词划分到k个图像语义中心,其中第i个名词属于第l和语义中心的概率为:ICML 2024 Oral|外部引导的深度聚类新范式
其中sim表示余弦相似性,ICML 2024 Oral|外部引导的深度聚类新范式表示第i个名词经过预训练好的文本编码器后得到的表征。保留每个语义中心对应概率最高的名词,作为组成文本空间的候选词。

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

ICML 2024 Oral|外部引导的深度聚类新范式

其中,ICML 2024 Oral|外部引导的深度聚类新范式表示第i张图像对应的文本模态中的表征,ICML 2024 Oral|外部引导的深度聚类新范式表示组成文本空间的第j个候选名词,ICML 2024 Oral|外部引导的深度聚类新范式控制检索的平滑程度。

至此,作者为每张图像构建出了其在文本模态中的表征。此时可通过在文本和图像的拼接表征ICML 2024 Oral|外部引导的深度聚类新范式直接使用经典k-means聚类方法来实现图像聚类。

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

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

二、跨模态互蒸馏

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

具体地,为每张图像构建邻居集合ICML 2024 Oral|外部引导的深度聚类新范式,并引入一个聚类网络f对每个图像表征做出聚类指派,在每次迭代中,计算所有图像和其邻居集合中随机的一个图像的聚类指派,记为:

ICML 2024 Oral|外部引导的深度聚类新范式其中ICML 2024 Oral|外部引导的深度聚类新范式ICML 2024 Oral|外部引导的深度聚类新范式分别对应图像i及其邻居的聚类指派,P和ICML 2024 Oral|外部引导的深度聚类新范式均为n*K的矩阵,其中K表示目标聚类个数。

相类似的,引入另一个聚类网络g来对每个文本表征做出聚类指派,同样为每个文本表征构建邻居集合ICML 2024 Oral|外部引导的深度聚类新范式,在每次迭代中,计算所有文本和其邻居集合中随机的一个文本的聚类指派,记为:
ICML 2024 Oral|外部引导的深度聚类新范式其中ICML 2024 Oral|外部引导的深度聚类新范式ICML 2024 Oral|外部引导的深度聚类新范式分别对应文本i及其邻居的聚类指派,Q和Q^N同样均为n*K的矩阵。

为了协同图像和文本两个模态,要求网络对于图像和其对应文本模态的邻居具有类似的聚类指派,同时对于文本和其对应图像模态的邻居也具有类似的聚类指派。为实现该目标,本文设计了如下的损失函数:
ICML 2024 Oral|外部引导的深度聚类新范式
其中ICML 2024 Oral|外部引导的深度聚类新范式分别表示聚类指派矩阵P,P^N,Q,Q^N的第i列,ICML 2024 Oral|外部引导的深度聚类新范式为温度系数。该损失函数一方面能通过跨模态邻居之间的聚类指派一致性实现图文模态的协同,另一方面能扩大不同的类簇之间的差异性。

此外,为了使训练过程更加稳定,本文设计了另外两个正则项损失函数。首先,为了鼓励模型做出更加置信的聚类指派,提出如下损失函数:ICML 2024 Oral|外部引导的深度聚类新范式该损失函数在ICML 2024 Oral|外部引导的深度聚类新范式ICML 2024 Oral|外部引导的深度聚类新范式均为独热(One-hot)编码时被最小化,因此能提升聚类指派的置信度。另外,为了防止模型将大量图像和文本都分配到个别类簇中,提出了以下损失函数:
ICML 2024 Oral|外部引导的深度聚类新范式
其中ICML 2024 Oral|外部引导的深度聚类新范式ICML 2024 Oral|外部引导的深度聚类新范式分别表示图像和文本模态中整体的聚类分布。

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

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

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

实验

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

ICML 2024 Oral|外部引导的深度聚类新范式

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

ICML 2024 Oral|外部引导的深度聚类新范式

                               表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.

今天带大家了解了的相关知识,希望对你有所帮助;关于科技周边的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~

版本声明
本文转载于:机器之心 如有侵犯,请联系study_golang@163.com删除
辅导作业这么费劲的事,还是交给这个大模型吧辅导作业这么费劲的事,还是交给这个大模型吧
上一篇
辅导作业这么费劲的事,还是交给这个大模型吧
Stability AI开源47秒音频生成模型,虫鸣鸟叫、摇滚、鼓点都能生成
下一篇
Stability AI开源47秒音频生成模型,虫鸣鸟叫、摇滚、鼓点都能生成
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    542次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    508次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    497次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    484次学习
查看更多
AI推荐
  • AI Make Song:零门槛AI音乐创作平台,助你轻松制作个性化音乐
    AI Make Song
    AI Make Song是一款革命性的AI音乐生成平台,提供文本和歌词转音乐的双模式输入,支持多语言及商业友好版权体系。无论你是音乐爱好者、内容创作者还是广告从业者,都能在这里实现“用文字创造音乐”的梦想。平台已生成超百万首原创音乐,覆盖全球20个国家,用户满意度高达95%。
    18次使用
  • SongGenerator.io:零门槛AI音乐生成器,快速创作高质量音乐
    SongGenerator
    探索SongGenerator.io,零门槛、全免费的AI音乐生成器。无需注册,通过简单文本输入即可生成多风格音乐,适用于内容创作者、音乐爱好者和教育工作者。日均生成量超10万次,全球50国家用户信赖。
    14次使用
  •  BeArt AI换脸:免费在线工具,轻松实现照片、视频、GIF换脸
    BeArt AI换脸
    探索BeArt AI换脸工具,免费在线使用,无需下载软件,即可对照片、视频和GIF进行高质量换脸。体验快速、流畅、无水印的换脸效果,适用于娱乐创作、影视制作、广告营销等多种场景。
    14次使用
  • SEO标题协启动:AI驱动的智能对话与内容生成平台 - 提升创作效率
    协启动
    SEO摘要协启动(XieQiDong Chatbot)是由深圳协启动传媒有限公司运营的AI智能服务平台,提供多模型支持的对话服务、文档处理和图像生成工具,旨在提升用户内容创作与信息处理效率。平台支持订阅制付费,适合个人及企业用户,满足日常聊天、文案生成、学习辅助等需求。
    17次使用
  • Brev AI:零注册门槛的全功能免费AI音乐创作平台
    Brev AI
    探索Brev AI,一个无需注册即可免费使用的AI音乐创作平台,提供多功能工具如音乐生成、去人声、歌词创作等,适用于内容创作、商业配乐和个人创作,满足您的音乐需求。
    19次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码