[发明专利]基于多教师知识蒸馏和归一化注意力的岩石类别预测方法在审
申请号: | 202211022055.3 | 申请日: | 2022-08-25 |
公开(公告)号: | CN115393671A | 公开(公告)日: | 2022-11-25 |
发明(设计)人: | 秦宗韬;和子杰;张宇韬;王源静;蒋若男 | 申请(专利权)人: | 河海大学 |
主分类号: | G06V10/774 | 分类号: | G06V10/774;G06V10/82;G06N3/04;G06N3/08 |
代理公司: | 南京千语知识产权代理事务所(普通合伙) 32394 | 代理人: | 尚于杰 |
地址: | 210000 *** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种基于多教师知识蒸馏和归一化注意力的岩石类别预测方法,包括S1:采集岩石样品图像,建立岩石数据集,并对原始岩石样品图像数据进行预处理;S2:将在原有的网络结构基础上插入基于归一化的注意力模块,构建轻量级神经网络;S3:采用多教师知识蒸馏方法,即采用以Swin‑Transformer为代表的自注意力模块和以ResNet为代表的卷积模型,训练出轻量级学生模型;S4:将数据集分为训练集和测试集,记录分类结果和各项性能指标。本发明以MobileNetV3为学生模型,结合CNN和Self‑Attetion两种范式的优点,训练出一个性能更好的轻量级模型,通过批量归一化得到的缩放因子,通过缩放因子反映出各个通道的变化的大小,也表示了该通道的重要性,使得不需要额外的参数就能实现注意力机制。 | ||
搜索关键词: | 基于 教师 知识 蒸馏 归一化 注意力 岩石 类别 预测 方法 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于河海大学,未经河海大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202211022055.3/,转载请声明来源钻瓜专利网。