基于联邦集成学习的电量需求预测方法和系统

摘要

本发明提出了一种基于联邦集成学习的电量需求预测方法和系统，方法包括以下步骤：包括以下步骤：总电站从所有参与训练的子电站中随机抽取部分子电站参加本轮训练，并将预训练的全局模型发送给相应的子电站；每个子电站收到全局模型，并对自身的本地模型进行初始化，并且子电站使用自身的本地数据集进行本地训练；子电站在进行本地训练之后将自身的本地模型上传给总电站，总电站对收到的本地模型执行聚合操作，得到新的全局模型；循环执行以上步骤，直至全局模型达到收敛，当前每个子电站的本地模型为电量需求预测模型，子电站采用对应的电量需求预测模型执行电量需求预测任务。根据本发明实施例的方法，能够实现对于子电站的电量需求的预测。