《万方数据-数字化期刊群》全文上网期刊
CNKI《中国学术期刊(网络版)》全文收录期刊
《中文科技期刊数据库》(维普网)全文收录期刊
超星期刊域出版平台、博看网全文收录期刊
日本JST中文数据库来源期刊
美国《乌利希期刊指南(网络版)》(Ulrichsweb)收录

留言板

尊敬的读者、作者、审稿人, 关于本刊的投稿、审稿、编辑和出版的任何问题, 您可以本页添加留言。我们将尽快给您答复。谢谢您的支持!

姓名
邮箱
手机号码
标题
留言内容
验证码

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

赵宇凯 徐高威 刘敏

赵宇凯, 徐高威, 刘敏. 基于VGG16迁移学习的轴承故障诊断方法[J]. 航天器环境工程, 2020, 37(5): 446-451 doi:  10.12126/see.2020.05.005
引用本文: 赵宇凯, 徐高威, 刘敏. 基于VGG16迁移学习的轴承故障诊断方法[J]. 航天器环境工程, 2020, 37(5): 446-451 doi:  10.12126/see.2020.05.005
ZHAO Y K, XU G W, LIU M. Method for fault diagnosis of bearing based on transfer learning with VGG16 model[J]. Spacecraft Environment Engineering, 2020, 37(5): 446-451 doi:  10.12126/see.2020.05.005
Citation: ZHAO Y K, XU G W, LIU M. Method for fault diagnosis of bearing based on transfer learning with VGG16 model[J]. Spacecraft Environment Engineering, 2020, 37(5): 446-451 doi:  10.12126/see.2020.05.005

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

doi: 10.12126/see.2020.05.005
基金项目: 国家自然科学基金项目(编号:61573257);北京卫星环境工程研究所创新基金项目(编号:CAST-BISEE 2019-022)
详细信息
    作者简介:

    赵宇凯(1997—),男,硕士研究生,主要研究方向为深度学习模型在故障诊断方面的应用。E-mail: zhaoyukaijake@tongji.edu.cn

    刘   敏(1970—),男,博士学位,教授,博士生导师,从事互联网与大数据环境下智能维护与服务优化配置的研究。E-mail: lmin@tongji.edu.cn

  • 中图分类号: TP18; TH133.3

