[发明专利]基于注意力的模型训练方法、装置及电子设备在审
申请号: | 202210102176.2 | 申请日: | 2022-01-27 |
公开(公告)号: | CN114494814A | 公开(公告)日: | 2022-05-13 |
发明(设计)人: | 王健;韩钧宇 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
主分类号: | G06V10/82 | 分类号: | G06V10/82;G06N3/04;G06N3/08 |
代理公司: | 北京银龙知识产权代理有限公司 11243 | 代理人: | 尹倩 |
地址: | 100085 北京市*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本公开提供了一种基于注意力的模型训练方法、装置及电子设备,涉及人工智能技术领域,具体为深度学习、计算机视觉技术领域,可应用于图像处理、图像检测等场景。具体实现方案为:获取神经网络模型中注意力模块的注意力输出矩阵,基于神经网络模型的池化层对所述注意力输出矩阵的样本维度和数据块维度进行降维计算,确定池化后的第一输出矩阵;基于神经网络模型的卷积层对第一输出矩阵进行卷积操作,确定卷积后的第二输出矩阵,对第二输出矩阵中各个头的输出值进行归一化处理和加权处理,获得更新后的第二输出矩阵;基于更新后的第二输出矩阵获取更新后的注意力输出矩阵,并基于所述更新后的注意力输出矩阵训练所述神经网络模型。 | ||
搜索关键词: | 基于 注意力 模型 训练 方法 装置 电子设备 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202210102176.2/,转载请声明来源钻瓜专利网。