买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
摘要:本发明公开了一种基于孤立森林的联邦学习中毒攻击的防御方法,包括如下步骤:用干净公开数据集构建验证模型并对验证模型参数注入扰动;在联邦学习的每个迭代中采样上传各个节点更新好的模型参数,将前者和经过扰动处理的验证模型参数形成特征矩阵;使用孤立森林来划分特征矩阵的数据空间,并计算出每个参与者的异常概率分数;根据验证模型的分数在所有上传模型的分数的分布,不断调整阈值,以排除攻击者,并尽量减少防御模型对良性用户的影响。该方法可以自动防御中毒攻击,通过调整预先训练好的验证模型,显著提升联邦学习的鲁棒性,确保联邦学习全局模型的正常训练。
主权项:1.一种基于孤立森林的联邦学习中毒攻击的防御方法,其特征在于,实现该防御策略的场景是一个由数个参与方和一个中央服务器组成的水平联邦学习模型,每个参与者只访问自己的本地数据集,每个节点的数据都不与服务器共享,所述方法包括如下步骤:步骤1:联邦学习模型初始化:设置整体训练轮次、参与者的数量及其本地数据集和中央服务器;步骤2:验证模型预训练:配置干净的公共数据集训练验证模型,部分验证模型参数注入扰动;训练孤立森林异常检测模块使其能区分经过扰动和未经扰动的验证模型;步骤3:联邦学习第t轮训练时,每个节点利用中央服务器下发的全局模型参数和本地数据集Di在本地训练并产生新的模型参数n表示模型参数的维度,随机选择k个客户将其更新后的模型参数同步上传至中央服务器;步骤4:对采样上传的模型参数,不计算节点模型参数之间的距离,而是构造孤立森林异常检测模块,投入采样的模型参数和经过扰动处理的验证模型参数共同参与检测;步骤4.1:构造特征矩阵Xk×m,递归地随机分割Xk×m,采用子采样的方式构造多颗决策树iTree,将返回的iTree集合并准备进行评估;步骤4.2:将节点在所有iTree的高度平均值比值归一化得到统计量sx,k,将统计量sx,k作为异常用户的评价指标;步骤5:所有参与训练的模型参数经过孤立森林模块得到对应的异常分数后,根据验证模型的分数在所有上传模型的分数的分布,不断调整阈值;步骤6:排除标签为异常的更新参数,中央服务器接收正常的训练参数运行联邦平均算法,聚合产生新的全局模型并再次将计算好的全局模型参数广播给每个参与方,为下一轮训练做准备,然后重复步骤3至步骤6直至联邦学习全局模型收敛。
全文数据:
权利要求:
百度查询: 广西师范大学 基于孤立森林的联邦学习中毒攻击的防御方法
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。