深度超球面变分聚类网络
作者:
作者单位:

作者简介:

通讯作者:

基金项目:

国家自然科学基金青年科学基金项目(No.12101098);重庆市自然科学基金面上项目(No.cstc2021jcyj-msxmX1053);重庆市教育委员会科学技术研究计划重点项目(No.KJQN201800116)


Deep Hyperspherical Variational Clustering Network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    基于变分自动编码器的深度聚类方法因较好的生成能力和聚类性能而受到广泛关注。然而现有的基于变分自动编码器的聚类算法通常需处理具有挑战性的证据下界(Evidence lower bound, ELBO)来实现聚类。同时,这些算法需要先验知识来假设类别分布,大多数方法还依赖于高斯分布或高斯混合模型。为了克服这些问题,提出了一种基于超球面变分自动编码器的端到端深度聚类网络。该方法采用2个后验分布的KL散度代替了复杂的ELBO,并将von Mises-Fisher混合分布作为潜在空间嵌入的模型。此外它最大化了潜在表示和预测聚类分配之间的扩展互信息以实现更有区分性和平衡性的分配。通过与最先进的深度聚类技术在基准数据集上进行比较,验证了所提出的深度聚类方法的有效性。所建立的模型为复杂数据特征学习和聚类分析提供了一套新的方法,拓展了贝叶斯推断方法在深度聚类上的应用。

    Abstract:

    In recent years, the deep clustering approach based on variational autoencoders (VAE) has garnered significant attention due to its remarkable generative capabilities and clustering performance. However, existing VAE-based clustering algorithms typically revolve around dealing with the challenging Evidence Lower Bound (ELBO) to achieve clustering. Moreover, they often require prior knowledge of the class distribution, with most relying on Gaussian distributions or Gaussian mixture models. A deep clustering network based on Hyperspherical Variational Auto-Encoders (HVAE) is proposed, named DHVC (Deep Hyperspherical Variational Clustering Network). This approach substitutes the complex Evidence Lower Bound (ELBO) with the Kullback-Leibler Divergence of two posterior distributions and employs the von Mises-Fisher Mixture Model (vMFMM) distribution as an embedding in the latent space. Additionally, it maximizes the Extended Mutual Information (EMI) between latent representations and predicted cluster assignments for more discriminative and balanced allocations. The effectiveness of the proposed deep clustering method is validated through comparisons with state-of-the-art deep clustering techniques on benchmark datasets. The established model provides a novel approach for complex data feature learning and cluster analysis, extending the application of Bayesian inference methods in deep clustering.

    参考文献
    相似文献
    引证文献
引用本文

余南骏,吴昊旻,陈飞宇,刘学文.深度超球面变分聚类网络[J].重庆师范大学学报自然科学版,2024,41(3):100-112

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-09-20