Method for fault diagnosis of bearing based on transfer learning with VGG16 model

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

    Figure  1.  Signal-to-image conversion method

    图  2  VGG16卷积神经网络结构

    Figure  2.  Structure chart of VGG16 convolutional neural network

    图  3  卷积神经网络模型迁移

    Figure  3.  Model transfer of CNN

    图  4  凯斯西储大学的轴承实验装置

    Figure  4.  The bearing test bench in Case Western Reserve University

    图  5  数据预处理

    Figure  5.  Data preprocessing

    图  6  3种诊断方法的准确率曲线

    Figure  6.  Accuracy curve for three diagnostic methods

    表  1  实验数据集

    Table  1.   Experimental data set

    编号样本类别数据集正常内圈滚动体外圈
    数据集10 HP训练集500500500500
    测试集500500500500
    1 HP训练集500500500500
    测试集500500500500
    2 HP训练集500500500500
    测试集500500500500
    3 HP训练集500500500500
    测试集500500500500
    数据集2IMS训练集500500500500
    测试集500500500500
    下载: 导出CSV

    表  2  轴承故障诊断准确率

    Table  2.   Bearing fault diagnosis results

    单位:%
    诊断方法VGG16CNNSVM
    故障类别0 HP10010096.2
    1 HP10010096.9
    2 HP10010097.5
    3 HP10099.595.7
    IMS10099.594.8
    0~1 HP10099.893.5
    0~2 HP99.994.192.8
    0~3 HP99.992.590.5
    0~3 HP+IMS99.690.487.8
    平均值99.797.393.9
    下载: 导出CSV
  • [1] 闻新, 陈镝, 乔羽. 国内神经网络故障诊断技术及其在航天器中的应用[J]. 沈阳航空航天大学学报, 2018, 35(3): 17-26

    WEN X, CHEN D, QIAO Y. Nerual network fault diagnosis technology and its application on spacecraft in China[J]. Journal of Shenyang Aerospace University, 2018, 35(3): 17-26
    [2] 陈彬强, 何昱超, 姚斌, 等. 航天器传动部件振动数据稀疏多分类智能故障诊断[J]. 航天器环境工程, 2018, 35(4): 359-364

    CHEN B Q, HE Y C, YAO B, et al. Sparsity assisted intelligent multi-classifier for fault diagnosis using vibration data of spacecraft transmission chain[J]. Spacecraft Environment Engineering, 2018, 35(4): 359-364
    [3] RANDALL R B. Vibration-based condition monitoring: industrial, automotive and aerospace applications[M]. Hoboken: Wiely, 2011: 110
    [4] 于景洋, 宁德军, 毛建华. 基于1D卷积神经网络的滚动轴承故障诊断方法[J]. 工业控制计算机, 2019, 32(6): 90-92

    YU J Y, NING D J, MAO J H. Rolling bearing fault diagnosis method based on 1D convolutional neural network[J]. Industrial Control Computer, 2019, 32(6): 90-92
    [5] 康守强, 胡明武, 王玉静, 等. 基于特征迁移学习的变工况下滚动轴承故障诊断方法[J]. 中国电机工程学报, 2019, 39(3): 764-772

    KANG S Q, HU M W, WANG Y J, et al. Fault diagnosis method of a rolling bearing under variable working conditions based on feature transfer learning[J]. Proceedings of the CSEE, 2019, 39(3): 764-772
    [6] 张根保, 李浩, 冉琰, 等. 一种用于轴承故障诊断的迁移学习模型[J/OL]. 吉林大学学报(工学版), 2019, 49(2): 1-9

    ZHANG G B, LI H, RAN Y, et al. A transfer learning model for bearing fault diagnosis[J/OL]. Journal of Jilin University: Engineering and Technology Edition, 2019, 49(2): 1-9
    [7] 陈超, 齐峰. 卷积神经网络的发展及其在计算机视觉领域中的应用综述[J]. 计算机科学, 2019, 46(3): 63-73

    CHEN C, QI F. Review on development of convolutional neural network and its application in computer vision[J]. Computer Science, 2019, 46(3): 63-73
    [8] 赵璐, 马野. 基于一维卷积神经网络的齿轮箱故障诊断研究[J]. 测试技术学报, 2019, 33(4): 302-306

    ZHAO L, MA Y. Research on fault diagnosis of gearbox based on 1D convolutional neural network[J]. Journal of Test and Measurement Technology, 2019, 33(4): 302-306
    [9] SIMONYAN K, ZISSERMAN A. Very deep convolutional networks for large-scale image recognition[C]//ICLR, 2015. arXiv: 1409.1556
    [10] 贺思艳, 任利娟, 田新诚. 基于卷积神经网络的滚动轴承故障诊断[J]. 兵工自动化, 2019, 38(3): 39-41

    HE S Y, REN L J, TIAN X C. Rolling bearing fault diagnosis based on convolution neural network[J]. Ordnance Industry Automation, 2019, 38(3): 39-41
    [11] 张海涛, 刘景鑫, 赵晓晴, 等. 基于VGG16的急性淋巴细胞白血病血液细胞显微图像人工智能辅助诊断分类研究[J]. 中国医疗设备, 2019, 34(7): 1-4

    ZHANG H T, LIU J X, ZHAO X Q, et al. An artificial intelligence-assisted diagnosis classification research on blood cell microscopic image of acute lymphoblastic leukemia research on VGG16 model[J]. China Medical Devices, 2019, 34(7): 1-4
    [12] PAN S J, YANG Q. A survey on transfer learning[J]. IEEE Transactions on Knowledge and Data Engineering, 2010, 22(10): 1345-1359 doi:  10.1109/TKDE.2009.191
    [13] ZEILER M D, FERGUS R. Visualizing and understanding convolutional network[C]//European Conference on Computer Vision, 2014. arXiv: 1311.2901
    [14] http://csegroups.case.edu/bearingdatacenter/pages/download-data-file[DS/OL]. [2019-06-15]
    [15] https://ti.arc.nasa.gov/tech/dash/groups/pcoe/prognostic-data-repository[DS/OL]. [2019-06-15]
  • [1] 周嘉明, 董龙雷, 赵建平, 丁镇军, 王潇屹.  基于概率神经网络的航天结构响应映射预示方法 . 航天器环境工程, 2023, 40(4): 392-399. doi: 10.12126/see.2023070
    [2] 刘洋洋, 韩潇, 张鹏嵩, 沈平单, 曹海林.  一种用于提高构件损伤边缘成像精度的方法 . 航天器环境工程, 2023, 40(3): 301-308. doi: 10.12126/see.2023004
    [3] 彭瑾坤, 武文华, 顾乃建, 胡奇.  基于CNN的卫星内部结构频域响应间接测量方法 . 航天器环境工程, 2023, 40(4): 400-407. doi: 10.12126/see.2023067
    [4] 李振, 石佳东, 马国政, 张执南, 李国禄, 郭伟玲, 王海斗.  自润滑基础运动部件模拟空间环境试验装置研究进展 . 航天器环境工程, 2023, 40(4): 423-429. doi: 10.12126/see.2023016
    [5] 刘云帆, 林亮行, 马国政, 孙建芳, 苏峰华, 郭伟玲, 朱丽娜, 王海斗.  基于CNN和LSTM的航天用涂层型自润滑关节轴承寿命预测及可靠性评估 . 航天器环境工程, 2023, 40(5): 531-540. doi: 10.12126/see.2023012
    [6] 张潇, 刘沐阳.  基于集成学习方法的机电作动器故障诊断框架 . 航天器环境工程, 2023, 40(5): 559-566. doi: 10.12126/see.2023113
    [7] 伍济钢, 文港.  基于优化多尺度排列熵和卷积神经网络的滚动轴承故障诊断方法 . 航天器环境工程, 2023, 40(1): 99-106. doi: 10.12126/see.2022102
    [8] 范军红, 朱云飞, 王文彪, 鲍官军, 姜利祥.  基于前馈神经网络的柔性连续臂建模与仿真 . 航天器环境工程, 2022, 39(2): 133-139. doi: 10.12126/see.2022.02.003
    [9] 刘大禹, 徐二宝.  数据驱动的卫星关键部件剩余寿命预测模型 . 航天器环境工程, 2022, 39(2): 148-152. doi: 10.12126/see.2022.02.005
    [10] 马健, 刘柱, 石育榕, 罗冰显, 郑锋.  基于太阳黑子群数据的多模态太阳耀斑预报模型 . 航天器环境工程, 2021, 38(3): 256-262. doi: 10.12126/see.2021.03.004
    [11] 陈彬强, 何昱超, 姚斌, 王栋, 贺王鹏.  航天器传动部件振动数据稀疏多分类智能故障诊断 . 航天器环境工程, 2018, 35(4): 359-364. doi: 10.12126/see.2018.04.009
    [12] 杜刚, 何朔, 于海鹏.  基于径向基函数神经网络的空间碎片撞击模式识别研究 . 航天器环境工程, 2015, 32(4): 357-360. doi: 10.12126/see.2015.04.003
    [13] 谢一村, 樊世超, 杨艳静.  应用经验模态分解的箱型结构裂纹诊断方法 . 航天器环境工程, 2015, 32(5): 489-495. doi: 10.12126/see.2015.05.006
    [14] 安若铭, 高阳.  基于分层神经网络的航天器故障诊断技术 . 航天器环境工程, 2013, 30(2): 203-208.
    [15] 张 森, 于登云, 王九龙.  非迭代动态多故障诊断方法研究 . 航天器环境工程, 2011, 28(6): 562-567.
    [16] 张 利, 付国庆, 梁 伟.  振动监测与故障诊断技术在“风云二号” 扫描辐射计中的应用 . 航天器环境工程, 2010, 27(6): 727-730.
    [17] 罗 成, 韩晓健.  动力学环境试验技术在航天器结构故障诊断中的应用 . 航天器环境工程, 2009, 26(2): 147-149.
    [18] 王婉秋, 杨 松.  航天器动力学环境试验故障诊断专家系统 . 航天器环境工程, 2008, 25(1): 36-40.
    [19] 焦 捷.  基于贝叶斯网络的飞机变流器故障诊断 决策系统的设计与实现 . 航天器环境工程, 2007, 24(2): 121-124.
    [20] 航天器热控系统故障诊断策略 . 航天器环境工程, 2003, 20(3): 44-47.
  • 加载中
