吉林大学学报(工学版) ›› 2026, Vol. 56 ›› Issue (7): 1811-1824.doi: 10.13229/j.cnki.jdxbgxb.20241377

• 车辆工程·机械工程 • 上一篇    

基于多通道融合的齿轮箱故障诊断方法

冯志刚(),王颖,王宇   

  1. 沈阳航空航天大学 自动化学院,沈阳 110136
  • 收稿日期:2024-12-30 出版日期:2026-07-01 发布日期:2026-08-12
  • 作者简介:冯志刚(1980-),男,教授,博士.研究方向:故障诊断.E-mail:fzg1023@yeah.net
  • 基金资助:
    国家自然科学基金青年基金项目(51605309)

Gear box fault diagnosis method based on multi-channel fusion

Zhi-gang FENG(),Ying WANG,Yu WANG   

  1. Department of Automation,Shenyang Aerospace University,Shenyang 110136,China
  • Received:2024-12-30 Online:2026-07-01 Published:2026-08-12

摘要:

针对单通道的振动信号在全面表征齿轮箱故障特征方面存在局限性的问题,提出了一种将多通道的振动信号变换为时频图后经过加权融合成RGB图像,再输入到诊断模型中的数据级融合方法MFTDCN。该方法能有效整合多个通道的信息,获得比单个振动信号更丰富的特征。首先,对多通道振动信号进行主成分分析(PCA)处理,降维成三个分量,用短时傅里叶变换(STFT)将3个分量变换为3个时频图像;然后,用Hoyer稀疏度对余弦相似度进行改进,将其应用于基于稀疏性和相似度的权重计算,利用得到的权重对三个时频图像进行加权处理,从而融合成RGB图像,作为诊断模型的输入。诊断模型部分为可变形卷积(DCN)与Transformer相结合,提取特征后进行分类。在东南大学的齿轮箱数据集和北京工业大学与北京交通大学联合发布的数据集上进行了相关实验验证,结果证实了该方法的有效性和可靠性。

关键词: 故障诊断, 数据级融合, 短时傅里叶变换, 余弦相似度, 可变形卷积, Transformer

Abstract:

To address the limitations of single-channel vibration signals in comprehensively characterizing gearbox fault features, this paper proposes a data-level fusion method called MFTDCN is proposed.This method effectively integrates information from multiple channels to obtain richer features compared to single vibration signals.The approach begins by applying Principal Component Analysis (PCA) to multi-channel vibration signals to reduce them into three components. These components are then transformed into three time-frequency images using Short-Time Fourier Transform (STFT). An improved cosine similarity measure, based on Hoyer sparsity, is employed to calculate weights by combining sparsity and similarity metrics. These weights are used to perform weighted fusion of the three time-frequency images, resulting in a single RGB image, which is then used as input for the diagnostic model. The diagnostic model combines Deformable Convolution Networks (DCN) with Transformer architectures to extract features and classify faults. Experimental results on the gearbox data set of Southeast University and the data set jointly published by Beijing University of Technology and Beijing Jiaotong University confirm the effectiveness and reliability of the proposed method.

Key words: fault diagnosis, data-level fusion, short-time Fourier transform, cosine similarity, deformable convolution network, Transformer

中图分类号: 

  • TH17

图1

可变形卷积示意图"

图2

Transformer结构图"

图3

基于Hoyer稀疏度改进的余弦相似度计算结构图"

图4

MFTDCN模型图"

图5

故障诊断流程"

表1

东南大学数据集中轴承和齿轮数据集划分表"

轴承/齿轮箱类型训练集测试集标签
轴承数据滚动体故障210900
内圈故障210901
外圈故障210902
复合故障210903
健康状态210904
齿轮数据齿轮断齿210900
齿轮缺口210901
齿面磨损210902
齿根裂纹210903
健康状态210904

图6

不同图像转换方法的对比效果图"

表2

不同图像转换方法准确率对比"

