[发明专利]一种基于通道剪枝的深度卷积神经网络模型的定位方法在审
| 申请号: | 202010548786.6 | 申请日: | 2020-06-16 |
| 公开(公告)号: | CN111814953A | 公开(公告)日: | 2020-10-23 |
| 发明(设计)人: | 赵宇;鲁敏;周斌;卜智勇 | 申请(专利权)人: | 上海瀚讯信息技术股份有限公司 |
| 主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/08;H04W4/02;H04W4/021;H04W4/33;H04W64/00 |
| 代理公司: | 上海泰能知识产权代理事务所 31233 | 代理人: | 宋缨 |
| 地址: | 200335 上海市*** | 国省代码: | 上海;31 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 一种 基于 通道 剪枝 深度 卷积 神经网络 模型 定位 方法 | ||
1.一种基于通道剪枝的深度卷积神经网络模型的定位方法,其特征在于,包括:
步骤(1):构造基于深度卷积神经网络模型的位置定位基线识别模型;
步骤(2):获取待定位区域中移动设备的无线电信号数据;
步骤(3):将获取到的无线电信号数据进行预处理,得到无线电信号数据集;利用所述无线电信号数据集对所述位置定位基线识别模型进行训练和验证,得到预训练深度卷积神经网络模型;
步骤(4):对所述预训练深度卷积神经网络模型进行通道剪枝,得到压缩的深度卷积神经网络模型;
步骤(5):评估所述压缩的深度卷积神经网络模型的性能,并通过精确率、召回率和准确率来评估所述的压缩的深度卷积神经网络模型在预设类别下的定位性能。
2.根据权利要求1所述的基于通道剪枝的深度卷积神经网络模型的定位方法,其特征在于,所述位置定位基线识别模型中预设有待定位区域的历史信息。
3.根据权利要求1所述的基于通道剪枝的深度卷积神经网络模型的定位方法,其特征在于,所述无线电信号数据集包括数据集、验证集和测试集。
4.根据权利要求1所述的基于通道剪枝的深度卷积神经网络模型的定位方法,其特征在于,所述步骤(4)还包括:
步骤(41):将所述预训练深度卷积神经网络模型中所有卷积层重新编号并排序,并统计所有卷积层的通道数;计算每个卷积层的所有通道性能参数α,公式为:
其中,表示第l层卷积层中第n号的滤波器的权重,||·||2表示向量的L2范数,Nl表示第l层卷积层中滤波器的个数;
步骤(42):将所述预训练深度卷积神经网络模型的剪枝个数设为P,对所有卷积层的通道性能参数α从大到小进行排序,选取通道性能参数最小的P个通道,记录并返回所述P个通道的卷积层号和通道编号;
步骤(43):根据拟剪枝列表mask去除所述通道性能参数最小的P个通道及其对应的卷积核;
步骤(44):对剪枝后的预训练深度卷积神经网络模型进行全网参数更新;
步骤(45):判断剪枝是否结束,若剪枝结束,则使用所述无线电信号数据集来微调模型,微调结束后得到压缩的深度卷积神经网络模型并保存;若剪枝未结束,则重复所述步骤(41)至步骤(44)。
5.根据权利要求4所述的基于通道剪枝的深度卷积神经网络模型的定位方法,其特征在于,所述步骤(43)还包括:所有卷积层剪枝结束后,通过BatchNorm层的继承方式将被去除通道的前后通道相连,使被去除通道的卷积层的输出通道数与下一个卷积层的输入通道数相同。
6.根据权利要求4所述的基于通道剪枝的深度卷积神经网络模型的定位方法,其特征在于,所述步骤(43)还包括:最后一个卷积层的输出通道为分类数目,无需剪枝。
7.根据权利要求4所述的基于通道剪枝的深度卷积神经网络模型的定位方法,其特征在于,所述步骤(44)具体为:创建关于被去除卷积核的第i层卷积层和第i+1层卷积层的新权重参数,并将未被去除的卷积核的权重参数复制到所述剪枝后的预训练深度卷积神经网络模型中,完成参数更新。
8.根据权利要求1所述的基于通道剪枝的深度卷积神经网络模型的定位方法,其特征在于,所述步骤(5)具体为:所述压缩的深度卷积神经网络模型预测准确率的公式为:
其中,Ntrue表示预测正确的样本个数,Ntest表示测试集样本总数;
设测试数据为X={x,y},x表示所述测试数据的特征向量,y表示所述测试数据的真实标签,则所述压缩的深度卷积神经网络模型在预设类别下的定位性能通过精确率、召回率和准确率来表示,公式为:
其中,Precision为精确率,Recall为召回率,Acc为准确率;TP为真正类,表示为y=k,y'=k;TN为真负类,表示为y≠k,y'≠k;FP为假正类,表示为y≠k,y'=k;FN为假负类,表示为y=k,y'≠k,y'表示所述压缩的深度卷积神经网络输出的预测标签,k表示预设类别。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于上海瀚讯信息技术股份有限公司,未经上海瀚讯信息技术股份有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010548786.6/1.html,转载请声明来源钻瓜专利网。





