arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2508.08022cs.DCcs.LG

面向微电网可扩展电力需求预测的联邦学习优化

Optimizing Federated Learning for Scalable Power-demand Forecasting in Microgrids

Department of Computational and Data Sciences, Indian Institute of Science (IISc), Bangalore, India(计算与数据科学系,印度科学研究院(IISc),班加罗尔,印度);Department of Electrical Engineering, Indian Institute of Science (IISc), Bangalore, India(电气工程系,印度科学研究院(IISc),班加罗尔,印度);Accenture, India(印度德勤)

Roopkatha Banerjee, Sampath Koti, Gyanendra Singh, Anirban Chakraborty, Gurunath Gurrala, Bhushan Jagyasi, Yogesh Simmhan

AI总结:

针对微电网电力需求预测中数据隐私风险、非独立同分布数据及大规模客户端扩展难题,提出指数加权损失等联邦学习优化方案,经千级客户端实验验证,其精度和可扩展性优于传统基线方法。

摘要:

通过物联网(IoT)对城市和微电网的电力消耗进行实时监测,有助于预测未来需求并优化电网运营。但将所有用户级用电数据传输到云端,以进行细时间尺度的预测和分析,可能会泄露用户活动模式。联邦学习(Federated Learning,FL)是一种注重隐私的协作式深度神经网络(DNN)训练方法,它将数据保留在边缘设备上,在本地基于私有数据训练模型,并在云端聚合本地模型。但该方法存在关键挑战:(i)客户端的数据可能是非独立同分布(non-IID)的;(ii)学习过程的计算成本应较低,同时能扩展到数千个(未见过的)客户端。本文针对微电网和城市级公用事业的时间序列需求预测场景,开发并评估了多种适用于边缘与云端协同的FL训练优化方案,采用DNN在最小化训练成本的同时实现高预测精度。我们展示了训练时使用指数加权损失的优势,并证明它能进一步提升最终模型的预测效果。最后,我们基于OpenEIA语料库中美国三个州的数千个客户端数据,在伪分布式环境和树莓派(Pi)边缘集群上开展FL实验,对这些策略进行了验证。结果表明,与ARIMA、面向单个用户训练的DNN等不可扩展的基线方法相比,所提方法具备显著优势。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