工况GrayGAFMTFSTFT
均值86.8793.1180.0699.46
bearingset_20_082.1696.1674.24100.00
bearingset_30_284.2293.7684.2299.92
gearset_20_086.1888.7182.1198.27
gearset_30_294.9193.8079.6799.65

表3

不同故障类型的3个分量时频图与生成的RGB图像对比"

类型工况 bearingset_30_2工况gearset_30_2
滚动体内圈外圈复合正常缺口断牙裂纹磨损正常
分量1
分量2
分量3
RGB

表4

不同权重计算方法的准确率对比"

工况信息熵方差权重未改进的余弦相似度能量权重Hoyer稀疏度改进的余弦相似度
均值97.8898.0998.9098.2299.46
bearingset_20_099.4299.8399.8799.38100.00
bearingset_30_299.1799.63100.0099.5599.92
gearset_20_094.0293.9496.2495.0898.28
gearset_30_298.8998.9499.4898.8699.65

表5

不同模型的故障诊断结果对比表"

工况RGB-ResNetGray-LeNet-5Raw-TICNNMatrix-CNNCAM_MCFCNNMFTDCN
均值97.0093.8199.1297.6296.8999.46
bearingset_20_099.8495.1299.8499.5595.11100.00
bearingset_30_299.5599.2897.6099.9494.8999.92
gearset_20_092.2789.3399.0595.7698.6798.27
gearset_30_296.3291.5299.9795.2398.8899.65

图7

MFTDCN方法各工况训练准确率曲线"

图8

4种工况下的混淆矩阵"

图9

训练前后测试集t-SNE特征分布对比"

表6

小样本训练准确率"

工况1∶92∶83∶77∶3
均值95.0097.0497.5999.46
bearingset_20_098.1299.3699.59100.00
bearingset_30_297.1699.2799.4799.92
gearset_20_090.2592.9693.5598.27
gearset_30_294.4896.5897.7499.65

图10

小样本实验各工况训练曲线"

表 7

合并数据集准确率表"

工况准确率/%
20_098.26
30_298.38

图11

合并数据集训练准确率曲线"

图12

合并数据集30_2工况下混淆矩阵"

表8

泛化能力实验准确率表"

工况准确率/%工况准确率/%
1_2099.31_4098.5
1_2599.61_4598.3
1_3098.61_5098.1
1_3598.21_5599.6

图13

泛化能力实验1_55工况下训练曲线"

