Federated Multi-Task Clustering

TL;DR

This paper introduces FMTC, a federated multi-task clustering framework that enables personalized, privacy-preserving clustering for heterogeneous clients by capturing shared structures and avoiding unreliable pseudo-labels.

Contribution

The paper proposes a novel federated clustering framework with client-specific models and a server-side tensor correlation module, improving clustering performance in decentralized, heterogeneous environments.

Findings

FMTC significantly outperforms baseline federated clustering methods.

The framework effectively captures shared structures across heterogeneous clients.

The distributed algorithm ensures privacy-preserving, efficient optimization.

Abstract

Spectral clustering has emerged as one of the most effective clustering algorithms due to its superior performance. However, most existing models are designed for centralized settings, rendering them inapplicable in modern decentralized environments. Moreover, current federated learning approaches often suffer from poor generalization performance due to reliance on unreliable pseudo-labels, and fail to capture the latent correlations amongst heterogeneous clients. To tackle these limitations, this paper proposes a novel framework named Federated Multi-Task Clustering (i.e.,FMTC), which intends to learn personalized clustering models for heterogeneous clients while collaboratively leveraging their shared underlying structure in a privacy-preserving manner. More specifically, the FMTC framework is composed of two main components: client-side personalized clustering module, which learns a parameterized mapping model to support robust out-of-sample inference, bypassing the need for unreliable pseudo-labels; and server-side tensorial correlation module, which explicitly captures the shared knowledge across all clients. This is achieved by organizing all client models into a unified tensor and applying a low-rank regularization to discover their common subspace. To solve this joint optimization problem, we derive an efficient, privacy-preserving distributed algorithm based on the Alternating Direction Method of Multipliers, which decomposes the global problem into parallel local updates on clients and an aggregation step on the server. To the end, several extensive experiments on multiple real-world datasets demonstrate that our proposed FMTC framework significantly outperforms various baseline and state-of-the-art federated clustering algorithms.