有效

自适应边缘联邦学习客户端调度方法、装置及电子设备

束妮娜、颜康、余静波、吴韬、杨方、王晨、王怀习、周政伟
中国人民解放军国防科技大学

摘要

本发明公开了一种自适应边缘联邦学习客户端调度方法、装置及电子设备,该方法应用于服务器,包括:初始化生成结构相同的第一神经网络与第二神经网络;令,获取客户端在第轮训练时的状态向量集合;将状态向量集合输入至第一神经网络,得到客户端在第轮训练时的动作向量集合;将待训练的全局模型广播至各个客户端后,以使各客户端基于自身对应的动作向量和本地数据对全局模型进行训练,得到本地模型;获取多个本地模型的参数并聚合,得到新的全局模型;判断新的全局模型的当前精度是否满足预设条件;若是,则将新的全局模型作为训练完成的全局模型。本发明有效提升了全局模型精度和能量效率。 本发明公开了一种自适应边缘联邦学习客户端调度方法、装置及电子设备,该方法应用于服务器,包括:初始化生成结构相同的第一神经网络与第二神经网络;令<Image he="25" wi="38" file="ZY_1.GIF" imgContent="drawing" imgFormat="GIF" orientation="portrait" inline="yes"/>,获取客户端在第<Image he="18" wi="14" file="ZY_2.GIF" imgContent="drawing" imgFormat="GIF" orientation="portrait" inline="yes"/>轮训练时的状态向量集合<Image he="25" wi="24" file="ZY_3.GIF" imgContent="drawing" imgFormat="GIF" orientation="portrait" inline="yes"/>;将状态向量集合<Image he="27" wi="24" file="ZY_4.GIF" imgContent="drawing" imgFormat="GIF" orientation="portrait" inline="yes"/>输入至第一神经网络,得到客户端在第<Image he="17" wi="13" file="ZY_5.GIF" imgContent="drawing" imgFormat="GIF" orientation="portrait" inline="yes"/>轮训练时的动作向量集合<Image he="23" wi="22" file="ZY_6.GIF" imgContent="drawing" imgFormat="GIF" orientation="portrait" inline="yes"/>;将待训练的全局模型广播至各个客户端后,以使各客户端基于自身对应的动作向量和本地数据对全局模型进行训练,得到本地模型;获取多个本地模型的参数并聚合,得到新的全局模型;判断新的全局模型的当前精度是否满足预设条件;若是,则将新的全局模型作为训练完成的全局模型。本发明有效提升了全局模型精度和能量效率。