基于小波变换和改进卷积神经网络的刚性罐道故障诊断

现有刚性罐道故障诊断方法有的仅适用于小样本数据集，有的虽适用于大样本数据集，但忽略了实际工作环境中的多工况背景。基于卷积神经网络的刚性罐道故障诊断方法存在数据和运算量庞大，易产生过拟合等问题。针对上述问题，提出了一种基于小波变换和改进卷积神经网络的刚性罐道故障诊断方法。首先，在刚性罐道设置错位与间隙2种缺陷，采集多工况下提升容器振动加速度信号。其次，利用小波变换将采集的振动加速度信号转换为二维时频图像，采用试凑法最终确定经Complex Morlet小波基函数处理后的二维时频图像的时间和频率分辨率最佳。然后，通过改进卷积神经网络模型结构，即保留第1层和第5层池化层，将第2,3,4层池化层替换为小尺度卷积层，以防止过拟合现象。最后，将二维时频图像输入改进后的卷积神经网络模型。实验结果表明:(1)改进模型经过训练后，在训练集上的平均准确率为99%左右，在测试集上的平均准确率为99.5%。(2)当数据训练至200步后，改进模型的准确率达99%以上，改进模型的损失函数值趋近于0，说明改进模型收敛性能较好，模型的泛化能力得到了增强，在学习过程中对于过拟合的抑制效果明显。(3)在验证集混淆矩阵上，...