4 个月前

深度变换不变聚类

深度变换不变聚类

摘要

近期在图像聚类领域的进展通常集中在学习更好的深度表示上。相比之下,我们提出了一种正交方法,该方法不依赖抽象特征,而是学习预测图像变换,并直接在图像空间中进行聚类。这一学习过程自然地融入了K均值和高斯混合模型的基于梯度的训练中,无需任何额外的损失函数或超参数。这引导我们开发出两种新的深度变换不变聚类框架,它们联合学习原型和变换。具体而言,我们使用深度学习模块来解决对空间、颜色和形态变换的不变性问题。我们的方法概念简单,并具有多项优势,包括可以轻松适应任务所需的不变性以及聚类中心和聚类分配的高度可解释性。我们在标准图像聚类基准测试中展示了该新方法具有竞争力且前景广阔的结果。最后,通过可视化真实照片集合上的聚类结果,我们展示了其鲁棒性和改进后的可解释性的优势。

代码仓库

monniert/dti-clustering
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-clustering-on-fashion-mnistDTI-Clustering
Accuracy: 0.612
NMI: 0.637
image-clustering-on-mnist-fullDTI-Clustering
Accuracy: 0.979
NMI: 0.942
image-clustering-on-mnist-testDTI-Clustering
Accuracy: 0.978
NMI: 0.947
image-clustering-on-uspsDTI-Clustering
Accuracy: 0.864
NMI: 0.882
unsupervised-image-classification-on-mnistDTI-Clustering
Accuracy: 97.3
unsupervised-image-classification-on-svhnDTI-Clustering
# of clusters (k): 10
Acc: 57.4

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
深度变换不变聚类 | 论文 | HyperAI超神经