【技术实现步骤摘要】
一种量化模型的生成方法及装置、电子设备及存储介质
[0001]本公开涉及神经网络
,尤其涉及一种量化模型的生成方法及装置、电子设备及存储介质。
技术介绍
[0002]在针对图像分类模型压缩所采取的参数量化方法中,存在以下问题:图像分类模型量化在低比特约束下,能够有效表示的数据较少,但是量化后的模型精度会大幅度丢失,其中,低比特可以是4
‑
bit。
[0003]相关技术中,采用了块重建的量化方式,使用模型中每一个块的输出结果对相应的卷积块中的参数进行更新,从而达到一个结合局部信息与全局信息的方式;还可以采用量化与知识蒸馏相结合的方法,以每一层卷积层的输出以及激活函数的输出指导量化过程。
[0004]然而,在量化过程中,只采用逐层或者逐块的量化方式,没有较好的权衡局部信息与全局信息对量化造成的影响,从而导致精度丢失严重。
[0005]因此,在深度神经网络模型量化过程中,减少在低比特约束下丢失的精度是一个亟需解决的问题。
技术实现思路
[0006]为了解决上述技术问题或者至少部分地解决上述技术问题,本公开的实施例提供了一种量化模型的生成方法及装置、电子设备及存储介质。
[0007]第一方面,本公开的实施例提供了一种量化模型的生成方法,所述方法包括:对于预先训练的目标检测模型进行量化;根据所述目标检测模型每一网络层的量化前与量化后输出值的差值,确定每一网络层对目标检测模型的贡献度;对于贡献度小于预设阈值的所有网络层进行分组,确定每一组的网络层;对每一组的 ...
【技术保护点】
【技术特征摘要】
1.一种量化模型的生成方法,其特征在于,所述方法包括:对于预先训练的目标检测模型进行量化;根据所述目标检测模型每一网络层的量化前与量化后输出值的差值,确定每一网络层对目标检测模型的贡献度;对于贡献度小于预设阈值的所有网络层进行分组,确定每一组的网络层;对每一组的网络层和贡献度大于或等于预设阈值的每一网络层,进行并行量化,得到与所述目标检测模型对应的量化模型。2.根据权利要求1所述的方法,其特征在于,通过以下表达式,根据所述目标检测模型每一网络层的量化前与量化后输出值的差值,确定每一网络层对目标检测模型的贡献度:其中,S为当前网络层对目标检测模型的贡献度,为当前网络层量化前的第i个输出值,为当前网络层量化后的第i个输出值。3.根据权利要求1所述的方法,其特征在于,所述对于预先训练的目标检测模型进行量化,包括:对于预先训练的目标检测模型的每一网络层的每一个权重因子,从当前比特位调整为目标比特位,其中,所述目标比特位小于当前比特位。4.根据权利要求1所述的方法,其特征在于,所述对于贡献度小于预设阈值的所有网络层进行分组,确定每一组的网络层,包括:按照网络层在目标检测模型中的先后顺序,对贡献度小于预设阈值的所有网络层进行排序,得到排序后网络层;对于排序后网络层进行分组,确定每一组的网络层,其中,每一组的网络层的个数为预设值。5.根据权利要求1所述的方法,其特征在于,所述对每一组的网络层和贡献度大于或等于预设阈值的每一网络层,进行并行量化,包括:对贡献度大于或等于预设阈值的每一网络层分别分配一个并行线程,将贡献度大于或等于预设阈值的每一网络层作为第一目标网络层进行量化;对每一组的网络层分别分配一个并行线程,将每一组的网络层中的每一网络层作为第二目标网络层进行量化。6.根据权利要求5所述的方法,其特征在于,所述将每一组的网络层中的每一网络层作为第二目标网络层进行量化,包括:将每一组的网络层中的每一网络层作为第二目标网络层;按照每一组的网络层在目标检测模型中的先后顺序,依次对第二目标网络层进行量化。7.根据权利要求6所述的方法,其特征在于,所述对第二目标网络层进行量化,包括:调整第二目标网络层的所有已知待量化参数取值范围的最值,并根据多组调整后最值分别对第二目标网络层进行量化;在第二目标网络层为目标检测模型的第M网络层的情况下,对于用于量化第M网络层的
每一组调整后最值,按照网络层在目标检测模型中的先后顺序,依次计算第M网络层至第三目标网络层中,每一网络层在第M网络层量化前和量化后的输出值,其中,所述第三目标网络层为排序后网络层中第M网络层后面的第一个网络层;根据第M网络层至第三目标网络层中,每一网络层在第M网络层量化前和量化后的输出值,分别计算第M网络层至第三目标网络层中每一网络层的损失值;按照网络层在目标检测模型中的先后顺序,对第M网络层至第三目标网络层中每一网络层的损失值进行排序,根据排序后的损失值确定损失值拐点;将损失值拐点作为当前组调整后最值对第M网络层的量化过程的终止点;对比每一组调整后最值对应的损失值拐点对应的损失值,将损失值最小的调整后最值作为目标最值。8.根据权利要求5所述的方法,其特征在于,所述将贡献度大于或等...
【专利技术属性】
技术研发人员:刘艳,林金辉,王恒,石宇航,孙梦磊,杨思琪,
申请(专利权)人:北京科技大学,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。