中文

利用个性化层解决联邦负载预测中的异质性

机器学习 2024-04-03 v1 信号处理

摘要

智能电表的普及使得能够广泛收集能源消耗数据,用于训练短期负荷预测模型。为应对隐私问题,联邦学习(FL)被提出作为一种隐私保护的训练方法,但当客户端数据变得异质时,训练模型的质量会下降。本文提出在一个称为PL-FL的通用框架中使用个性化层进行负载预测。我们证明PL-FL优于FL和纯本地训练,同时所需的通信带宽低于FL。通过在NREL ComStock存储库中的三个不同数据集上进行大量仿真,验证了这一点。

关键词

引用

@article{arxiv.2404.01517,
  title  = {Addressing Heterogeneity in Federated Load Forecasting with Personalization Layers},
  author = {Shourya Bose and Yu Zhang and Kibaek Kim},
  journal= {arXiv preprint arXiv:2404.01517},
  year   = {2024}
}