基于VGG16迁移学习的轴承故障诊断方法

摘要

针对轴承故障诊断问题,提出基于VGG16卷积神经网络与迁移学习的故障诊断方法.首先将轴承原始振动信号数据利用信号转图像方法进行预处理,生成相应的目标数据集;然后将已经预训练过的VGG16模型在生成的目标数据集上训练并微调;最后将微调后的VGG16模型应用于故障诊断.将该方法分别在凯斯西储大学和辛辛那提大学的轴承数据集上进行验证,结果表明该方法能够取得接近100％的轴承故障诊断准确率,具有较好的应用前景.