arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2506.16295stat.CO

理解贝叶斯聚类分析中的不确定性

Beyond point estimation: explaining the posterior in Bayesian cluster analysis

Cecilia Balocchi, Sara Wade

AI总结:

WASABI通过多聚类估计近似后验分布,提升贝叶斯聚类分析中对不确定性的理解,尤其在聚类不明显或模型不恰当时表现优异。

摘要:

贝叶斯聚类方法常因其能够提供分区结构的不确定性而受到赞赏。然而,总结后验分布 over 分区结构可能具有挑战性,因为该空间是离散的、无序的且维度巨大。尽管最近的进展提供了一个单一的聚类估计来代表后验,但这忽略了不确定性,并且在后验是多模态的情况下可能不具有代表性。为了增强对不确定性的理解,我们提出了一种Wasserstein近似于贝叶斯聚类(WASABI),该方法用不止一个聚类估计来总结后验样本,每个对应于分区空间中具有大量后验质量的不同部分。具体来说,我们通过在分区空间上配备合适度量的Wasserstein距离近似后验分布来找到这些聚类估计。有趣的是,一个局部最优解可以通过在分区空间上使用k-medoids-like算法来找到,将后验样本分成若干组,每组由一个聚类估计表示。使用合成和实际数据集,我们证明WASABI有助于提高对不确定性的理解,特别是在聚类不明显分离或所用模型不恰当的情况下。

英文摘要:

The Bayesian approach to clustering is often appreciated for its ability to provide uncertainty in the partition structure. However, summarising the posterior distribution over partitions remains challenging, due to its discrete, unordered and extremely high-dimensional nature. Existing approaches typically focus on a single representative clustering, which can obscure important features of the posterior, particularly in the presence of multimodality. To address this limitation, we propose a WASserstein Approximation for Bayesian Inference (WASABI), which represents the posterior distribution using a small number of weighted clustering solutions ("particles"), each capturing a distinct region of substantial posterior mass. The approximation is obtained by minimising the Wasserstein distance defined on the space of partitions, equipped with a suitable metric. This formulation naturally leads to a k-means-like algorithm on the partition space that divides posterior samples in representative groups, each represented by one of the clustering solutions. Through synthetic and real-data applications, we demonstrate that WASABI provides an interpretable and computationally efficient low-dimensional summary of the posterior, particularly in settings with overlapping clusters or model misspecification.

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

↑