公开(公告)号：CN116992944B

公开(公告)日：2023-12-19

申请号：CN202311257199.1

申请日：2023-09-27

Applicant: 之江实验室

Inventor： 李超 ,  陈启运 ,  刁博宇 ,  宫禄齐

IPC: G06N3/082 ,  G06N3/0464 ,  G06N3/084 ,  G06V10/82

公开(公告)号：CN116992944A

公开(公告)日：2023-11-03

申请号：CN202311257199.1

申请日：2023-09-27

Applicant: 之江实验室

Inventor： 李超 ,  陈启运 ,  刁博宇 ,  宫禄齐

IPC: G06N3/082 ,  G06N3/0464 ,  G06N3/084 ,  G06V10/82

Abstract: 本发明公开了一种基于可学习重要性评判标准剪枝的图像处理方法及装置，属于人工智能模型压缩与推理加速领域，该方法包括：获取待剪枝的图像处理模型；在所述待剪枝的图像处理模型的前向传播中，加入重要性评判标准搜索空间，其中所述重要性评判标准搜索空间用于计算所述待剪枝的图像处理模型中每一层卷积层需要剪枝的部分；基于所述重要性评判标准搜索空间进行反向传播并更新所述重要性评判标准搜索空间的参数；对基于最终评判标准剪枝后的模型进行微调。