Addressing Heterogeneity in Federated Load Forecasting with Personalization Layers

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Bose, Shourya, Zhang, Yu, Kim, Kibaek
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910395477786624
author Bose, Shourya
Zhang, Yu
Kim, Kibaek
author_facet Bose, Shourya
Zhang, Yu
Kim, Kibaek
contents The advent of smart meters has enabled pervasive collection of energy consumption data for training short-term load forecasting models. In response to privacy concerns, federated learning (FL) has been proposed as a privacy-preserving approach for training, but the quality of trained models degrades as client data becomes heterogeneous. In this paper we propose the use of personalization layers for load forecasting in a general framework called PL-FL. We show that PL-FL outperforms FL and purely local training, while requiring lower communication bandwidth than FL. This is done through extensive simulations on three different datasets from the NREL ComStock repository.
format Preprint
id arxiv_https___arxiv_org_abs_2404_01517
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Addressing Heterogeneity in Federated Load Forecasting with Personalization Layers
Bose, Shourya
Zhang, Yu
Kim, Kibaek
Machine Learning
Signal Processing
The advent of smart meters has enabled pervasive collection of energy consumption data for training short-term load forecasting models. In response to privacy concerns, federated learning (FL) has been proposed as a privacy-preserving approach for training, but the quality of trained models degrades as client data becomes heterogeneous. In this paper we propose the use of personalization layers for load forecasting in a general framework called PL-FL. We show that PL-FL outperforms FL and purely local training, while requiring lower communication bandwidth than FL. This is done through extensive simulations on three different datasets from the NREL ComStock repository.
title Addressing Heterogeneity in Federated Load Forecasting with Personalization Layers
topic Machine Learning
Signal Processing
url https://arxiv.org/abs/2404.01517