arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.06385cs.LG
收藏

约束聚类的深度条件高斯混合模型

Deep Conditional Gaussian Mixture Model for Constrained Clustering

ETH Zürich(苏黎世联邦理工学院);University Children’s Hospital Regensburg (KUNO)(雷根斯堡大学儿童医院(KUNO));University of Regensburg(雷根斯堡大学)

Laura Manduchi, Kieran Chin-Cheong, Holger Michel, Sven Wellmann, Julia E. Vogt

摘要:

约束聚类在机器学习领域得到了极大的关注,因为它可以利用越来越多的部分标记数据的先验信息。根据深度生成模型的最新进展,我们提出了一种新的约束聚类框架,该框架直观,可解释,并且可以在随机梯度变分推理的框架中有效地训练。通过明确地整合领域知识的概率关系的形式,我们提出的模型(DC-GMM)揭示了潜在的分布数据的条件上事先聚类的偏好,表示为成对的约束。这些约束通过指示哪些样本应该或不应该属于同一聚类来引导聚类过程朝向数据的期望分区。我们提供了大量的实验来证明,DC-GMM显示出优越的聚类性能和鲁棒性相比,国家的最先进的深度约束聚类方法在广泛的数据集。我们进一步证明了我们的方法在两个具有挑战性的现实世界中的应用程序的有用性。

英文摘要:

Constrained clustering has gained significant attention in the field of machine learning as it can leverage prior information on a growing amount of only partially labeled data. Following recent advances in deep generative models, we propose a novel framework for constrained clustering that is intuitive, interpretable, and can be trained efficiently in the framework of stochastic gradient variational inference. By explicitly integrating domain knowledge in the form of probabilistic relations, our proposed model (DC-GMM) uncovers the underlying distribution of data conditioned on prior clustering preferences, expressed as pairwise constraints. These constraints guide the clustering process towards a desirable partition of the data by indicating which samples should or should not belong to the same cluster. We provide extensive experiments to demonstrate that DC-GMM shows superior clustering performances and robustness compared to state-of-the-art deep constrained clustering methods on a wide range of data sets. We further demonstrate the usefulness of our approach on two challenging real-world applications.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