[发明专利]一种基于联邦哈希学习的面向跨节点多模态检索方法有效
申请号: | 202211692053.5 | 申请日: | 2022-12-28 |
公开(公告)号: | CN115686868B | 公开(公告)日: | 2023-04-07 |
发明(设计)人: | 屠泽榕;杨展;龙军;费洪晓;陈云飞;李逸楠 | 申请(专利权)人: | 中南大学 |
主分类号: | G06F9/50 | 分类号: | G06F9/50;G06N20/00;G06F16/33;G06F16/53;G06F16/73 |
代理公司: | 长沙市融智专利事务所(普通合伙) 43114 | 代理人: | 欧阳迪奇 |
地址: | 410083 湖南*** | 国省代码: | 湖南;43 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 基于 联邦 学习 面向 节点 多模态 检索 方法 | ||
1.一种基于联邦哈希学习的面向跨节点多模态检索方法,其特征在于,包括:
步骤一,云中心服务器向各边缘节点下发初始化的深度哈希模型;
步骤二,云中心服务器收集各边缘节点训练一次深度哈希模型后更新的深度哈希模型参数,并对其中倒数第二层的全连接参数进行融合更新,从而获得一个统一的云中心全局参数;然后云中心服务器将云中心全局参数下发至各边缘节点来更新深度哈希模型参数,并继续循环以上过程直至达到训练停止条件,云中心服务器将最终的云中心全局参数下发至各边缘节点进行最终更新;其中各边缘节点是基于各自的更新目标函数来对深度哈希模型进行训练;
步骤三,云中心服务器接收各边缘节点所生成的哈希码以及对应的原始数据位置信息,并进行存储;其中哈希码是由各边缘节点基于最终更新后的深度哈希模型生成的;
步骤四,云中心服务器根据待检索数据,基于最终的云中心全局参数来以深度哈希模型生成待检索哈希码,然后计算待检索哈希码与存储的哈希码之间的汉明距离,然后以距离最小的哈希码对应的原始数据位置信息来提供原始数据检索结果;
所述的步骤二中,云中心全局参数是通过下式获取:
其中Ot表示云中心全局参数,M表示选择参与联邦学习的边缘节点,t表示第t轮参数更新,表示第m个边缘节点的深度哈希模型第t轮更新的倒数第二层全连接参数,表示第m个边缘节点的倒数第二层全连接参数对云中心全局参数的贡献度。
2.根据权利要求1所述的方法,其特征在于,的取值范围为0-1之间,通过下式获取:
其中nm为第m个节点样本数量,lm为第m个边缘节点中数据的类别总数。
3.根据权利要求1所述的方法,其特征在于,所述的步骤二中,云中心服务器将云中心全局参数下发至各边缘节点来更新深度哈希模型参数时,各边缘节点是基于以下方式来进行更新:
其中为模型的初始训练后参数,ρ为更新速率参数。
4.根据权利要求1所述的方法,其特征在于,所述的步骤二中,更新目标函数为:
L=L1+αL2+βL3
其中L为更新目标函数,L1为映射函数,L2为数据相似度函数,L3为不变特性函数,α和β为平衡参数。
5.根据权利要求4所述的方法,其特征在于,映射函数L1为:
其中表示第k个边缘节点的数据表征矩阵,表示第k个边缘节点中第i种模态的第j个数据的表征矩阵,c表示边缘节点中每个数据的模态总数;nk表示第k个边缘节点中的数据数量;Yk表示第k个边缘节点中数据的标签信息,lk表示第k个边缘节点数据的类别总数;V为映射参数,F为Frobenius范数。
6.根据权利要求4所述的方法,其特征在于,数据相似度函数L2为:
其中Si,j表示第i个样本和第j个样本间是否存在共同标签,Si,j=1表示二个样本存在共同标签,Si,j=0表示二个样本不存在共同标签;nk表示第k个边缘节点中的数据数量。
7.根据权利要求4所述的方法,其特征在于,不变特性函数L3为:
其中Ui表示第i种不同数据模态的数据表征矩阵,Uj表示第j种不同数据模态的数据表征矩阵,nk表示第k个边缘节点中的数据数量,F为Frobenius范数。
8.根据权利要求1所述的方法,其特征在于,所述的步骤二中,训练停止条件为深度哈希模型收敛或达到预设的训练次数。
9.根据权利要求1所述的方法,其特征在于,所述的步骤四中,待检索哈希码与存储的哈希码之间的汉明距离,是以待检索哈希码与云中心存储的哈希码进行异或计算获得的;以距离最小的哈希码对应的原始数据位置信息来提供原始数据检索结果,是根据汉明距离对哈希码进行排序,获得距离最小的多个哈希码,根据哈希码对应的原始数据位置信息返回多条原始数据。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中南大学,未经中南大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202211692053.5/1.html,转载请声明来源钻瓜专利网。