| Citation: |
JIANG Fan，SONG Hongyan，SHEN Xi，et al. Fault diagnosis method of mine hoist main bearing with small sample based on VAE-WGAN[J]. Coal Science and Technology，2025，53（S1）：1−15. DOI: 10.12438/cst.2023-1503
|
As a key component of the hoist, the main bearing may deteriorate in performance and cause faults during long-term high-speed and heavy-duty service. Therefore, conducting fault diagnosis of the main bearing of the hoist is of great significance for ensuring the safe and efficient operation of the mine hoist. However, the proportion of normal service and fault status in the monitoring data of the operation status of mine hoists is severely imbalanced, showing characteristics such as a large number of normal samples, a small number of fault samples, and insufficient label samples, resulting in unsatisfactory training results and low diagnostic accuracy of the main bearing fault model of mine hoists. In response to the problem of low accuracy in fault diagnosis of mine hoist main bearings under small samples, a VAE-WGAN based mine hoist main bearing sample augmentation model was constructed by fusing variational autoencoder and Wasserstein to generate adversarial networks. Furthermore, a fault diagnosis method based on CBAM-MobileNetV2 is proposed to achieve fault diagnosis of mine hoist main bearings under small sample data. At the algorithmic level, the Wasserstein distance metric is introduced to solve the problem of vanishing training gradients in generative adversarial networks. At the data level, VAE-WGAN was tested using the Case Western Reserve University dataset, and hyperparameters were optimized by evaluating the generation ability of VAE-WGAN through quantitative indicators. VAE-WGAN was trained using the bearing dataset of the mine hoist simulation test bench to achieve the expansion and expansion of the small sample dataset. In order to improve the feature extraction ability and fault diagnosis accuracy of fault diagnosis models, based on the lightweight convolutional neural network MobileNetV2, the convolutional block attention mechanism CBAM is integrated into the deep feature mapping of MobileNetV2, and an attention mechanism convolutional classification network CBAM-MobileNetV2 is constructed. By integrating cross channel and spatial information, more attention is paid to fault features. Finally, a comparative analysis was conducted with traditional generation models such as WGAN-GP, DCGAN, VAE, and WGAN. The fault diagnosis accuracy of VAE-WGAN+CBAM-MobileNetV2 on four small sample ratio datasets was higher than the other four methods, proving that the proposed sample augmentation and fault diagnosis methods have higher diagnostic accuracy on different small sample ratio fault datasets and can meet the fault diagnosis requirements under small samples.
| [1] |
张宏乐,韩雪华,史凯,等. 基于深度迁移学习的提升机主轴故障在线诊断系统研究[J]. 煤炭工程,2022,54(7):61−66.
ZHANG Hongle,HAN Xuehua,SHI Kai,et al. Online fault diagnosis system of hoist bearing based on deep transfer learning[J]. Coal Engineering,2022,54(7):61−66.
|
| [2] |
李娟莉,赵旭荣,孙梦祯. 矿井提升机主轴系统故障本体诊断方法[J]. 矿业研究与开发,2020,40(6):144−147.
LI Juanli,ZHAO Xurong,SUN Mengzhen. Fault ontology diagnosis method of spindle system of mine hoist[J]. Mining Research and Development,2020,40(6):144−147.
|
| [3] |
张梦瑜. 矿井提升机电控系统的研究与设计[J]. 机械管理开发,2022,37(3):275−276,279.
ZHANG Mengyu. Research and design of electric control system for mine hoist[J]. Mechanical Management and Development,2022,37(3):275−276,279.
|
| [4] |
张宏乐,韩雪华,史凯,等. 基于深度迁移学习的提升机主轴故障在线诊断系统研究[J]. 煤炭工程,2022,54(7):61−66.
ZHANG Hongle,HAN Xuehua,SHI Kai,et al. Online fault diagnosis system of hoist bearing based on deep transfer learning[J]. Coal Engineering,2022,54(7):61−66.
|
| [5] |
杨光友,刘浪,习晨博. 自适应辅助分类器生成式对抗网络样本生成模型及轴承故障诊断[J]. 中国机械工程,2022,33(13):1613−1621.
YANG Guangyou,LIU Lang,XI Chenbo. Bearing fault diagnosis based on SA-ACGAN data generation model[J]. China Mechanical Engineering,2022,33(13):1613−1621.
|
| [6] |
郭伟,邢晓松. 基于改进卷积生成对抗网络的少样本轴承智能诊断方法[J]. 中国机械工程,2022,33(19):2347−2355. doi: 10.3969/j.issn.1004-132X.2022.19.009
GUO Wei,XING Xiaosong. Intelligent fault diagnosis of bearings with few samples based on an improved convolutional generative adversarial network[J]. China Mechanical Engineering,2022,33(19):2347−2355. doi: 10.3969/j.issn.1004-132X.2022.19.009
|
| [7] |
CRESWELL A,WHITE T,DUMOULIN V,et al. Generative adversarial networks:An overview[J]. IEEE Signal Processing Magazine,2018,35(1):53−65. doi: 10.1109/MSP.2017.2765202
|
| [8] |
陈其. 改进GAN结合SDAE的传动系统主轴承故障诊断[J]. 微特电机,2023,51(2):20−25.
CHEN Qi. Fault diagnosis of main bearing of transmission system based on ACGAN-SDAE[J]. Small & Special Electrical Machines,2023,51(2):20−25.
|
| [9] |
PLAKIAS S,BOUTALIS Y S. Exploiting the generative adversarial framework for one-class multi-dimensional fault detection[J]. Neurocomputing,2019,332:396−405. doi: 10.1016/j.neucom.2018.12.041
|
| [10] |
DING Y,MA L,MA J,et al. A generative adversarial network-based intelligent fault diagnosis method for rotating machinery under small sample size conditions[J]. IEEE Access,2019,7:149736−149749. doi: 10.1109/ACCESS.2019.2947194
|
| [11] |
ARJOVSKY M,CHINTALA S,BOTTOU L,et al. Wasserstein generative adversarial networks[C]//Proceedings of the 34th International Conference on Machine Learning - Volume 70. New York:ACM,2017:214−223.
|
| [12] |
QI Guojun. Loss-sensitive generative adversarial networks on lipschitz densities[J]. International Journal of Computer Vision,2020,128(5):1118−1140. doi: 10.1007/s11263-019-01265-2
|
| [13] |
徐卓飞,李旭东,张婵婵,等. 基于孪生网络的小样本滚动轴承故障诊断研究[J]. 仪器仪表学报,2022,43(10):241−251.
XU Zhuofei,LI Xudong,ZHANG Chanchan,et al. Fault diagnosis of rolling bearings with limited samples based on Siamese network[J]. Chinese Journal of Scientific Instrument,2022,43(10):241−251.
|
| [14] |
LARSEN A B L, SØNDERBY S K, LAROCHELLE H, et al. Autoencoding beyond pixels using a learned similarity metric[EB/OL]. (2016‒02‒10) [2023‒08‒20]. https://arxiv.org/abs/1512.09300v2.
|
| [15] |
MOZTARZADEH O,JAMSHIDI M B,SARGOLZAEI S,et al. Metaverse and medical diagnosis:A blockchain-based digital twinning approach based on MobileNetV2 algorithm for cervical vertebral maturation[J]. Diagnostics,2023,13(8):1485. doi: 10.3390/diagnostics13081485
|
| [16] |
LIU Jiansheng,CUI Guolong,XIAO Chengdi. A real-time and efficient surface defect detection method based on YOLOv4[J]. Journal of Real-Time Image Processing,2023,20(4):77. doi: 10.1007/s11554-023-01333-4
|
| [17] |
李炳臻,刘克,顾佼佼,等. 卷积神经网络研究综述[J]. 计算机时代,2021(4):8−12,17.
LI Bingzhen,LIU Ke,GU Jiaojiao,et al. Review of the researches on convolutional neural networks[J]. Computer Era,2021(4):8−12,17.
|
| [18] |
PANG Bo. Classification of images using EfficientNet CNN model with convolutional block attention module (CBAM) and spatial group-wise enhance module (SGE)[C]//International Conference on Image,Signal Processing,and Pattern Recognition (ISPP 2022).Washington: SPIE,2022.
|
| [19] |
姚齐水,别帅帅,余江鸿,等. 一种结合改进Inception V2模块和CBAM的轴承故障诊断方法[J]. 振动工程学报,2022,35(4):949−957.
YAO Qishui,BIE Shuaishuai,YU Jianghong,et al. A bearing fault diagnosis method combining improved inception V2 module and CBAM[J]. Journal of Vibration Engineering,2022,35(4):949−957.
|
| [20] |
王生霄,侯兴松,黑夏萌. 嵌入CBAM结构的改进YOLOV3超宽带雷达生命信号检测算法[J]. 国外电子测量技术,2020,39(3):1−6.
WANG Shengxiao,HOU Xingsong,HEI Xiameng. Improved YOLOV3 UWB radar life signal detection algorithm embedded in CBAM structure[J]. Foreign Electronic Measurement Technology,2020,39(3):1−6.
|
| [21] |
任治洲,梁琨,王泽宇等. 结合改进CBAM和MobileNet V2算法对小麦病斑粒分类[J]. 南京农业大学学报,2023,47(3):583−591.
REN Zhizhou,LIANG Kun,WANG Zeyu,et al. Combining improved CBAM and MobileNet V2 algorithms for classifying diseased wheat kernels[J]. Journal of Nanjing Agricultural University,2023,47(3):583−591
|
| [22] |
CHONG M J,FORSYTH D. Effectively unbiased FID and inception score and where to find them[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). Piscataway,NJ:IEEE,2020:6069−6078.
|
| [23] |
ZHANG R,ISOLA P,EFROS A A,et al. The unreasonable effectiveness of deep features as a perceptual metric[C]//2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway,NJ:IEEE,2018:586−595.
|
Supported by: Beijing Renhe Information Technology Co., Ltd. 
DownLoad: