首页 专利交易 科技果 科技人才 科技服务 国际服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

基于本地差分隐私的联邦学习图像分类方法 

买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!

申请/专利权人:西安电子科技大学

摘要:本发明提出一种基于本地差分隐私的联邦学习图像分类方法,实现步骤为:构建联邦学习场景模型;本地客户端获取训练样本集和测试样本集;构建基于卷积神经网络的图像分类模型;本地客户端初始化参数;本地客户端对基于卷积神经网络的图像分类模型的权重参数进行本地更新;本地客户端对权重参数进行本地差分隐私扰动并上传;中心服务器获取权值参数全局更新后的基于卷积神经网络的图像分类模型并发送;中心服务器获取训练好的基于卷积神经网络的图像分类模型;本地客户端获取图像分类结果。本发明通过对本地更新后的权重参数进行本地差分隐私扰动后,再全局更新,降低了计算存储开销,进而提高了训练效率,且适用于不同隐私保护需求场景的图像分类。

主权项:1.一种基于本地差分隐私的联邦学习图像分类方法,其特征在于,包括如下步骤:1构建联邦学习场景模型:构建包括中心服务器和I个本地客户端L={Li|1≤i≤I}的联邦学习场景模型,其中,I≥2,Li表示第i个本地客户端;2本地客户端获取训练样本集和测试样本集:每个本地客户端Li获取包含M个目标类别的N幅图像,并对每幅图像进行标注,然后将半数以上图像及其标签作为训练样本集,将剩余的图像及其标签作为测试样本集其中,M≥2,N≥500;3构建基于卷积神经网络的图像分类模型H:本地客户端L构建基于卷积神经网络的图像分类模型,卷积神经网络包括依次层叠的输入层、m个卷积层-ReLU层-池化层、输出层,其中m≥2,输出层包括依次层叠的多个全连接层;4本地客户端L初始化参数:本地客户端L初始化迭代次数为t,最大迭代次数为T,T≥50,当前基于卷积神经网络的图像分类模型为Ht,Ht的权重参数为wt,wt的参数值数量为k,k≥10000,并令t=0,Ht=H;5本地客户端L对基于卷积神经网络的图像分类模型H的权重参数进行本地更新:5a本地客户端L将从训练样本集中有放回且随机选取的c·N个训练样本作为当前基于卷积神经网络的图像分类模型Ht的输入进行前向传播,得到预测标签集合其中,c为采样因子,表示第q个训练样本对应的预测标签,1≤q≤c·N;5b本地客户端L采用交叉熵损失函数,并通过每个预测标签和其对应的真实标签计算图像分类模型Ht的损失值然后求取对Ht的权重参数wt的偏导再采用随机梯度下降法,通过将在Ht中进行反向传播的方式对wt进行更新,得到本地更新后的权重参数Δiwt;6本地客户端L对权重参数Δiwt进行本地差分隐私扰动并上传:6a每个本地客户端Li选取更新后的权重参数Δiwt中绝对值最大的θuk个参数,并求取所选取的每个参数与Δiwt中绝对值最大的值C的商,得到标准化的扰动输入值其中θu表示选择参数因子,0.01≤θu≤1,xj表示第j个扰动输入值,1≤j≤θuk;6b每个本地客户端Li对每个扰动输入值xj生成随机数ut∈[0,1],并判断是否成立,若是,从区间中均匀采集xj的扰动输出值yj,否则,从区间中均匀采集xj的扰动输出值yj,其中,A表示扰动输出的边界,ε表示隐私预算,0.1≤ε≤10;6c本地客户端L将θuk个扰动输出值组合成上传向量并将上传至中心服务器;7中心服务器获取权值参数全局更新后的基于卷积神经网络的图像分类模型并发送:中心服务器对上传向量进行聚合,得到的聚合结果实现对本地更新后的权重参数Δiwt的全局更新,并将权重参数wt经过本地更新以及全局更新后的图像分类模型发送至每个本地客户端Li;8中心服务器获取训练好的基于卷积神经网络的图像分类模型:中心服务器判断t≥T是否成立,若是,得到训练好的基于卷积神经网络的图像分类模型H',若否,令t=t+1,并执行步骤5;9本地客户端获取图像分类结果:每个本地客户端Li将测试样本集作为训练好的基于卷积神经网络的图像分类模型H'的输入进行前向传播,得到所有测试样本的预测标签。

全文数据:

权利要求:

百度查询: 西安电子科技大学 基于本地差分隐私的联邦学习图像分类方法

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。