PTOPOFL: Topology-Guided Personalised Federated Learning with Persistent-Homology Client Descriptors

arXiv:2603.04323 · cs.LG, cs.CR, cs.DC, math.AT, stat.ML · Submitted 2026-03-04 · Read on arXiv

cs.LG, cs.CR, cs.DC, math.AT, stat.ML

Submitted: 2026-03-04

Updated: 2026-10-19

Comments: 32 pages, 6 Figures. Revised version with fine theoretical demonstration on convergence. Also real clinical dataset added

License: http://creativecommons.org/licenses/by-nc-nd/4.0/

The gist: Clustered and personalised federated learning (FL) needs a signal telling the server which clients should share a model.

Terminology

Abstract

Clustered and personalised federated learning (FL) needs a signal telling the server which clients should share a model. Most methods derive it from losses or model updates that must be recomputed during training. We study pTopoFL, which derives it once, before training, from a compact persistent-homology (PH) descriptor of each client's feature distribution: the server clusters clients on these descriptors, weights them within clusters, and runs blended clustered local gradient descent, returning one personalised model per cluster. For strongly convex, smooth losses we analyse the complete procedure: for any partition, one round is a contraction with factor (1-eta*mu) tau; the fixed point is characterised by a local-step drift bound, an exact formula for the effective strength of blending (which yields a calibration rule) and a stochastic floor exposing a bias-variance trade-off of clustering; and personalisation lowers the heterogeneity floor in proportion to a measurable share of the dispersion of client optima. Experiments on synthetic federations, a label-skew benchmark and a real four-hospital dataset confirm the theory and delimit the method. When feature geometry and decision rules co-vary, the PH partition recovers the latent client groups, including differences a linear-subspace descriptor misses, and pTopoFL is competitive with the strongest clustered baselines; on the real federation it lowers the test log-loss of FedAvg from 0.463 to 0.438, on par with the best personalised methods. When heterogeneity is invisible in feature geometry, it offers no advantage over FedAvg. pTopoFL transmits model parameters and is no more private than FedAvg; we give a record-level differential-privacy accounting and show that privatising the descriptor destroys the partition. Code reproduces every number.

Related papers