基于联邦学习的地铁基坑开挖风险识别方法及装置

摘要

本发明实施例公开了基于联邦学习的地铁基坑开挖风险识别方法及装置。该方法包括参与联邦学习的各个客户端节点获取地铁基坑的监测数据，并得到标准样本；各个客户端节点对本地的标准样本进行加密对齐处理，并从加密后的标准样本筛选出目标数据集；各个客户端节点基于目标数据集训练风险识别模型得到模型局部参数；服务端节点汇总所有模型局部参数，得到更新后的模型全局参数，并反馈给各客户端节点；各客户端节点持续地对模型局部参数进行迭代、上传和接收，直至服务端节点对应的损失函数收敛，得到优化后的风险识别模型；采用验证集测试优化后的风险识别模型，得到最优的风险识别模型；利用最优的风险识别模型确定识别结果。该方法准确性更高。