ResultsIndex
SearchAbout
CS
4/9/2025

A Scalable Approach to Clustering Embedding Projections

Donghao Ren, Fred Hohman, Dominik Moritz

Abstract

Interactive visualization of embedding projections is a useful technique for understanding data and evaluating machine learning models. Labeling data within these visualizations is critical for interpretation, as labels provide an overview of the projection and guide user navigation. However, most methods for producing labels require clustering the points, which can be computationally expensive as the number of points grows. In this paper, we describe an efficient clustering approach using kernel density estimation in the projected 2D space instead of points. This algorithm can produce high-quality cluster regions from a 2D density map in a few hundred milliseconds, orders of magnitude faster than current approaches. We contribute the design of the algorithm, benchmarks, and applications that demonstrate the utility of the algorithm, including labeling and summarization.

BibTeX

@article{2504.07285, title={A Scalable Approach to Clustering Embedding Projections}, author={Donghao Ren and Fred Hohman and Dominik Moritz}, journal={arXiv preprint arXiv:2504.07285}, year={2025} }
Publication Metadata
Published4/9/2025
Read the paper
arXivarXiv HTMLPDF
Implementations
apple/embedding-atlas
4,531•TypeScript

Topic

ML