全网唯一标准王
(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210328266.3 (22)申请日 2022.03.31 (71)申请人 中国石油大 学 (华东) 地址 266580 山东省青岛市黄岛区长江西 路66号 (72)发明人 刘宝弟 兴雷 邵帅 刘伟锋  袁新安 李伟 葛新民  (51)Int.Cl. G06V 10/764(2022.01) G06V 10/80(2022.01) G06V 10/44(2022.01) G06K 9/62(2022.01) G06N 3/08(2006.01) (54)发明名称 基于融合自监督的特征提取器的生成方法 及装置 (57)摘要 本发明涉及模式识别技术领域, 尤其涉及一 种基于融合自监督的特征提取器的生成方法及 装置, 旨在解决现有的特征提取器采用单一的自 监督方法, 容易引起归纳偏差的问题。 该基于融 合自监督的特征提取器的生 成方法及装置, 提出 了一种新的自监督学习方法 ‑融合自监督学习, 通过两种自监督机制分别训练特征提取器, 得到 基于旋转的自监督特征提取器和基于镜像的自 监督特征提取器, 从而挖掘更多有意义的信息; 通过多个维度对两个特征提取器(即基于旋转的 自监督特征提取器和基于镜像的自监督特征提 取器)进行学习, 融合得到最终的特征提取器, 减 少单一自监督方法引起的表征偏 差, 丰富和加强 表征信息, 为下游分类任务提供强大的特征表 示。 权利要求书3页 说明书8页 附图1页 CN 114758169 A 2022.07.15 CN 114758169 A 1.一种基于融合自监 督的特征提取器的生成方法, 其特 征在于, 所述方法包括: 使用resnet ‑12网络作为特征提取网络, 设计基于旋转的自监督特征提取器fRFE(·), 其中, 基于 旋转的自监 督特征提取器fRFE(·)的自监督损失函数LRFE为: LRFE=Lc+Lr 其中, Lc表示标准分类损失; Lr表示辅助旋转损失; 其中, 标准分类损失Lc为: 其中, c表示第c类样本; y(c,x)表示样本真实标签; p(c,x)表示样本的预测标签属于第c类 的概率; 其中, 辅助旋转损失Lr为: 其中, r={0 °,90°,180°,270°}表示以不同角度旋转的样本标签; y(r,x)表示真值标签; p(r,x)表示样本的预测标签属于第r类的概 率。 使用resnet ‑12网络作为特征提取网络, 设计基于镜像 的自监督特征提取器fMFE(·), 其中, 基于 镜像的自监 督特征提取器fMFE(·)的自监督损失函数LMFE为: LMFE=Lc+Lm 其中, Lc表示标准分类损失; Lm表示辅助镜像损失; 其中, 辅助镜像损失Lm为: 其中, m表示 m类数据, m={水平 镜像、 垂直镜像、 对角镜像}; y(m,x)表示真值标签; p(m,x)表 示样本的预测标签属于第m类的概 率。 通过多个维度对两个特征提取器(即基于旋转的自监督特征提取器和基于镜像的自监 督特征提取器)进行学习, 融合得到最终的特征提取器h( ·), 其中, h( ·)的训练损失函数 (即多维融合损失函数)为: LMDF=α LFA+β LSA+γLLA 其中, LFA表示特征对齐损失; LSA表示语义对齐损失; LLA表示标签对齐损失; α、 β和γ表 示超参数。 2.根据权利要求1所述的方法, 其特 征在于, 所述特 征对齐损失LFA为: LFA=||h(x)‑fRFE(x)||2+||h(x)‑fMFE(x)||2 其中, h(x), fRFE(x)和fMFE(x)分别表示由h( ·)、 fRFE(·)和fMFE(·)获得的样本x的特 征; ||·||2表示(·)的2范数正则化。 3.根据权利要求2所述的方法, 其特 征在于, 所述语义对齐损失LSA为: 权 利 要 求 书 1/3 页 2 CN 114758169 A 2其中, i表示第i类样本; 分别表示fRFE(x), fMFE(x), h(x)在蒸馏温度T下的softmax输出。 4.根据权利要求3所述的方法, 其特 征在于, 所述标签对齐损失LLA为: 其中, i表示第i类样本; y(i,x)表示样本真实标签; p(i,x)表示样本的预测标签属于第i类 的概率。 5.一种基于融合自监 督的特征提取器的生成装置, 其特 征在于, 所述装置包括: 第一自监督特征提取器生成模块, 用于使用resnet ‑12网络作 为特征提取网络, 设计基 于旋转的自监督特征提取器fRFE(·), 其中, 基于旋转的自监督特征提取器fRFE(·)的自监 督损失函数LRFE为: LRFE=Lc+Lr 其中, Lc表示标准分类损失; Lr表示辅助旋转损失; 其中, 标准分类损失Lc为: 其中, c表示第c类样本; y(c,x)表示样本真实标签; p(c,x)表示样本的预测标签属于第c类 的概率; 其中, 辅助旋转损失Lr为: 其中, r={0 °,90°,180°,270°}表示以不同角度旋转的样本标签; y(r,x)表示真值标签; p(r,x)表示样本的预测标签属于第r类的概 率。 第二自监督特征提取器生成模块, 用于使用resnet ‑12网络作 为特征提取网络, 设计基 于镜像的自监督特征提取器fMFE(·), 其中, 基于镜像的自监督特征提取器fMFE(·)的自监 督损失函数LMFE为: LMFE=Lc+Lm 其中, Lc表示标准分类损失; Lm表示辅助镜像损失; 其中, 辅助镜像损失Lm为: 其中, m表示 m类数据, m={水平 镜像、 垂直镜像、 对角镜像}; y(m,x)表示真值标签; p(m,x)表 示样本的预测标签属于第m类的概 率。 最终特征提取器生成模块, 用于通过多个维度对两个特征提取器(即基于旋转的自监 督特征提取器和基于镜像的自监督特征提取器)进行学习, 融合得到最终的特征提取器h (·), 其中, h( ·)的训练损失函数(即多维融合损失函数)为: LMDF=α LFA+β LSA+γLLA权 利 要 求 书 2/3 页 3 CN 114758169 A 3

PDF文档 专利 基于融合自监督的特征提取器的生成方法及装置

文档预览
中文文档 13 页 50 下载 1000 浏览 0 评论 0 收藏 3.0分
温馨提示:本文档共13页,可预览 3 页,如浏览全部内容或当前文档出现乱码,可开通会员下载原始文档
专利 基于融合自监督的特征提取器的生成方法及装置 第 1 页 专利 基于融合自监督的特征提取器的生成方法及装置 第 2 页 专利 基于融合自监督的特征提取器的生成方法及装置 第 3 页
下载文档到电脑,方便使用
本文档由 SC 于 2024-03-03 12:06:52上传分享
友情链接
站内资源均来自网友分享或网络收集整理,若无意中侵犯到您的权利,敬请联系我们微信(点击查看客服),我们将及时删除相关资源。