论文标题

深层目标聚类

Deep Goal-Oriented Clustering

论文作者

Shi, Yifeng, Bender, Christopher M., Oliva, Junier B., Niethammer, Marc

论文摘要

聚类和预测分别是无监督和监督学习领域的两个主要任务。尽管机器学习最近的许多进步都围绕着这两个任务,但很少探索它们之间相互依存的,互惠互利的关系。可以合理地期望适当地聚类数据将有助于下游预测任务,相反,对下游任务的更好预测性能可能会为更合适的聚类策略提供信息。在这项工作中,我们专注于这种互惠关系的后半部分。为此,我们引入了深层目标聚类(DGC),这是一个概率的框架,该框架通过通过侧面信息和无监督的建模以端到端的方式共同使用监督来簇数据。我们通过实现与最先进的预测准确性来展示模型对一系列数据集的有效性,而在我们的环境中,更重要的是同时学习一致的聚类策略。

Clustering and prediction are two primary tasks in the fields of unsupervised and supervised learning, respectively. Although much of the recent advances in machine learning have been centered around those two tasks, the interdependent, mutually beneficial relationship between them is rarely explored. One could reasonably expect appropriately clustering the data would aid the downstream prediction task and, conversely, a better prediction performance for the downstream task could potentially inform a more appropriate clustering strategy. In this work, we focus on the latter part of this mutually beneficial relationship. To this end, we introduce Deep Goal-Oriented Clustering (DGC), a probabilistic framework that clusters the data by jointly using supervision via side-information and unsupervised modeling of the inherent data structure in an end-to-end fashion. We show the effectiveness of our model on a range of datasets by achieving prediction accuracies comparable to the state-of-the-art, while, more importantly in our setting, simultaneously learning congruent clustering strategies.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源