公开(公告)号：CN113408709A

公开(公告)日：2021-09-17

申请号：CN202110785452.5

申请日：2021-07-12

Applicant: 浙江大学

Inventor： 周泓 ,  杨涛 ,  楼震宇

IPC: G06N3/04 ,  G06N3/08 ,  G06K9/62

Abstract: 本发明公开了一种基于单元重要度的条件计算方法，包含：S1：预先训练主干残差网络M；S2：构建门控网络G；S3：计算所述主干残差网络M中每个所述残差单元对每一张输入图像的重要度；S4：将所述输入图像及其对应的各所述残差单元的重要度组成为输入‑标签对，构建数据集，通过所述数据集训练所述门控网络G；S5：对所述主干残差网络M进行微调以适应动态裁剪；S6：重复步骤S3‑S5直到模型的裁剪率和精度满足预设条件。本发明的基于单元重要度的条件计算方法，计算主干残差网络M中每个残差单元对每一张输入图像的重要度，并以此构建数据集用于训练门控网络G，使门控网络G能够根据输入图像与中间特征图，预测出不同残差单元的重要度。

公开(公告)号：CN113408709B

公开(公告)日：2023-04-07

申请号：CN202110785452.5

申请日：2021-07-12

Applicant: 浙江大学

Inventor： 周泓 ,  杨涛 ,  楼震宇

IPC: G06N3/0464 ,  G06N3/048 ,  G06N3/0442 ,  G06N3/08 ,  G06F18/2415

Abstract: 本发明公开了一种基于单元重要度的条件计算方法，包含：S1：预先训练主干残差网络M；S2：构建门控网络G；S3：计算所述主干残差网络M中每个所述残差单元对每一张输入图像的重要度；S4：将所述输入图像及其对应的各所述残差单元的重要度组成为输入‑标签对，构建数据集，通过所述数据集训练所述门控网络G；S5：对所述主干残差网络M进行微调以适应动态裁剪；S6：重复步骤S3‑S5直到模型的裁剪率和精度满足预设条件。本发明的基于单元重要度的条件计算方法，计算主干残差网络M中每个残差单元对每一张输入图像的重要度，并以此构建数据集用于训练门控网络G，使门控网络G能够根据输入图像与中间特征图，预测出不同残差单元的重要度。