[1] Saufi S R, Ahmad Z A B, Leong M S, et al. Gearbox fault diagnosis using a deep learning model with limited data sample[J]. IEEE Transactions on Industrial Informatics, 2020, 16(10): 6263-6271.
[2] 侯召国, 王华伟, 熊明兰, 等.基于迁移学习与加权多通道融合的齿轮箱故障诊断[J]. 振动与冲击,2023, 42(9): 236-246.
Hou Zhao-guo, Wang Hua-wei, Xiong Ming-lan,et al. Gearbox fault diagnosis based on transfer learning and weighted multi-channel fusion[J]. Journal of Vibration and Shock,2023,42(9):236-246.
[3] She D M, Yang Z C, Duan Y D, et al. A meta transfer learning-driven few-shot fault diagnosis method for combine harvester gearboxes[J]. Computers and Electronics in Agriculture, 2024, 227: 109605.
[4] 王进花, 刘秦玮, 曹洁, 等.基于SCACGAN的小样本齿轮箱故障诊断[J]. 北京航空航天大学学报, 2026, 52(3): 713-723.
Wang Jin-hua, Liu Qin-wei, CAO Jie, et al. Fault diagnosis of gearbox with small-sample based on SCACGAN[J].Journal of Beijing University of Aeronautics and Astronautics, 2026, 52(3): 713-723.
[5] 刘辉, 李阳, 侯一民. 用于轴承故障诊断任务的轻量化卷积网络[J]. 哈尔滨理工大学学报, 2024(4): 80-88.
Liu Hui, Li Yang, Hou Yi-ming. Lightweight convolutional network for bearing fault diagnosis[J]. Journal of Harbin University of Science and Technology,2024(4): 80-88.
[6] Lei Y G, Jia F, Lin J, et al. An intelligent fault diagnosis method using unsupervised feature learning towards mechanical big data[J].IEEE Transactions on Industrial Electronics, 2016, 63(5): 3137-3147.
[7] Hu Q, Si X S, Zhang Q H, et al. A rotating machinery fault diagnosis method based on multi-scale dimensionless indicators and random forests[J]. Mechanical Systems and Signal Processing, 2020, 139: 106609.
[8] 石永芳, 徐庆宏, 姜宏, 等. 基于特征差异性学习卷积神经网络的齿轮箱故障诊断方法[J].机床与液压,2023,51(24):176-183.
Shi Yong-fang, XU Qing-hong, JIANG Hong, et al.Gearbox fault diagnosis method based on feature difference learning convolutional neural network[J]. Machine Tool & Hydraulics,2023,51(24):176-183.
[9] Zhao D Z, Cui L L, Liu D D. Bearing weak fault feature extraction under time-varying speed conditions based on frequency matching demodulation transform[J]. IEEE/ASME Transactions on Mechatronics, 2022, 28(3): 1627-1637.
[10] Goyal D, Dhami S S, Pabla B S. Non-contact fault diagnosis of bearings in machine learning environment[J]. IEEE Sensors Journal,2020, 20(9): 4816-4823.
[11] Zhang N N, Wu L F, Yang J, et al. Naive bayes bearing fault diagnosis based on enhanced independence of data[J]. Sensors, 2018, 18(2): 18020463.
[12] Zhao X L, Jia M P, Liu Z. Semisupervised deep sparse auto-encoder with local and nonlocal information for intelligent fault diagnosis of rotating machinery[J]. IEEE Transactions on Instrumentation and Measurement, 2020, 70: 1-13.
[13] Wang X H, Meng R X, Wang G T, et al. The research on fault diagnosis of rolling bearing based on current signal CNN-SVM[J]. Measurement Science and Technology, 2023, 34(12): 125021.
[14] Rezaeianjouybari B, Shang Y. Deep learning for prognostics and health management: state of the art, challenges, and opportunities[J]. Measurement, 2020, 163: 107929.
[15] Zhu X X, Hou D N, Zhou P, et al. Rotor fault diagnosis using a convolutional neural network with symmetrized dot pattern images[J]. Measurement, 2019, 138: 526-535.
[16] Wang S Y, Tian J Y, Liang P F, et al. Single and simultaneous fault diagnosis of gearbox via wavelet transform and improved deep residual network under imbalanced data[J]. Engineering Applications of Artificial Intelligence,2024, 133: 108146.
[17] Li C Y, Hu Y H, Jiang J W, et al. Fault diagnosis of a marine power-generation diesel engine based on the Gramian angular field and a convolutional neural network[J].Journal of Zhejiang University-SCIENCE A, 2024, 25(6): 470-482.
[18] Ding X X, He Q B. Energy-fluctuated multiscale feature learning with deep convnet for intelligent spindle bearing fault diagnosis[J]. IEEE Transactions on Instrumentation and Measurement, 2017, 66(8): 1926-1935.
[19] Liu D D, Cui L L, Cheng W D. A review on deep learning in planetary gearbox health state recognition: methods, applications, and dataset publication[J]. Measurement Science and Technology,2023, 35(1): 012002.
[20] Wen L L, Li X Y, Gao L. A transfer convolutional neural network for fault diagnosis based on ResNet-50[J]. Neural Computing and Applications, 2020, 32(10): 6111-6124.
[21] Wen L, Li X Y, Gao L, et al. A new convolutional neural network-based data-driven fault diagnosis method[J]. IEEE Transactions on Industrial Electronics, 2017, 65(7): 5990-5998.
[22] Zhang W, Li C H, Peng G L, et al. A deep convolutional neural network with new training methods for bearing fault diagnosis under noisy environment and different working load[J]. Mechanical Systems and Signal Processing, 2018, 100: 439-453.
[23] Xia M, Li T, Xu L, et al. Fault diagnosis for rotating machinery using multiple sensors and convolutional neural networks[J]. IEEE/ASME Transactions on Mechatronics, 2017, 23(1): 101-110.
[24] Li H M, Huang J Y, Gao M J, et al. Multi-view information fusion fault diagnosis method based on attention mechanism and convolutional neural network[J]. Applied Sciences, 2022, 12(22): 11410.
[25] Xie T L, Huang X F, Choi S K. Intelligent mechanical fault diagnosis using multisensor fusion and convolution neural network[J]. IEEE Transactions on Industrial Informatics, 2021, 18(5): 3213-3223.
[1] 刘昕晖,陈晫群,吕岩. 基于深度学习的工业故障诊断综述[J]. 吉林大学学报(工学版), 2026, 56(7): 1759-1779.
[2] 曹洁,陈志峰,王进花,陈莉. 基于扩散模型和DenseNet的少样本齿轮箱故障诊断方法[J]. 吉林大学学报(工学版), 2026, 56(7): 1787-1797.
[3] 冯志刚,任梦媛,董冰,于明月. 基于多频带特征图和改进SqueezeNet的滚动轴承故障诊断[J]. 吉林大学学报(工学版), 2026, 56(1): 96-108.
[4] 于江波,翁剑成,林鹏飞,孙宇星,柴娇龙. 基于混合Transformer的对外客运枢纽抵站客流预测模型[J]. 吉林大学学报(工学版), 2025, 55(7): 2251-2259.
[5] 冯志刚,王首起,于明月. 基于变分模态提取及轻量级网络的滚动轴承故障诊断[J]. 吉林大学学报(工学版), 2025, 55(6): 1883-1891.
[6] 刘广文,谢欣月,付强,才华,王伟刚,马智勇. 基于时空模板焦点注意的Transformer目标跟踪算法[J]. 吉林大学学报(工学版), 2025, 55(3): 1037-1049.
[7] 王娜,崔月磊,李杨,王子从. 基于小波包对数能量图的滚动轴承故障诊断方法[J]. 吉林大学学报(工学版), 2025, 55(2): 494-502.
[8] 冯志刚,张志远,董冰,于明月. 基于改进VME算法和mRVM的滚动轴承小样本复合故障诊断[J]. 吉林大学学报(工学版), 2025, 55(12): 3831-3839.
[9] 王鹏,宋亚飞,王晓丹,路艳丽,向前. FATIDS:面向类不平衡样本的物联网入侵检测方法[J]. 吉林大学学报(工学版), 2025, 55(12): 3986-3999.
[10] 陈永,安卓奥博,张娇娇. 基于旋转自注意力改进Mask RCNN的桥梁裂缝检测方法[J]. 吉林大学学报(工学版), 2025, 55(11): 3660-3672.
[11] 胡宏宇,张争光,曲优,蔡沐雨,高菲,高镇海. 基于双分支和可变形卷积网络的驾驶员行为识别方法[J]. 吉林大学学报(工学版), 2025, 55(1): 93-104.
[12] 余萍,赵康,曹洁. 基于优化A-BiLSTM的滚动轴承故障诊断[J]. 吉林大学学报(工学版), 2024, 54(8): 2156-2166.
[13] 王长建,刘久明,张锦洲,李斌. 基于高速摄影技术的行星减速箱故障激光序列脉冲诊断方法[J]. 吉林大学学报(工学版), 2024, 54(7): 1869-1875.
[14] 蒋磊,王子其,崔振宇,常志勇,时小虎. 基于循环结构的视觉Transformer[J]. 吉林大学学报(工学版), 2024, 54(7): 2049-2056.
[15] 梁礼明,周珑颂,尹江,盛校棋. 融合多尺度Transformer的皮肤病变分割算法[J]. 吉林大学学报(工学版), 2024, 54(4): 1086-1098.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!