arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 1909.08787stat.MLcs.LG
收藏

基于瓦瑟斯坦距离的有效多层聚类

On Efficient Multilevel Clustering via Wasserstein Distances

Monash University(莫纳什大学);University of California, Berkeley(加州大学伯克利分校);University of Michigan(密歇根大学);IBM Research(IBM研究院);VinAI Research(VinAI研究院)

Viet Huynh, Nhat Ho, Nhan Dam, XuanLong Nguyen, Mikhail Yurochkin, Hung Bui, and Dinh Phung

摘要:

我们提出了一种新的方法来解决多级聚类的问题,其目的是同时划分每个组中的数据,并在一个潜在的大层次结构的数据语料库中发现组之间的分组模式。我们的方法涉及到一个联合优化制定了几个空间的离散概率测度,这是赋予瓦瑟斯坦距离度量。我们提出了这个问题的几个变种,承认快速优化算法,通过利用连接的问题,找到Wasserstein重心。建立了局部和全局聚类估计的一致性性质。最后,合成和真实数据的实验结果表明所提出的方法的灵活性和可扩展性。

英文摘要:

We propose a novel approach to the problem of multilevel clustering, which aims to simultaneously partition data in each group and discover grouping patterns among groups in a potentially large hierarchically structured corpus of data. Our method involves a joint optimization formulation over several spaces of discrete probability measures, which are endowed with Wasserstein distance metrics. We propose several variants of this problem, which admit fast optimization algorithms, by exploiting the connection to the problem of finding Wasserstein barycenters. Consistency properties are established for the estimates of both local and global clusters. Finally, experimental results with both synthetic and real data are presented to demonstrate the flexibility and scalability of the proposed approach.

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

↑