用联邦学习优化微电网用电预测,兼顾隐私与大规模扩展。
Optimizing Federated Learning for Scalable Power-demand Forecasting in Microgrids
- 在边缘设备上本地训练,云端聚合模型,保护用户隐私。
- 采用加权损失函数,提升上千客户端的预测准确率。
- 适合需要隐私保护的大规模城市能源系统建模者。
通过物联网实时监控城市和微电网的电力消耗,有助于预测未来需求并优化电网运行。但将所有用户级用电数据上传至云端进行细粒度分析可能暴露活动模式。联邦学习(FL)是一种注重隐私的分布式深度神经网络训练方法,数据保留在边缘设备上,本地训练模型后在云端聚合。然而存在两大挑战:(i) 客户端数据非独立同分布(non-IID),(ii) 学习过程需计算高效,且能扩展至数千个(未知)客户端。本文针对微电网与城市级电力系统的时间序列需求预测,优化了跨边缘与云端的联邦学习训练流程,使用DNN实现高精度预测的同时最小化训练开销。我们展示了在训练中采用指数加权损失函数的优势,进一步提升了最终模型的预测性能。通过在来自OpenEIA语料库的美国三个州超过1000个客户端上验证,并在伪分布式环境与树莓派边缘集群中执行联邦学习,结果表明所提方法优于ARIMA及为单个用户训练的DNN,具有更好的可扩展性。
原文摘要 · Abstract (English)
Real-time monitoring of power consumption in cities and micro-grids through the Internet of Things (IoT) can help forecast future demand and optimize grid operations. But moving all consumer-level usage data to the cloud for predictions and analysis at fine time scales can expose activity patterns. Federated Learning~(FL) is a privacy-sensitive collaborative DNN training approach that retains data on edge devices, trains the models on private data locally, and aggregates the local models in the cloud. But key challenges exist: (i) clients can have non-independently identically distributed~(non-IID) data, and (ii) the learning should be computationally cheap while scaling to 1000s of (unseen) clients. In this paper, we develop and evaluate several optimizations to FL training across edge and cloud for time-series demand forecasting in micro-grids and city-scale utilities using DNNs to achieve a high prediction accuracy while minimizing the training cost. We showcase the benefit of using exponentially weighted loss while training and show that it further improves the prediction of the final model. Finally, we evaluate these strategies by validating over 1000s of clients for three states in the US from the OpenEIA corpus, and performing FL both in a pseudo-distributed setting and a Pi edge cluster. The results highlight the benefits of the proposed methods over baselines like ARIMA and DNNs trained for individual consumers, which are not scalable.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。