Heterogeneous Federated Learning System for Sparse Healthcare Time-Series Prediction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Syu, Jia-Hao, Lin, Jerry Chun-Wei
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912197389582336
author Syu, Jia-Hao
Lin, Jerry Chun-Wei
author_facet Syu, Jia-Hao
Lin, Jerry Chun-Wei
contents In this paper, we propose a heterogeneous federated learning (HFL) system for sparse time series prediction in healthcare, which is a decentralized federated learning algorithm with heterogeneous transfers. We design dense and sparse feature tensors to deal with the sparsity of data sources. Heterogeneous federated learning is developed to share asynchronous parts of networks and select appropriate models for knowledge transfer. Experimental results show that the proposed HFL achieves the lowest prediction error among all benchmark systems on eight out of ten prediction tasks, with MSE reduction of 94.8%, 48.3%, and 52.1% compared to the benchmark systems. These results demonstrate the effectiveness of HFL in transferring knowledge from heterogeneous domains, especially in the smaller target domain. Ablation studies then demonstrate the effectiveness of the designed mechanisms for heterogeneous domain selection and switching in predicting healthcare time series with privacy, model security, and heterogeneous knowledge transfer.
format Preprint
id arxiv_https___arxiv_org_abs_2501_12125
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Heterogeneous Federated Learning System for Sparse Healthcare Time-Series Prediction
Syu, Jia-Hao
Lin, Jerry Chun-Wei
Machine Learning
In this paper, we propose a heterogeneous federated learning (HFL) system for sparse time series prediction in healthcare, which is a decentralized federated learning algorithm with heterogeneous transfers. We design dense and sparse feature tensors to deal with the sparsity of data sources. Heterogeneous federated learning is developed to share asynchronous parts of networks and select appropriate models for knowledge transfer. Experimental results show that the proposed HFL achieves the lowest prediction error among all benchmark systems on eight out of ten prediction tasks, with MSE reduction of 94.8%, 48.3%, and 52.1% compared to the benchmark systems. These results demonstrate the effectiveness of HFL in transferring knowledge from heterogeneous domains, especially in the smaller target domain. Ablation studies then demonstrate the effectiveness of the designed mechanisms for heterogeneous domain selection and switching in predicting healthcare time series with privacy, model security, and heterogeneous knowledge transfer.
title Heterogeneous Federated Learning System for Sparse Healthcare Time-Series Prediction
topic Machine Learning
url https://arxiv.org/abs/2501.12125