计算机应用 ›› 2018, Vol. 38 ›› Issue (8): 2224-2229.DOI: 10.11772/j.issn.1001-9081.2017112702

• 人工智能 • 上一篇    下一篇

基于融合型深度学习的滚动轴承亚健康识别算法

张利1,2, 孙军1, 李大伟2, 牛明航2, 高一丹2   

  1. 辽宁大学 信息学院, 沈阳 110036
  • 收稿日期:2017-11-17 修回日期:2018-01-09 出版日期:2018-08-10 发布日期:2018-08-11
  • 通讯作者: 张利
  • 作者简介:张利(1971-),男,吉林榆树人,教授,博士,主要研究方向:模式识别、数据聚类;孙军(1994-),男,山东威海人,硕士研究生,主要研究方向:机器学习;李大伟(1991-),男,山东枣庄人,硕士研究生,主要研究方向:机器学习;牛明航(1995-),男,山东菏泽人,硕士研究生,主要研究方向:机器学习。

Rolling bearing sub-health recognition algorithm based on fusion deep learning

ZHANG Li1,2, SUN Jun1, LI Dawei2, NIU Minghang2, GAO Yidan2   

  1. College of Information, Liaoning University, Shenyang Liaoning 110036, China
  • Received:2017-11-17 Revised:2018-01-09 Online:2018-08-10 Published:2018-08-11

摘要: 深度学习模型增加了隐含层的层数,使得该模型在语音识别、图像视频分类等方面取得了不错的效果;但建立一个适合特定对象的模型需要大量的数据集来训练,而且需要较长时间才能获得合适的权重和偏置,为此提出一种基于深度自动编码器-相关向量机网络模型的滚动轴承亚健康诊断方法。首先,采集滚动轴承振动信号并进行傅里叶变换和归一化处理;其次,设计改进的自动编码器-稀疏边缘降噪自动编码器,结合了稀疏自动编码器和边缘降噪自动编码器的特点;接着建立深度自动编码器-相关向量机网络模型,用有监督的函数对各个隐含层的参数进行微调,并利用相关向量机(RVM)进行训练;最后将得到的分类根据D-S证据理论融合并得出最终的分类结果。实验结果表明所提算法能有效提高滚动轴承"亚健康"状态的识别精度,纠正错误分类。

关键词: 深度学习, 亚健康识别, 相关向量机, D-S证据理论, 滚动轴承

Abstract: The deep learning model increases the number of hidden layers, which makes the model have a good effect on speech recognition, image video classification and so on. However, to establish a model suitable for a specific object, a large number of data sets are required to train it for a long time to get the appropriate weights and biases. To resolve the above problems, a sub-health diagnosis method for rolling bearing was proposed based on depth autoencoder-relevance vector machine network model. Firstly, the bearing vibration signal was collected and transformed by Fourier transform and normalization. Secondly, the improved automatic encoder, named sparse edge noise reduction autoencoder, was designed, which combined the features of sparse automatic encoder and edge noise reduction automatic encoder. Then the depth autoencoder-relevance vector machine network model was designed, in which the supervised function was used to finely tune the parameters of each hidden layer, and it was trained by Relevance Vector Machine (RVM). Finally, the final classification results were obtained according to D-S (Dempster-Shafer) evidence fusion theory. The experimental results show that the proposed algorithm can effectively improve the recognition precision of the "sub-health" state of the rolling bearing and correct the error classification.

Key words: deep learning, sub-health recognition, Relevance Vector Machine (RVM), D-S (Dempster-Shafer) evidence theory, rolling bearing

中图分类号: