Addressing Heterogeneity in Federated Load Forecasting with Personalization Layers
Fuente:
arXiv
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _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 |