一种基于强化网络剪枝的轴承故障诊断网络压缩方法

本发明涉及故障诊断神经网络自主压缩方法，更具体的说是一种基于强化网络剪枝的轴承故障诊断网络压缩方法，包括以下核心步骤：采集轴承的原始故障信号，并进行预处理；选择故障诊断神经网络作为待剪枝网络；构建智能体模型，智能体模型包含一个动作网络和一个评价网络，用以观测剪枝过程中环境的实时状态信息，根据状态信息输出网络各层的剪枝率作为当前剪枝策略，并评估当前剪枝策略的价值；构建待剪枝网络剪枝环境，用于实现网络剪枝动作、将当前最佳剪枝策略优化后的网络进行网络再训练及验证每次剪枝后网络的分类性能；基于DDPG算法优化智能体网络参数和迭代剪枝策略，对待剪枝网络进行循环剪枝。