Differentially Private Federated $k$-Means Clustering with Server-Side Data

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Scott, Jonathan, Lampert, Christoph H., Saulpic, David
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909645683032064
author Scott, Jonathan
Lampert, Christoph H.
Saulpic, David
author_facet Scott, Jonathan
Lampert, Christoph H.
Saulpic, David
contents Clustering is a cornerstone of data analysis that is particularly suited to identifying coherent subgroups or substructures in unlabeled data, as are generated continuously in large amounts these days. However, in many cases traditional clustering methods are not applicable, because data are increasingly being produced and stored in a distributed way, e.g. on edge devices, and privacy concerns prevent it from being transferred to a central server. To address this challenge, we present FedDP-KMeans, a new algorithm for $k$-means clustering that is fully-federated as well as differentially private. Our approach leverages (potentially small and out-of-distribution) server-side data to overcome the primary challenge of differentially private clustering methods: the need for a good initialization. Combining our initialization with a simple federated DP-Lloyds algorithm we obtain an algorithm that achieves excellent results on synthetic and real-world benchmark tasks. We also provide a theoretical analysis of our method that provides bounds on the convergence speed and cluster identification success.
format Preprint
id arxiv_https___arxiv_org_abs_2506_05408
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Differentially Private Federated $k$-Means Clustering with Server-Side Data
Scott, Jonathan
Lampert, Christoph H.
Saulpic, David
Cryptography and Security
Machine Learning
Clustering is a cornerstone of data analysis that is particularly suited to identifying coherent subgroups or substructures in unlabeled data, as are generated continuously in large amounts these days. However, in many cases traditional clustering methods are not applicable, because data are increasingly being produced and stored in a distributed way, e.g. on edge devices, and privacy concerns prevent it from being transferred to a central server. To address this challenge, we present FedDP-KMeans, a new algorithm for $k$-means clustering that is fully-federated as well as differentially private. Our approach leverages (potentially small and out-of-distribution) server-side data to overcome the primary challenge of differentially private clustering methods: the need for a good initialization. Combining our initialization with a simple federated DP-Lloyds algorithm we obtain an algorithm that achieves excellent results on synthetic and real-world benchmark tasks. We also provide a theoretical analysis of our method that provides bounds on the convergence speed and cluster identification success.
title Differentially Private Federated $k$-Means Clustering with Server-Side Data
topic Cryptography and Security
Machine Learning
url https://arxiv.org/abs/2506.05408