1月28日,国家知识产权局信息显示,京东科技控股股份有限公司申请一项名为“一种模型剪枝方法、装置、设备及存储介质”的专利,公开号CN 119358622 A,申请日期为2023年7月。
专利摘要显示,本发明公开了一种模型剪枝方法、装 置、设备及存储介质,涉及人工智能技术领域,该方法包括:将测试数据集输入到训练完成的待剪枝网络模型中;针对所述待剪枝网络模型中的每个网络层,获取所述网络层基于各所述测试数据分别输出的网络向量集,并基于各所述测试标签和各所述网络向量集,对未训练完成的初始分类器进行训练得到与所述网络层对应的目标分类器。
基于至少两个目标分类器分别对应的分类效果数据,对所述待剪枝网络模型执行剪枝操作得到剪枝网络模型,本发明实施例解决了传统的剪枝评估算法准确度不高的问题,在保证剪枝后的网络模型的模型性能的同时,尽可能提高了剪枝后的网络模型的推理速度。