【技术实现步骤摘要】
联邦学习建模优化方法、设备、可读存储介质及程序产品
[0001]本申请涉及金融科技(Fintech)的人工智能
,尤其涉及一种联邦学习建模优化方法、设备、可读存储介质及程序产品。
技术介绍
[0002]随着金融科技,尤其是互联网科技金融的不断发展,越来越多的技术(如分布式、人工智能等)应用在金融领域,但金融业也对技术提出了更高的要求,如对金融业对应待办事项的分发也有更高的要求。
[0003]随着计算机软件和人工智能、大数据云服务应用的不断发展,目前,在联邦学习场景中每个参与方维护一个本地模型和一个全局模型。各个参与方通过本地模型学习本地特有的知识,通过全局模型共享所有参与方的知识,然后各个参与方将本地模型和全局模型聚合在一起。但是这种方法的缺点是联邦服务器可以通过各个参与方的全局模型反推出参与方的原始数据,因此有数据泄露的风险,也即,现有的联邦学习方法存在泄露参与方数据隐私的风险。
技术实现思路
[0004]本申请的主要目的在于提供一种联邦学习建模优化方法、设备、可读存储介质及程序产品,旨在解决 ...
【技术保护点】
【技术特征摘要】
1.一种联邦学习建模优化方法,其特征在于,应用于第一设备,所述联邦学习建模优化方法包括:获取训练好的特征提取模型和分类模型,并提取第一训练样本、第一噪音数据、所述第一训练样本对应的第一真实分类标签;获取所述特征提取模型针对于所述第一训练样本生成的第一样本特征,以及待训练特征生成模型针对于所述第一噪音数据和所述第一真实分类标签生成的第二样本特征;通过所述分类模型对所述第二样本特征进行分类以及通过待训练样本区分模型对所述第一样本特征与所述第二样本特征进行样本区分,在固定所述特征提取模型和所述分类模型的情况下对所述待训练特征生成模型进行迭代优化,得到特征生成模型;将所述特征生成模型和所述分类模型发送至第二设备,以供第二设备依据各所述第一设备发送的特征生成模型,迭代优化聚合各所述特征生成模型得到的全局特征生成模型以及聚合各所述分类模型得到的全局分类模型,得到目标全局特征生成模型和目标全局分类模型;接收所述第二设备发送的目标全局特征生成模型和目标全局分类模型,并依据所述目标全局特征生成模型,对所述特征提取模型和所述目标全局分类模型进行迭代优化,得到目标特征提取模型和目标分类模型。2.如权利要求1所述联邦学习建模优化方法,其特征在于,所述通过所述分类模型对所述第二样本特征进行分类以及通过待训练样本区分模型对所述第一样本特征与所述第二样本特征进行样本区分,在固定所述特征提取模型和所述分类模型的情况下对所述待训练特征生成模型进行迭代优化,得到特征生成模型的步骤包括:通过所述分类模型对所述第二样本特征进行分类,以及通过待训练第一样本区分模型对所述第一样本特征和所述第二样本特征进行样本区分,计算第一分类损失以及样本区分损失;依据所述第一分类损失和所述样本区分损失,迭代优化所述待训练特征生成模型,得到所述特征生成模型。3.如权利要求2所述联邦学习建模优化方法,其特征在于,所述通过所述分类模型对所述第二样本特征进行分类,以及通过待训练第一样本区分模型对所述第一样本特征和所述第二样本特征进行样本区分,计算第一分类损失以及样本区分损失的步骤包括:依据所述分类模型,对所述第二样本特征进行分类,得到第一预测分类标签;依据所述第一预测分类标签和所述第一真实分类标签,计算第一分类损失;依据所述样本区分模型,分别对所述第一样本特征和所述第二样本特征进行二分类,得到二分类结果;依据所述二分类结果以及所述第一样本特征和所述第二样本特征共同对应的正负样本标签,计算所述样本区分损失。4.如权利要求1所述联邦学习建模优化方法,其特征在于,所述依据所述目标全局特征生成模型,对所述特征提取模型和所述目标全局分类模型进行迭代优化,得到目标特征提取模型和目标分类模型的步骤包括:提取第二训练样本、第二噪音数据以及所述第二训练样本对应的第二真实分类标签;获取所述特征提取模型针对于所述第二训练样本生成的本地样本特征,以及获取所述
目标全局特征生成模型针对于所述第二噪音数据和所述第二真实分类标签生成的全局样本特征;依据所述第二真实分类标签以及所述目标全局分类模型针对于所述本地样本特征生成的第二预测分类标签,计算第二分类损失;依据所述本地样本特征和所述全局样本特征之间的相似度,计算特征相似度损失;依据所述第二分类损失和所述特征相似度损失,在固定所述目标全局特征生成模型的情况下迭代优化所述特征提取模型和所述目标全局分类模型,得到所述目标特征提取模型和所述目标分类模型。5.如权利要求4所述联邦学习建模优化方法,其特征在于,所述依据所述第二分类损失和所述特征相似度损失,在固定所述目标全局特征生成模型的情况下迭代优化所述特征提取模型和所述目标全局分类模型,得到所述目标特征提取模型和所述目标分类模型的步骤包括:若所述目标全局特征生成模型和所述目标全局分类模型满足预设联邦结束条件,则依据所述第二分类损失和所述特征相似度损失,在固定所述目标全局特征生成模型的情况下迭代更新所述特征提取模型和所述目标全局分类模型,得到所述目标特征提取模型和所述目标...
【专利技术属性】
技术研发人员:康焱,吴岳洲,
申请(专利权)人:深圳前海微众银行股份有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。