arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.02162stat.MLcs.LGmath.STstat.TH
收藏

贝叶斯聚类有效性指数

A Bayesian cluster validity index

King Mongkut’s University of Technology Thonburi(国王 Mongkut 理工学院(吞武里))

Nathakhun Wiroonsri, Onthada Preedasawakul

AI总结:

本研究提出贝叶斯聚类有效性指数,通过贝叶斯方法优化聚类数量选择,适用于用户专业知识指导的场景。

摘要:

选择适当的聚类数量是应用聚类算法的关键步骤。为了帮助完成此过程,已开发出各种聚类有效性指数(CVIs)。这些指数旨在识别数据集中最佳的聚类数量。然而,用户可能并不总是寻求绝对最优的聚类数量,而是寻求一个更符合其特定应用的次优选项。这一认识促使我们引入了贝叶斯聚类有效性指数(BCVI),该指数基于现有指数构建。BCVI使用狄利克雷或广义狄利克雷先验,导致相同的后验分布。我们通过Wiroonsri指数(硬聚类)和Wiroonsri-Preedasawakul指数(软聚类)作为底层指数来评估我们的BCVI。我们将所提出的BCVI的性能与原始底层指数以及几种其他现有CVIs(包括Davies-Bouldin、Starczewski、Xie-Beni和KWON2指数)进行比较。BCVI在用户专业知识有价值的情况下提供了明显优势,允许用户指定最终聚类数量的期望范围。为此,我们进行了三种不同场景的实验。此外,我们通过现实世界的数据集(如MRI脑肿瘤图像)展示了我们方法的实用性。这些工具将作为新的R包'BayesCVI'发布。

英文摘要:

Selecting the appropriate number of clusters is a critical step in applying clustering algorithms. To assist in this process, various cluster validity indices (CVIs) have been developed. These indices are designed to identify the optimal number of clusters within a dataset. However, users may not always seek the absolute optimal number of clusters but rather a secondary option that better aligns with their specific applications. This realization has led us to introduce a Bayesian cluster validity index (BCVI), which builds upon existing indices. The BCVI utilizes either Dirichlet or generalized Dirichlet priors, resulting in the same posterior distribution. We evaluate our BCVI using the Wiroonsri index for hard clustering and the Wiroonsri-Preedasawakul index for soft clustering as underlying indices. We compare the performance of our proposed BCVI with that of the original underlying indices and several other existing CVIs, including Davies-Bouldin, Starczewski, Xie-Beni, and KWON2 indices. Our BCVI offers clear advantages in situations where user expertise is valuable, allowing users to specify their desired range for the final number of clusters. To illustrate this, we conduct experiments classified into three different scenarios. Additionally, we showcase the practical applicability of our approach through real-world datasets, such as MRI brain tumor images. These tools will be published as a new R package 'BayesCVI'.

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

↑