arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.03587cs.LGstat.ML
收藏

信息论主动相关聚类

Information-Theoretic Active Correlation Clustering

Department of Computer Science and Engineering(计算机科学与工程系);Chalmers University of Technology(楚姆勒大学);University of Gothenburg(哥德堡大学)

Linus Aronsson, Morteza Haghir Chehreghani

AI总结:

本文提出基于信息论的主动学习方法,用于在相似性信息有限或昂贵时提高相关聚类的效率和准确性。

摘要:

相关聚类是一种基于成对相似性或不相似性信息对数据进行分组的灵活框架,不需要将簇数作为输入。然而,在许多实际场景中,这些成对相似性并不事先可用,必须通过昂贵的测量或人类反馈来获得。这促使使用主动学习来查询最有信息量的成对比较,从而在预算限制下实现有效的聚类。在本文中,我们通过引入几种信息论获取函数,开发了一种原则性的主动学习方法用于相关聚类,这些函数根据熵和预期信息增益优先选择查询。这些策略旨在尽可能高效地减少对聚类结构的不确定性。我们评估了我们的方法在各种合成和现实世界设置中的表现,并展示了它们在聚类准确性和查询效率方面显著优于现有基线。我们的结果突显了在相似性信息昂贵或有限的情况下,将主动学习与相关聚类结合的好处。

英文摘要:

Correlation clustering is a flexible framework for partitioning data based solely on pairwise similarity or dissimilarity information, without requiring the number of clusters as input. However, in many practical scenarios, these pairwise similarities are not available a priori and must be obtained through costly measurements or human feedback. This motivates the use of active learning to query only the most informative pairwise comparisons, enabling effective clustering under budget constraints. In this work, we develop a principled active learning approach for correlation clustering by introducing several information-theoretic acquisition functions that prioritize queries based on entropy and expected information gain. These strategies aim to reduce uncertainty about the clustering structure as efficiently as possible. We evaluate our methods across a range of synthetic and real-world settings and show that they significantly outperform existing baselines in terms of clustering accuracy and query efficiency. Our results highlight the benefits of combining active learning with correlation clustering in settings where similarity information is costly or limited.

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

↑