[发明专利]基于多任务预测模型的数据处理方法、装置、设备和介质在审
申请号: | 202110257360.X | 申请日: | 2021-03-09 |
公开(公告)号: | CN112990289A | 公开(公告)日: | 2021-06-18 |
发明(设计)人: | 刘颖;徐进;解鑫;许铭;刘建林 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
主分类号: | G06K9/62 | 分类号: | G06K9/62;G06K9/46;G06N3/04;G06N3/08 |
代理公司: | 北京品源专利代理有限公司 11332 | 代理人: | 孟金喆 |
地址: | 100085 北京市*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 基于 任务 预测 模型 数据处理 方法 装置 设备 介质 | ||
1.一种基于多任务预测模型的数据处理方法,其中,所述多任务预测模型包括输入层、至少一层特征共享层和输出层,所述特征共享层包括至少两个特征提取模块、特征拼接模块、和至少两个特征融合模块,所述输出层包括至少两个预测模块;所述方法包括:
通过所述输入层对产品至少两个尺寸维度的原始数据分别执行预处理操作,以得到至少两组标准原始数据;
将至少两组所述标准原始数据分别输入至少两个所述特征提取模块,对所述标准原始数据进行特征提取,以得到至少两组尺寸特征数据;
将至少两组所述尺寸特征数据输入所述特征拼接模块进行拼接,以得到特征拼接数据;
将所述特征拼接数据分别输入至少两个所述特征融合模块,分别对所述特征拼接数据进行特征融合,以得到至少两组特征融合数据;
将至少两组所述特征融合数据分别输入至少两个所述预测模块,以根据所述特征融合数据确定所述产品每个尺寸维度的预测结果。
2.根据权利要求1所述的方法,其中,所述特征提取模块的网络结构为全连接神经网络,每个所述特征提取模块对所述标准原始数据进行特征提取具体包括:
每个所述特征提取模块通过如下公式得到尺寸特征数据:
F1=σ1(W1x1+b1)
其中,F1表示尺寸特征数据,x1表示标准原始数据,W1表示所述特征提取模块的网络权重,b1表示所述特征提取模块的网络偏置项,σ1()表示所述特征提取模块的激活函数。
3.根据权利要求1所述的方法,其中,将至少两组所述尺寸特征数据输入所述特征拼接模块进行拼接,以得到特征拼接数据包括:
将至少两组所述尺寸特征数据输入所述特征拼接模块进行矩阵拼接,得到特征拼接数据。
4.根据权利要求1所述的方法,其中,各所述特征融合模块之间的模型参数不同。
5.根据权利要求4所述的方法,其中,所述特征融合模块的网络结构为全连接神经网络,每个所述特征融合模块对所述特征拼接数据进行特征融合具体包括:
通过如下公式得到特征融合数据:
F2=σ2(W2x2+b2)
其中,F2表示特征融合数据,x2表示特征拼接数据,W2表示所述特征融合模块的网络权重,b2表示所述特征融合模块的网络偏置项,σ2()表示所述特征融合模块的激活函数。
6.根据权利要求1所述的方法,其中,所述预测模块的网络结构为全连接神经网络,每个所述预测模块根据所述特征融合数据确定所述产品每个尺寸维度的预测结果具体包括:
通过如下公式确定所述产品每个尺寸维度的预测结果:
F3=σ3(W3x3+b3)
其中,F3表示所述产品对应每个尺寸维度的预测结果,x3表示特征融合数据,W3表示所述预测模块的网络权重,b3表示所述预测模块的网络偏置项,σ3()表示所述预测模块的激活函数。
7.根据权利要求1所述的方法,其中,所述特征共享层还包括至少两个特征提取辅助模块和至少两个特征求和模块,所述方法还包括:
将至少两组所述标准原始数据分别输入至少两个所述特征提取辅助模块进行特征提取,以得到至少两组辅助特征数据;
将所述特征融合数据与同一尺寸维度的辅助特征数据输入所述特征求和模块相加,以得到的特征求和数据更新所述特征融合数据。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202110257360.X/1.html,转载请声明来源钻瓜专利网。