Topological Federated Clustering via Gravitational Potential Fields Under Local Differential Privacy

Authors

  • Yunbo Long Department of Engineering, University of Cambridge
  • Jiaquan Zhang Artificial Intelligence Innovation and Incubation Institute, Fudan University Shanghai Innovation Institute
  • Xi Chen Artificial Intelligence Innovation and Incubation Institute, Fudan University Shanghai Academy of AI for Science
  • Alexandra Brintrup Department of Engineering, University of Cambridge The Alan Turing Institute

DOI:

https://doi.org/10.1609/aaai.v40i28.39582

Abstract

Clustering non-independent and identically distributed (non-IID) data under local differential privacy (LDP) in federated settings presents a critical challenge: preserving privacy while maintaining accuracy without iterative communication. Existing one-shot methods rely on unstable pairwise centroid distances or neighborhood rankings, degrading severely under strong LDP noise and data heterogeneity. We present Gravitational Federated Clustering (GFC), a novel approach to privacy-preserving federated clustering that overcomes the limitations of distance-based methods under varying LDP. Addressing the critical challenge of clustering non-IID data with diverse privacy guarantees, GFC transforms privatized client centroids into a global gravitational potential field where true cluster centers emerge as topologically persistent singularities. Our framework introduces two key innovations: (1) a client-side compactness-aware perturbation mechanism that encodes local cluster geometry as "mass" values, and (2) a server-side topological aggregation phase that extracts stable centroids through persistent homology analysis of the potential field's superlevel sets. Theoretically, we establish a closed-form bound between the privacy budget ε and centroid estimation error, proving the potential field's Lipschitz smoothing properties exponentially suppress noise in high-density regions. Empirically, GFC outperforms state-of-the-art methods on ten benchmarks, especially under strong LDP constraints (ε < 1), while maintaining comparable performance at lower privacy budgets. By reformulating federated clustering as a topological persistence problem in a synthetic physics-inspired space, GFC achieves unprecedented privacy-accuracy trade-offs without iterative communication, providing a new perspective for privacy-preserving distributed learning.

Downloads

Published

2026-03-14

How to Cite

Long, Y., Zhang, J., Chen, X., & Brintrup, A. (2026). Topological Federated Clustering via Gravitational Potential Fields Under Local Differential Privacy. Proceedings of the AAAI Conference on Artificial Intelligence, 40(28), 24044–24051. https://doi.org/10.1609/aaai.v40i28.39582

Issue

Section

AAAI Technical Track on Machine Learning V