DeepAI AI Chat
Log In Sign Up

Personalized Federated Learning with Clustering: Non-IID Heart Rate Variability Data Application

by   Joo Hun Yoo, et al.

While machine learning techniques are being applied to various fields for their exceptional ability to find complex relations in large datasets, the strengthening of regulations on data ownership and privacy is causing increasing difficulty in its application to medical data. In light of this, Federated Learning has recently been proposed as a solution to train on private data without breach of confidentiality. This conservation of privacy is particularly appealing in the field of healthcare, where patient data is highly confidential. However, many studies have shown that its assumption of Independent and Identically Distributed data is unrealistic for medical data. In this paper, we propose Personalized Federated Cluster Models, a hierarchical clustering-based FL process, to predict Major Depressive Disorder severity from Heart Rate Variability. By allowing clients to receive more personalized model, we address problems caused by non-IID data, showing an accuracy increase in severity prediction. This increase in performance may be sufficient to use Personalized Federated Cluster Models in many existing Federated Learning scenarios.


PFL-MoE: Personalized Federated Learning Based on Mixture of Experts

Federated learning (FL) is an emerging distributed machine learning para...

Dynamic Clustering in Federated Learning

In the resource management of wireless networks, Federated Learning has ...

Federated Learning with Adaptive Batchnorm for Personalized Healthcare

There is a growing interest in applying machine learning techniques for ...

FedHealth 2: Weighted Federated Transfer Learning via Batch Normalization for Personalized Healthcare

The success of machine learning applications often needs a large quantit...

Adaptive Personlization in Federated Learning for Highly Non-i.i.d. Data

Federated learning (FL) is a distributed learning method that offers med...

Three Approaches for Personalization with Applications to Federated Learning

The standard objective in machine learning is to train a single model fo...