公开(公告)号：CN118941848A

公开(公告)日：2024-11-12

申请号：CN202410967635.2

申请日：2024-07-18

申请人： 中国铁道科学研究院集团有限公司电子计算技术研究所 ,  北京交通大学 ,  中国铁道科学研究院集团有限公司 ,  中国国家铁路集团有限公司

发明人： 祝咏升 ,  王伟 ,  单杏花 ,  刘冲 ,  吕晓婷 ,  傅一馨 ,  张军峰 ,  陈政 ,  许向蕊 ,  郝玉蓉

IPC分类号： G06V10/764 ,  G06N20/00 ,  G06F21/55

摘要： 本发明涉及鲁棒性检测技术领域，涉及一种基于联邦学习的鲁棒性检测方法、装置、设备及介质，所述方法包括获取第一全局模型，所述第一全局模型为服务器端训练的模型；将所述第一全局模型发送至本地端，并利用本地端的数据集对所述全局模型进行更新训练，得到至少一个本地模型；根据至少一个所述本地模型生成第一恶意模型；将所述第一恶意模型发送至服务器端进行聚合并根据聚合后的全局模型进行鲁棒性检测，本发明通过在本地生成第一恶意模型再将第一恶意模型上传至服务器端干扰全局模型更新，继而通过观察测试前后全局模型性能表现来判断联邦学习系统的鲁棒性，有效的提高了联邦学习系统的鲁棒性检测效率和速度。