图(6) / 表ll (2)
计量
  • PDF下载量:  47
  • 文章访问数:  533
  • HTML全文浏览量:  195
文章相关
  • 中图分类号:  TP18; TH133.3
  • 收稿日期:  2019-10-15
  • 修回日期:  2020-09-09
  • 网络出版日期:  2020-10-22
  • 刊出日期:  2020-10-22

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

doi: 10.12126/see.2020.05.005
    基金项目:  国家自然科学基金项目(编号:61573257);北京卫星环境工程研究所创新基金项目(编号:CAST-BISEE 2019-022)
    作者简介:

    赵宇凯(1997—),男,硕士研究生,主要研究方向为深度学习模型在故障诊断方面的应用。E-mail: zhaoyukaijake@tongji.edu.cn

    刘   敏(1970—),男,博士学位,教授,博士生导师,从事互联网与大数据环境下智能维护与服务优化配置的研究。E-mail: lmin@tongji.edu.cn

  • 中图分类号: TP18; TH133.3

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

English Abstract

赵宇凯, 徐高威, 刘敏. 基于VGG16迁移学习的轴承故障诊断方法[J]. 航天器环境工程, 2020, 37(5): 446-451 doi:  10.12126/see.2020.05.005
引用本文: 赵宇凯, 徐高威, 刘敏. 基于VGG16迁移学习的轴承故障诊断方法[J]. 航天器环境工程, 2020, 37(5): 446-451 doi:  10.12126/see.2020.05.005
ZHAO Y K, XU G W, LIU M. Method for fault diagnosis of bearing based on transfer learning with VGG16 model[J]. Spacecraft Environment Engineering, 2020, 37(5): 446-451 doi:  10.12126/see.2020.05.005
Citation: ZHAO Y K, XU G W, LIU M. Method for fault diagnosis of bearing based on transfer learning with VGG16 model[J]. Spacecraft Environment Engineering, 2020, 37(5): 446-451 doi:  10.12126/see.2020.05.005
    • 传动装置是日常生活和工业生产当中最常用的机械装置之一,而传动装置的安全平稳运行离不开轴承的正常工作。在各种各样的工作环境下,轴承的工作状况是急剧变化的,对轴承故障的早期诊断可以有效预防安全问题的发生。然而,基于传统可靠性工程的诊断方法已经很难适应轴承类传动部件故障诊断的需求[1-2];同时,随着物联网技术与大数据技术的飞速发展,越来越多的轴承状态监测数据被采集并存储下来[3]。因此,数据驱动的故障诊断方法在学术界与工业界被广泛研究与应用,如:陈彬强等[2]将分形小波变换和稀疏分类的方法应用到航天器传动机械零部件的故障诊断问题中,解决了数据稀缺的问题;于景洋等[4]提出1D卷积神经网络的滚动轴承诊断方法,可以对一维振动信号进行处理。然而,这些传统方法存在如下问题:1)需要对原始数据进行烦琐的预处理;2)需要大量的训练数据来对网络进行训练,训练时间长,人力物力耗费巨大。针对以上问题,康守强等[5]利用特征迁移的学习方法,针对变工况下的滚动轴承故障诊断问题提出基于变分模态分解的特征提取和迁移方法;张根保等[6]提出了一种用于轴承故障诊断的迁移学习模型,由栈式稀疏编码器和柔性最大值函数组成,当工况发生变化时只需要少量样本就可以重新执行诊断任务。

      本文提出一种基于VGG16卷积神经网络的迁移学习方法,将在源域预训练过的VGG16网络应用到轴承故障数据集中。首先利用信号转化的处理方法,将原始一维振动信号转化为二维(224×224)灰度图像,再对单通道的灰度图像进行复制、叠加成为三通道(224×224×3)图像;然后将处理后的图片样本分为训练集和测试集,对训练集在预先训练过的网络当中进行微调,以达到网络最佳参数;最后通过测试集验证该模型的故障分类能力。

    • 基于卷积神经网络的故障诊断方法很难直接对原始振动信号进行处理,因此,对原始数据进行预处理尤为重要。当前的数据预处理方法是通过时频分析对原始信号的特征进行提取,通常需要花费大量的时间,并且人工提取的特征对最终的分类结果影响很大。而本文设计一种直接的信号‒图像转换方法,无须依赖任何专家经验。

      通常轴承故障数据采集的是时域信号,是一种一维矩阵的格式。要将原始的时域信号转换成为一个N×N像素数的图片格式,首先需要从连续的信号中随机取样,抽取N个长度为N的时间信号,组成一个二维矩阵L,经过对矩阵作0~255的归一化之后,即得到一个N×N像素的灰度图像格式。具体变换过程可表示为

      $$P = {\rm{round}}\left( {\frac{{{{L}} - {\rm{min}}({{L}})}}{{\max ({{L}}) - \min ({{L}})}} \times 255} \right)\text{,}$$ (1)

      其中round(·)函数返回是四舍五入的函数值。该方法对轴承正常情况下信号的转化过程如图1所示,一维的振动信号输出在MatLab上为振动波形式,经过该信号转化方法处理后,原来的振动信号被转化为相应大小的矩阵,再对矩阵进行归一化之后输出灰度图像格式。

      图  1  信号转图像处理方法

      Figure 1.  Signal-to-image conversion method

    • 卷积神经网络(Convolutional Neural Network, CNN)仿照生物的视知觉机制构建,是机器学习的代表算法之一[7]。CNN的基本结构包括输入层、卷积层、池化层、全连接层和输出层,其中:卷积层和池化层可以多次叠加来提高网络性能;卷积层利用卷积核进行卷积操作,对输入样本的局部加权求和,提取输入的特征;池化层进行特征降维,减少学习量并降低网络复杂度[8]。在训练过程中,可以根据相应的需求自由调整卷积核的步长。经过多次卷积、池化操作之后,输入样本的特征能够被网络提取,再经由全连接层和输出层激活函数对结果进行分类。

    • VGG卷积神经网络模型[9]是牛津大学在2014年提出的,在众多的VGG变形当中,VGG16在图像分类和目标检测任务中都表现良好,故本文选择使用VGG16卷积神经网络。该网络使用连续的3×3的卷积核代替原本较大的卷积核,并且通过增加网络深度使得对样本的学习更加深入。VGG16卷积神经网络结构如图2所示,通过反复叠加的卷积层(convolution layer)和池化层(pooling layer),共有13层卷积层和3层全连接层(fully connected layer),而池化层不计权重故不算在总层数之内。卷积层和池化层其实就是对输入图像的一种提取过程,多层卷积池化层相互堆叠,使得网络具有更大感受野的同时又能降低网络参数,并且通过ReLU激活函数使得原本的单一线性变化变得多样化,学习能力也因此增强[10]。经由全连接层和输出层可以将样本进行分类处理,通过softmax激活函数可以得到当前样本属于不同种类的概率分布[11]

      图  2  VGG16卷积神经网络结构

      Figure 2.  Structure chart of VGG16 convolutional neural network

    • 迁移学习[12]是一种能够将从其他领域学习到的知识应用到所需研究领域的机器学习方法。当一个神经网络进行训练时,一般认为训练集和测试集是同分布的,并且需要成千上万个样本对该模型进行训练优化。因此即使样本足够,处理样本数据也是一个成本很高的过程;而迁移学习方法能够很好地弥补因为数据量不足或者训练时间过长而带来的问题。

      VGG16卷积神经网络是一个对图像进行提取并不断深入提炼的过程,它具有很深的网络结构。起初的卷积层只对图像的特征进行提取,而网络层较深时才处理特定的任务[13]。因此在迁移学习过程当中,可以应用已经在源域预先训练过的模型,保留较低层的权重,只对高层进行重新训练并微调相关参数。因为与源域数据分布不同的特点,需要利用目标域数据集对模型进行微调,去掉原来的顶层并添加新的输出层,加入softmax函数对新的问题进行分类。图3所示即为一个对模型进行迁移的过程,通过对模型当中在源域(自然图像)训练的权重进行迁移,再利用目标域(任务数据)进行微调,大大简化了模型在新领域问题当中的训练过程。

      图  3  卷积神经网络模型迁移

      Figure 3.  Model transfer of CNN

    • 本文的实验数据集1来自凯斯西储大学的轴承数据库[14],其实验装置如图4所示,测试台由2 HP(1 HP=735 W)电机、扭矩传感器、测力计和控制电子组成。测试轴承支撑电机轴,采用电火花加工将单点故障人为引入实验轴承,使用加速度计收集振动数据,使用扭矩传感器收集速度和功率数据。故障数据集分别在12 kHz和48 kHz采样率下的驱动端和风扇端处产生,又各自分成不同位置(内圈、外圈、滚动体)、不同故障直径(0.007、0.014、0.021、0.028英寸)、不同负载(0、1、2、3 HP)下的实验数据。

      图  4  凯斯西储大学的轴承实验装置

      Figure 4.  The bearing test bench in Case Western Reserve University

      实验数据集2来自辛辛那提大学的IMS轴承数据库[15],其实验装置通过1个主轴将4个轴承连接在一起,用交流电机将轴承转速保持在2000 r/min,每个轴承配备一个对应的加速度计记录轴承数据。测试数据一共分为3组,每组都记录4个轴承从测试开始直到轴承失效的振动信号,轴承故障分为内圈、外圈、滚动体3类。

      表1所示,为了保证轴承数据的多工况条件,本实验选择了数据集1当中轴承故障直径为0.007英寸、不同负载条件下,正常、内圈故障、滚动体故障和外圈故障的轴承振动信号;并从每组数据中通过我们的预处理方法选择出500个样本组成训练集,同时另选出500个样本组成测试集,共计2000个样本的训练集和2000个样本的测试集。除此之外,本实验还使用相同的方法从数据集2中相同故障状态下挑选了相同样本数量的训练集和测试集,以验证不同工况条件下VGG16故障分类的准确率。

      表 1  实验数据集

      Table 1.  Experimental data set

      编号样本类别数据集正常内圈滚动体外圈
      数据集10 HP训练集500500500500
      测试集500500500500
      1 HP训练集500500500500
      测试集500500500500
      2 HP训练集500500500500
      测试集500500500500
      3 HP训练集500500500500
      测试集500500500500
      数据集2IMS训练集500500500500
      测试集500500500500
    • 图5所示为数据集1当中0 HP负载下、故障直径为0.007英寸的轴承振动信号,经过信号‒图像转换后所产生的灰度图像。VGG16网络的输入格式为224×224×3,像素为224×224,故选择N为224。经过转换的振动信号是单通道的灰度图像,要经过复制才能成为VGG16网络的输入格式。图5下半部分以正常轴承的灰度图像为例,将转换好的灰度图像进行复制、叠加成为VGG16网络可输入的三通道格式。

      图  5  数据预处理

      Figure 5.  Data preprocessing

    • 模型应用过程中,首先对训练集在VGG16网络中进行微调,训练出新的网络层参数之后利用测试集进行验证。训练过程中参数选择迭代次数为10代,每一次的梯度下降样本数(batch_size)为50,每一轮训练结束后都进行验证,最后由测试集测试模型的分类准确率。如表2所示,另外选取了一维CNN和支持向量机(Support Vector Machines, SVM)这2种比较经典的故障诊断方法与本文方法进行比较。

      表 2  轴承故障诊断准确率

      Table 2.  Bearing fault diagnosis results

      单位:%
      诊断方法VGG16CNNSVM
      故障类别0 HP10010096.2
      1 HP10010096.9
      2 HP10010097.5
      3 HP10099.595.7
      IMS10099.594.8
      0~1 HP10099.893.5
      0~2 HP99.994.192.8
      0~3 HP99.992.590.5
      0~3 HP+IMS99.690.487.8
      平均值99.797.393.9

      本文将实验分成了9组,依次是0 HP、1 HP、2 HP、3 HP、IMS这5组四分类任务,以及0~1 HP、0~2 HP、0~3 HP、0~3 HP+IMS这4组多分类任务。从这9组实验结果来看,VGG16与CNN、SVM相比,平均准确率分别高出2.4和5.8个百分点。从图6可以看出:当面对四分类任务时,VGG16和CNN都能保持非常高的诊断准确率,SVM的准确率要相对低一些;而随着分类种类增多,任务变得复杂,3种算法的准确率都有所下降,但VGG16只下降了0.4个百分点,而CNN和SVM的准确率分别下降了9.6和12.2个百分点。

      图  6  3种诊断方法的准确率曲线

      Figure 6.  Accuracy curve for three diagnostic methods

      由上述实验结果可知,VGG16在不同工况条件下均能轻松完成故障的分类任务,且当分类任务变得更加复杂时依然能够保持性能的优越。

    • 本文针对轴承故障诊断任务,提出基于VGG16迁移学习的诊断方法,主要创新点如下:

      1)提出了一种高效快捷的信号转图像处理方法,将原本的一维振动信号转换为灰度图像信号,大大降低了信号特征提取的复杂度;

      2)设计了一种可用于轴承故障诊断的VGG16迁移学习模型,将完成预训练的VGG16卷积神经网络模型中的浅层结构与权重保留,并用目标任务中的数据集对VGG16模型较高层结构中的权重进行微调,微调后的VGG16模型直接用于对轴承故障类型进行分类。

      应用本文所提出的方法分别对凯斯西储大学的轴承故障数据集和辛辛那提大学的IMS轴承数据集进行了实验验证,结果表明,在实际应用中,本文提出的信号转图像处理方法具有良好的特征提取能力,设计的VGG16迁移学习模型能够在不同工况条件下和多分类任务中保持良好的算法性能。

参考文献 (15)

目录

    /

    返回文章
    返回