Saved in:
Bibliographic Details
Main Authors: Ghari, Pouya M., Shen, Yanning
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.21547
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912090824900608
author Ghari, Pouya M.
Shen, Yanning
author_facet Ghari, Pouya M.
Shen, Yanning
contents Federated learning is renowned for its efficacy in distributed model training, ensuring that users, called clients, retain data privacy by not disclosing their data to the central server that orchestrates collaborations. Most previous work on federated learning assumes that clients possess static batches of training data. However, clients may also need to make real-time predictions on streaming data in non-stationary environments. In such dynamic environments, employing pre-trained models may be inefficient, as they struggle to adapt to the constantly evolving data streams. To address this challenge, clients can fine-tune models online, leveraging their observed data to enhance performance. Despite the potential benefits of client participation in federated online model fine-tuning, existing analyses have not conclusively demonstrated its superiority over local model fine-tuning. To bridge this gap, the present paper develops a novel personalized federated learning algorithm, wherein each client constructs a personalized model by combining a locally fine-tuned model with multiple federated models learned by the server over time. Theoretical analysis and experiments on real datasets corroborate the effectiveness of this approach for real-time predictions and federated model fine-tuning.
format Preprint
id arxiv_https___arxiv_org_abs_2410_21547
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Personalized Federated Learning with Mixture of Models for Adaptive Prediction and Model Fine-Tuning
Ghari, Pouya M.
Shen, Yanning
Machine Learning
Federated learning is renowned for its efficacy in distributed model training, ensuring that users, called clients, retain data privacy by not disclosing their data to the central server that orchestrates collaborations. Most previous work on federated learning assumes that clients possess static batches of training data. However, clients may also need to make real-time predictions on streaming data in non-stationary environments. In such dynamic environments, employing pre-trained models may be inefficient, as they struggle to adapt to the constantly evolving data streams. To address this challenge, clients can fine-tune models online, leveraging their observed data to enhance performance. Despite the potential benefits of client participation in federated online model fine-tuning, existing analyses have not conclusively demonstrated its superiority over local model fine-tuning. To bridge this gap, the present paper develops a novel personalized federated learning algorithm, wherein each client constructs a personalized model by combining a locally fine-tuned model with multiple federated models learned by the server over time. Theoretical analysis and experiments on real datasets corroborate the effectiveness of this approach for real-time predictions and federated model fine-tuning.
title Personalized Federated Learning with Mixture of Models for Adaptive Prediction and Model Fine-Tuning
topic Machine Learning
url https://arxiv.org/abs/2410.21547