一种模型剪枝方法、装置、电子设备和存储介质

本发明公开了一种模型剪枝方法、装置、电子设备和存储介质。其中，该方法包括：对嵌入了混合注意力模块的目标剪枝模型进行模型训练，得到中间决策模型；根据中间决策模型中第二目标剪枝层的通道注意力权重和空间注意力权重，对目标剪枝模型中的第一目标剪枝层进行剪枝处理，得到目标剪枝模型对应的目标剪枝优化模型；其中，第一目标剪枝层和第二目标剪枝层一一对应。本方案通过在目标剪枝模型中嵌入通道注意力模块和空间注意力模块，综合考虑通道和空间位置两个维度的信息，能够更准确地识别出对模型精度至关重要的通道和空间位置，同时保留了更广泛的上下文特征关系，实现更全面且更精确的剪枝效果，从而有效保证剪枝后的模型精度。