[发明专利]一种ViT网络模型的剪枝方法及其系统在审
申请号: | 202210879913.X | 申请日: | 2022-07-25 |
公开(公告)号: | CN115249061A | 公开(公告)日: | 2022-10-28 |
发明(设计)人: | 张天魁;徐邦国;陈泽仁;李娟;朱禹涛;李俊涛;李照波;陈星 | 申请(专利权)人: | 北京邮电大学;江西鑫铂瑞科技有限公司 |
主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/08 |
代理公司: | 北京卓特专利代理事务所(普通合伙) 11572 | 代理人: | 王治兰 |
地址: | 100876 *** | 国省代码: | 北京;11 |
权利要求书: | 暂无信息 | 说明书: | 暂无信息 |
摘要: | 本申请公开了一种ViT网络模型的剪枝方法及其系统,其中ViT网络模型的剪枝方法具体包括以下步骤:对ViT网络模型的每一层的稀疏化比率进行初始化;响应于完成稀疏化比率的初始化,进行ViT网络模型的初始修剪;响应于完成初始修剪,对ViT网络模型进行蒸馏训练;对训练完成的ViT网络模型进行再次修剪,完成剪枝。本申请对传统ViT网络模型的结构进行了优化,具体引入重要性分数判决模块能够快速得到修剪维度的重要性,相较于之前的基于L1范数这一标准,本发明得到的判决标准更加合理,能够保证修剪的部分为模型不重要的部分,提高了修剪的精准度。 | ||
搜索关键词: | 一种 vit 网络 模型 剪枝 方法 及其 系统 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京邮电大学;江西鑫铂瑞科技有限公司,未经北京邮电大学;江西鑫铂瑞科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202210879913.X/,转载请声明来源钻瓜专利网。