用于神经网络模型的剪枝方法及装置、电子设备

本申请涉及智能家居技术领域，公开一种用于神经网络模型的剪枝方法，包括：确定每个网络单元在前向传播阶段对神经网络模型性能的第一贡献度；确定每个网络单元在模型训练阶段对神经网络模型性能的第二贡献度；根据第一贡献度和/或第二贡献度，确定每个网络单元对神经网络模型性能的目标贡献度，目标贡献度表示网络单元对神经网络模型性能的贡献大小；根据每个网络单元对神经网络模型性能的目标贡献度，确定需要剪枝的目标网络单元，并进行剪枝操作以获得新的神经网络模型。该方法能够最大程度保持模型性能甚至优化模型性能。本申请还公开一种用于神经网络模型的剪枝装置及电子设备。