4 个月前

基于判别增强的全卷积自编码器图像聚类

基于判别增强的全卷积自编码器图像聚类

摘要

传统的图像聚类方法通常采用两步法,即特征学习和聚类,这两个步骤依次进行。然而,近期的研究结果表明,将这两个分离的阶段结合在一个统一的框架中并进行联合训练可以取得更好的性能。在本文中,我们首先介绍了用于图像特征学习的全卷积自编码器(fully convolutional auto-encoders),然后提出了一种基于全卷积自编码器和软$k$-均值得分的统一聚类框架,以联合学习图像表示和聚类中心。在学习过程的初期阶段,从自编码器提取的表示可能对后续的聚类不够区分性强。为了解决这一问题,我们采用了增强型判别分布(boosted discriminative distribution),其中高得分分配被突出显示,而低得分分配则被弱化。随着判别能力的逐步增强,聚类分配得分变得更加区分性,并且聚类纯度得以提高。我们在多个视觉基准数据集上的实验结果表明,所提出的方法能够达到当前最先进的性能。

代码仓库

waynezhanghk/gacluster
pytorch
GitHub 中提及

基准测试

基准方法指标
image-clustering-on-coil-100DBC
Accuracy: 0.775
NMI: 0.905
image-clustering-on-coil-20DBC
Accuracy: 0.793
NMI: 0.895
image-clustering-on-mnist-fullDBC
Accuracy: 0.976
NMI: 0.937
image-clustering-on-uspsDBC
Accuracy: 0.743
NMI: 0.724

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于判别增强的全卷积自编码器图像聚类 | 论文 | HyperAI超神经