一种深度智能故障诊断模型动态两极化剪枝方法

摘要

本发明提供了一种深度智能故障诊断模型动态两极化剪枝方法。该方法包括：根据诊断任务的实际情况设计机械设备智能故障诊断模型，确定机械设备智能故障诊断模型中需要剪枝的网络层；将两极化剪枝模块嵌入机械设备智能故障诊断模型，并与需要剪枝的网络层连接，将需要剪枝的网络层的输出特征图作为两极化剪枝模块的输入；利用机械设备故障诊断数据集对机械设备智能故障诊断模型和两极化剪枝模块进行同步训练，两极化剪枝模块输出经过剪枝和通道增强的所述需要剪枝的网络层的特征图。本发明方法可以在诊断模型训练过程中同步地完成大网络粒度单位的动态结构化剪枝，剪枝方式为结构化剪枝而无须特定硬件设备支持；剪枝不影响诊断精度等诸多优点。