(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210880632.6 (22)申请日 2022.07.25 (71)申请人 美的集团 （上海） 有限公司 地址 201700 上海市青浦区盈 港东路168号 9幢2层20 5室 申请人 美的集团股份有限公司 (72)发明人 刘宁　车正平　黄尧民　刘新妹　 徐志远　张法朝　奉飞飞　唐剑　 (74)专利代理 机构 北京路浩知识产权代理有限 公司 11002 专利代理师 陈威 (51)Int.Cl. G06N 3/08(2006.01) G06N 3/04(2006.01) G06F 16/36(2019.01)G06V 10/80(2022.01) G06V 10/82(2022.01) (54)发明名称 模型剪枝方法及装置 (57)摘要 本申请涉及人工智能技术领域， 提供一种模 型剪枝方法及装置。 方法包括： 获取初始检测模 型和样本点云数据； 将所述样 本点云数据输入所 述初始检测模型进行稀 疏训练， 确定所述初始检 测模型的通道重要性分数； 其中， 所述通道重要 性分数基于所述样本点云数据的采样点特征、 所 述样本点云数据的未采样点特征和所述样本点 云数据的采样点坐标特征确定； 基于所述通道重 要性分数对 所述初始检测模型进行剪枝， 确定目 标检测模型。 本申请提出一种基于空间信息和特 征融合的剪枝重要性指示器， 设计了一种知 识再 利用的剪枝方式， 通过未采样点来增强剪枝 方法 的鲁棒性。 权利要求书2页 说明书11页 附图4页 CN 115169556 A 2022.10.11 CN 115169556 A 1.一种模型剪枝方法， 其特 征在于， 包括： 获取初始检测模型和样本点云数据； 将所述样本点云数据输入所述初始检测模型进行稀疏训练， 确定所述初始检测模型的 通道重要性分数； 其中， 所述通道重要性分数基于所述样本点云数据的采样点特征、 所述样本点云数据 的未采样点特 征和所述样本点云数据的采样点 坐标特征确定； 基于所述 通道重要性分数对所述初始检测模型进行剪枝， 确定目标检测模型。 2.根据权利要求1所述的模型剪枝方法， 其特 征在于， 构建所述采样点 坐标特征， 包括： 对所述样本点云数据进行降采样， 确定采样点； 对所述采样点进行 特征提取， 确定所述样本点云数据的采样点特 征； 对所述采样点的坐标信息进行 特征提取， 确定所述采样点的初始坐标 特征； 对所述样本点云数据的采样点特征和所述采样点的初始坐标特征进行特征融合， 确定 所述样本点云数据的采样点 坐标特征。 3.根据权利要求2所述的模型剪枝方法， 其特征在于， 所述对所述样本点云数据的采样 点特征和所述采样点的初始坐标 特征进行特征融合， 包括： 通过交叉注意力机制对所述采样点特 征和所述采样点的初始坐标 特征进行特征融合。 4.根据权利要求2所述的模型剪枝方法， 其特 征在于， 构建所述未采样点特 征， 包括： 确定所述样本点云数据中未被降采样的未采样点； 提取所述未采样点的特 征信息， 确定所述未采样点特 征。 5.根据权利要求4所述的模型剪枝方法， 其特征在于， 确定所述初始检测模型的通道重 要性分数， 包括： 对所述样本点云数据的采样点特征、 所述样本点云数据的未采样点特征和所述样本点 云数据的采样点坐标特征分别进行归一化， 确定采样点特征分数、 未采样点特征分数和采 样点坐标特征分数； 将所述采样点特征分数、 所述未采样点特征分数和所述采样点坐标特征分数进行加权 计算， 确定通道重要性分数。 6.根据权利要求1至5任一所述的模型剪枝方法， 其特征在于， 所述基于所述通道重要 性分数对所述初始检测型进行剪枝， 确定目标检测模型， 包括： 基于所述 通道重要性分数对所述初始检测模型进行通道选择， 确定目标掩膜； 基于所述目标掩膜和稀疏训练后的所述初始检测模型， 确定目标检测模型。 7.根据权利要求6所述的模型剪枝方法， 其特征在于， 所述基于所述目标掩膜和稀疏训 练后的所述初始检测模型， 确定目标检测模型， 包括： 基于所述目标掩膜对稀疏训练后的初始检测模型的压缩层梯度进行掩码， 确定掩码后 的压缩层； 对所述稀疏训练后初始检测模型的卷积层和掩码后的压缩层进行融合， 确定目标检测 模型。 8.一种电子设备， 包括存储器、 处理器及存储在存储器上并可在处理器上运行的计算 机程序， 其特征在于， 所述处理器执行所述程序时实现如权利要求1至7任一项所述的模型 剪枝方法。权　利　要　求　书 1/2 页 2 CN 115169556 A 29.一种非暂态计算机可读存储介质， 其上存储有计算机程序， 其特征在于， 该计算机程 序被处理器执行时实现如权利要求1至7任一项所述的模型剪枝方法。 10.一种计算机程序产品， 包括计算机程序， 其特征在于， 所述计算机程序被处理器执 行时实现如权利要求1至7任一项所述的模型剪枝方法。权　利　要　求　书 2/2 页 3 CN 115169556 A 3