中文

利用空间变换层改进深度图像聚类

计算机视觉与模式识别 2019-10-25 v2 机器学习 机器学习

摘要

图像聚类是机器学习中重要但具挑战性的任务。如同大多数图像处理领域,最新改进来自基于深度学习方法的模型。然而,经典深度学习方法难以处理尺度与旋转等空间图像变换。本文中,我们提出使用视觉注意力技术来缓解图像聚类方法中的该问题。我们评估了称为深度自适应聚类(DAC)的深度图像聚类模型与空间变换网络(STN)的组合。所提模型在 MNIST 与 FashionMNIST 数据集上评估,并优于基线模型。

关键词

引用

@article{arxiv.1902.05401,
  title  = {Improving Deep Image Clustering With Spatial Transformer Layers},
  author = {Thiago V. M. Souza and Cleber Zanchettin},
  journal= {arXiv preprint arXiv:1902.05401},
  year   = {2019}
}