Fuzzy c-Means Clustering for Persistence Diagrams

Thomas Davies, Jack Aspinall, Bryan Wilder, Long Tran-Thanh

Keywords: Topological data analysis, Fuzzy c-means, fuzzy clustering, materials science, model selection

TL;DR: We develop Fuzzy c-Means clustering for persistence diagrams, with experiments on lattice structures and decision boundaries.

Abstract: Persistence diagrams concisely represent the topology of a point cloud whilst having strong theoretical guarantees. Most current approaches to integrating topological information into machine learning implicitly map persistence diagrams to a Hilbert space, resulting in deformation of the underlying metric structure whilst also generally requiring prior knowledge about the true topology of the space. In this paper we give an algorithm for Fuzzy c-Means (FCM) clustering directly on the space of persistence diagrams, enabling unsupervised learning that automatically captures the topological structure of data, with no prior knowledge or additional processing of persistence diagrams. We prove the same convergence guarantees as traditional FCM clustering: every convergent subsequence of iterates tends to a local minimum or saddle point. We end by presenting experiments where the fuzzy nature of our topological clustering is capitalised on: lattice structure classification in materials science and pre-trained model selection in machine learning.

Previous Submission: Yes

Community Implementations: [ 4 code implementations](https://www.catalyzex.com/paper/fuzzy-c-means-clustering-for-persistence/code)