Bayesian Level Set Clustering

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Buch, David, Dewaskar, Miheer, Dunson, David B.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915666876956672
author Buch, David
Dewaskar, Miheer
Dunson, David B.
author_facet Buch, David
Dewaskar, Miheer
Dunson, David B.
contents Classically, Bayesian clustering interprets each component of a mixture model as a cluster. The inferred clustering posterior is highly sensitive to any inaccuracies in the kernel within each component. As this kernel is made more flexible, problems arise in identifying the underlying clusters in the data. To address this pitfall, this article proposes a fundamentally different approach to Bayesian clustering that decouples the problems of clustering and flexible modeling of the data density $f$. Starting with an arbitrary Bayesian model for $f$ and a loss function for defining clusters based on $f$, we develop a Bayesian decision-theoretic framework for density-based clustering. Within this framework, we develop a Bayesian level set clustering method to cluster data into connected components of a level set of $f$. We provide theoretical support, including clustering consistency, and highlight performance in a variety of simulated examples. An application to astronomical data illustrates improvements over the popular DBSCAN algorithm in terms of accuracy, insensitivity to tuning parameters, and providing uncertainty quantification.
format Preprint
id arxiv_https___arxiv_org_abs_2403_04912
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Bayesian Level Set Clustering
Buch, David
Dewaskar, Miheer
Dunson, David B.
Methodology
62G99
Classically, Bayesian clustering interprets each component of a mixture model as a cluster. The inferred clustering posterior is highly sensitive to any inaccuracies in the kernel within each component. As this kernel is made more flexible, problems arise in identifying the underlying clusters in the data. To address this pitfall, this article proposes a fundamentally different approach to Bayesian clustering that decouples the problems of clustering and flexible modeling of the data density $f$. Starting with an arbitrary Bayesian model for $f$ and a loss function for defining clusters based on $f$, we develop a Bayesian decision-theoretic framework for density-based clustering. Within this framework, we develop a Bayesian level set clustering method to cluster data into connected components of a level set of $f$. We provide theoretical support, including clustering consistency, and highlight performance in a variety of simulated examples. An application to astronomical data illustrates improvements over the popular DBSCAN algorithm in terms of accuracy, insensitivity to tuning parameters, and providing uncertainty quantification.
title Bayesian Level Set Clustering
topic Methodology
62G99
url https://arxiv.org/abs/2403.04912