[发明专利]一种基于CNN的低精度训练与8位整型量化推理方法在审
申请号: | 201910154088.5 | 申请日: | 2019-03-01 |
公开(公告)号: | CN109902745A | 公开(公告)日: | 2019-06-18 |
发明(设计)人: | 严敏佳;王永松;刘丹 | 申请(专利权)人: | 成都康乔电子有限责任公司;电子科技大学 |
主分类号: | G06K9/62 | 分类号: | G06K9/62;G06K9/00;G06N3/08;G06N3/04;G06N5/04 |
代理公司: | 暂无信息 | 代理人: | 暂无信息 |
地址: | 610000 四川省成都市*** | 国省代码: | 四川;51 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 量化 推理 整型 浮点 权重 定点算法 整型数据 卷积神经网络 精度模型 精度损失 模型训练 目标检测 权重参数 算法训练 推理过程 训练模型 低位 卷积 前向 激活 输出 优化 | ||
1.一种基于CNN的低精度训练与8位整型量化推理方法,其特征在于,包括如下步骤:
低精度模型训练;用16位浮点型的低精度定点算法进行模型训练,得到用于目标检测的模型;
量化权重;提出8位整型量化方案,按层将卷积神经网络的权重参数从16位浮点型量化为8位整型;
8位整型量化推理;把激活值量化为8位整型数据,即CNN的每一层都接受int8类型量化输入并生成int8量化输出。
2.根据权利要求1所述的一种基于CNN的低精度训练与8位整型量化推理方法,其特征在于:低精度模型训练包括在大型服务器端用GPU加速来训练模型,计算过程中用16位浮点型保存数据。
3.根据权利要求2所述的一种基于CNN的低精度训练与8位整型量化推理方法,其特征在于:计算过程中的16位浮点型数据中,用2位保存整数部分,用14位保存小数部分,用四舍五入保留浮点型数据的14位精度。
4.根据权利要求1所述的一种基于CNN的低精度训练与8位整型量化推理方法,其特征在于,量化权重包括提出量化方案:,其中x表示浮点型数据,分别表示待量化数组中数据的最小值、最大值,即,
再通过四舍五入函数Round()取整得到量化值q。
5.根据权利要求4所述的一种基于CNN的低精度训练与8位整型量化推理方法,其特征在于:把权重按层分成一系列数组,求取每一个权重数组的最值,将同一个数组内的数据等比例放缩成8位的整数。
6.根据权利要求1所述的一种基于CNN的低精度训练与8位整型量化推理方法,其特征在于,8位整型量化推理过程包括如下步骤:
(a)BN算法预处理;在卷积运算之前计算输入样本的均值和方差,进行归一化预处理;
(b)整型卷积运算;用8位整型量化方案将浮点型乘法运算转化为整型乘法运算;
(c)优化激活函数;逐层选取每一个卷积层的激活区域[a,b],优化激活函数来把卷积运算结果映射到已知区域[a,b]。
7.根据权利要求1至6任一种基于CNN的低精度训练与8位整型量化推理方法,其特征在于:在推理过程中采用全整型数据计算。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于成都康乔电子有限责任公司;电子科技大学,未经成都康乔电子有限责任公司;电子科技大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910154088.5/1.html,转载请声明来源钻瓜专利网。