arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 0907.0808cs.LG
收藏

Dirichlet过程先验下监督聚类的贝叶斯模型

A Bayesian Model for Supervised Clustering with the Dirichlet Process Prior

Information Sciences Institute(信息科学研究所);University of Southern California(南加州大学)

Hal Daumé, Daniel Marcu

摘要:

我们开发了一个贝叶斯框架来解决监督聚类问题,在任务中遇到的一般问题,如参考匹配,共指消解,身份不确定性和记录链接。我们的聚类模型是基于Dirichlet过程的先验,这使我们能够定义分布在自然出现在这个问题的可数无限集。我们通过假设存在一组未观察到的随机变量(我们称之为“参考类型”)来为我们的模型添加监督,这些变量在所有集群中都是通用的。在我们的框架中,这需要整合无限多个参数的推理,解决了使用马尔可夫链蒙特卡罗技术。我们提出了共轭和非共轭先验的算法。我们提出了一个简单的-但一般-我们的模型参数化的基础上高斯假设。我们在一个人工任务和三个现实任务上评估了这个模型,并将其与无监督和最先进的监督算法进行了比较。我们的研究结果表明,我们的模型能够在各种任务和性能指标上优于其他模型。

英文摘要:

We develop a Bayesian framework for tackling the supervised clustering problem, the generic problem encountered in tasks such as reference matching, coreference resolution, identity uncertainty and record linkage. Our clustering model is based on the Dirichlet process prior, which enables us to define distributions over the countably infinite sets that naturally arise in this problem. We add supervision to our model by positing the existence of a set of unobserved random variables (we call these "reference types") that are generic across all clusters. Inference in our framework, which requires integrating over infinitely many parameters, is solved using Markov chain Monte Carlo techniques. We present algorithms for both conjugate and non-conjugate priors. We present a simple--but general--parameterization of our model based on a Gaussian assumption. We evaluate this model on one artificial task and three real-world tasks, comparing it against both unsupervised and state-of-the-art supervised algorithms. Our results show that our model is able to outperform other models across a variety of tasks and performance metrics.

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

↑