✅作者简介：热爱科研的Matlab仿真开发者，擅长数据处理、建模仿真、程序设计、完整代码获取、论文复现及科研仿真。
🍎 往期回顾关注个人主页：Matlab科研工作室
🍊个人信条：格物致知,完整Matlab代码及仿真咨询内容私信。
🔥内容介绍
针对旋转机械故障诊断中，传统方法难以有效提取非平稳振动信号的时频特征、诊断精度不足且泛化能力有限的问题，本文提出基于连续小波变换（CWT）与多种深度学习及混合模型的故障诊断方案，以凯斯西储大学（CWRU）轴承故障数据集为研究对象，系统对比卷积神经网络（CNN）、残差网络（ResNet）、CNN-SVM、CNN-BiGRU、CNN-LSTM五种模型的故障诊断性能。首先通过连续小波变换将原始振动信号转换为高分辨率时频图，实现时域与频域特征的同步提取，解决传统时域、频域分析分离的局限；随后构建五种不同结构的诊断模型，分别完成特征深度挖掘与故障分类；最后基于CWRU数据集的多工况数据（不同故障类型、故障直径、负载条件）开展对比实验，验证各模型的诊断准确率、收敛速度及抗干扰能力。实验结果表明，ResNet模型在复合故障诊断中表现最优，诊断准确率可达99.8%，CNN-LSTM模型在强时序依赖故障识别中优势显著，而CNN-SVM模型在小样本场景下具有更高的训练效率，CWT的引入使所有模型的诊断精度均得到不同程度提升。本文研究可为旋转机械故障的精准、高效诊断提供理论支撑与实践参考，创新点在于构建多模型对比框架，结合CWT时频特征提取技术，突破单一模型的应用局限，适配不同工业场景的诊断需求。
关键词
故障诊断；连续小波变换；深度学习；凯斯西储大学数据集；卷积神经网络；残差网络；支持向量机；循环神经网络
1 引言
1.1 研究背景与意义
旋转机械作为工业生产、航空航天、交通运输等领域的核心设备，其运行状态直接决定了整个系统的可靠性与安全性，而轴承作为旋转机械的关键零部件，易因磨损、疲劳、冲击等因素产生故障，若未能及时诊断，可能导致设备停机、生产中断，甚至引发严重安全事故。据统计，旋转机械故障中约80%与轴承故障相关，因此实现轴承故障的早期精准诊断，对降低运维成本、保障系统稳定运行具有重要的工程价值与现实意义。
传统故障诊断方法主要依赖人工经验或基于信号处理的特征提取，如时域分析（峰值、均方根）、频域分析（快速傅里叶变换FFT）等，但这类方法存在明显局限：时域分析仅能捕捉幅值变化，无法定位频率特征；频域分析丢失时间信息，难以诊断瞬态故障；且手动特征提取依赖专家经验，泛化性差，难以适应复杂工业环境中的非平稳振动信号分析需求。
随着深度学习技术的快速发展，其强大的自动特征提取与模式识别能力为故障诊断提供了新的技术路径。CNN凭借出色的空间特征提取能力，广泛应用于故障时频图的分析；ResNet通过残差连接缓解深层网络的梯度消失问题，提升复杂特征的学习能力；SVM作为经典的分类算法，在小样本、高维特征分类中表现稳定；BiGRU与LSTM则擅长捕捉时序信号的依赖关系，适用于振动信号这类长序列数据的分析。然而，单一模型往往存在应用局限，且原始振动信号的特征冗余的问题，会影响模型的诊断性能。
连续小波变换（CWT）作为一种高效的时频分析方法，通过小波基函数的缩放与平移，实现时频局部化分析，能够自适应调整时频分辨率，有效捕捉非平稳振动信号中的瞬态冲击特征，恰好弥补传统信号处理方法的不足。凯斯西储大学轴承故障数据集作为故障诊断领域的基准数据集，包含多种故障类型、故障程度及运行工况，数据可靠性高、覆盖范围广，为本文的模型验证提供了理想的实验载体。
1.2 研究现状
近年来，国内外学者围绕小波变换与深度学习结合的故障诊断方法开展了大量研究。Hou等提出基于CWT与双流卷积融合的模拟电路故障诊断方法，通过CWT将原始波形转换为二维时频图，结合注意力机制提升模型的特征表达能力，在多种滤波电路实验中取得了优异的诊断效果。部分研究将CWT与CNN结合，实现了轴承故障的端到端诊断，避免了手动特征提取的繁琐流程，但单一CNN模型在深层特征挖掘与时序信息捕捉上存在不足。
ResNet的出现为深层网络训练提供了有效解决方案，研究者将ResNet与小波变换结合，用于复杂故障模式的诊断，显著提升了模型的特征表达能力。同时，混合模型成为研究热点，CNN-SVM通过CNN提取特征、SVM完成分类，在小样本场景下展现出良好的性能；CNN与BiGRU、LSTM的结合，则实现了空间特征与时序特征的协同提取，进一步提升了时序相关故障的诊断精度。
然而，现有研究多聚焦于单一模型或两种模型的对比，缺乏对多种主流模型的系统对比分析，且针对CWRU数据集的多工况适配性研究不够全面，未能明确不同模型在不同故障场景下的应用优势与局限。基于此，本文选取五种主流模型，结合CWT时频特征提取技术，以CWRU数据集为研究对象，开展系统的对比实验，明确各模型的性能差异，为实际工业故障诊断中的模型选择提供参考。
1.3 研究内容与创新点
1.3.1 研究内容
数据预处理与时频特征提取：获取凯斯西储大学轴承故障数据集，进行数据清洗、归一化、样本划分等预处理；采用连续小波变换将原始振动信号转换为二维时频图，实现时域与频域特征的同步提取，为模型输入提供高质量特征。
诊断模型构建：分别构建CWT-CNN、CWT-ResNet、CWT-CNN-SVM、CWT-CNN-BiGRU、CWT-CNN-LSTM五种故障诊断模型，明确各模型的网络结构、超参数设置及训练策略，适配CWT时频图与振动信号的特征特点。
实验验证与对比分析：基于CWRU数据集的多工况数据（正常状态、内圈故障、外圈故障、滚动体故障，不同故障直径、负载条件），开展模型训练与测试，对比五种模型的诊断准确率、训练时间、收敛速度及抗干扰能力，分析各模型的应用优势与局限。
模型优化与总结：针对实验中发现的问题，提出模型优化方向；总结各模型的适用场景，为实际故障诊断中的模型选择提供指导。
1.3.2 创新点
构建多模型系统对比框架，选取五种主流诊断模型（CNN、ResNet、CNN-SVM、CNN-BiGRU、CNN-LSTM），结合CWT时频特征提取技术，全面对比各模型在CWRU数据集上的故障诊断性能，突破单一模型研究的局限。
优化CWT时频特征提取流程，结合CWRU数据集的振动信号特点，选择适配的小波基函数与参数，提升时频图的特征辨识度，解决原始振动信号特征冗余、瞬态特征难以捕捉的问题，进一步提升模型诊断精度。
基于CWRU数据集的多工况条件（不同故障类型、故障直径、负载）开展实验，分析各模型在不同场景下的适配性，明确模型的应用边界，为实际工业故障诊断提供更具针对性的参考。
1.4 论文结构
本文共分为6章，具体结构如下：第1章为引言，阐述研究背景、意义、现状、内容及创新点；第2章为相关理论基础，介绍连续小波变换、五种诊断模型的核心原理；第3章为实验数据集与预处理方法，详细说明凯斯西储大学数据集的特点及CWT时频特征提取流程；第4章为诊断模型构建与训练，明确各模型的网络结构与训练策略；第5章为实验结果与分析，通过对比实验验证各模型的性能；第6章为总结与展望，总结本文研究成果，提出未来研究方向。
2 相关理论基础
2.1 连续小波变换（CWT）
3 诊断模型构建与训练
3.1 模型整体框架
本文构建的故障诊断模型整体框架分为三层：数据输入层、特征提取层、故障分类层。数据输入层为CWT生成的128×128二维时频图；特征提取层由CWT与时频特征提取网络（CNN、ResNet）组成，负责提取时频图中的空间特征及时序特征；故障分类层由全连接层、SVM、BiGRU、LSTM组成，负责将提取的特征映射到具体的故障类别，实现故障诊断。五种模型的核心差异在于特征提取层与故障分类层的结构不同，具体如下：
CWT-CNN模型：CWT+CNN（卷积层+池化层+全连接层）；
CWT-ResNet模型：CWT+ResNet（残差模块+卷积层+池化层+全连接层）；
CWT-CNN-SVM模型：CWT+CNN（卷积层+池化层）+SVM；
CWT-CNN-BiGRU模型：CWT+CNN（卷积层+池化层）+BiGRU+全连接层；
CWT-CNN-LSTM模型：CWT+CNN（卷积层+池化层）+LSTM+全连接层。
3.2 各模型具体结构
3.2.1 CWT-CNN模型结构
CWT-CNN模型的网络结构如下（输入为128×128×1的灰度时频图）：
卷积层1：32个3×3卷积核，激活函数为ReLU，步长为1， padding为“same”，输出特征图尺寸为128×128×32；
池化层1：最大值池化，池化核为2×2，步长为2，输出特征图尺寸为64×64×32；
卷积层2：64个3×3卷积核，激活函数为ReLU，步长为1， padding为“same”，输出特征图尺寸为64×64×64；
池化层2：最大值池化，池化核为2×2，步长为2，输出特征图尺寸为32×32×64；
扁平化层：将池化层2的输出转换为一维特征向量，维度为32×32×64=65536；
全连接层1：神经元数量为128，激活函数为ReLU；
全连接层2（输出层）：神经元数量为13（对应13种工况），激活函数为Softmax，输出故障类别概率。
3.2.2 CWT-ResNet模型结构
CWT-ResNet模型基于ResNet18构建，核心包含4个残差模块，每个残差模块包含2个卷积层和1个残差连接，具体结构如下：
输入层：128×128×1的灰度时频图；
初始卷积层：64个3×3卷积核，激活函数为ReLU，步长为1， padding为“same”，输出特征图尺寸为128×128×64；
残差模块1：2个64×3×3卷积核，激活函数为ReLU，残差连接，输出特征图尺寸为128×128×64；
残差模块2：2个128×3×3卷积核，激活函数为ReLU，残差连接，输出特征图尺寸为64×64×128；
残差模块3：2个256×3×3卷积核，激活函数为ReLU，残差连接，输出特征图尺寸为32×32×256；
残差模块4：2个512×3×3卷积核，激活函数为ReLU，残差连接，输出特征图尺寸为16×16×512；
全局平均池化层：输出特征图尺寸为1×1×512；
全连接层（输出层）：神经元数量为13，激活函数为Softmax，输出故障类别概率。
3.2.3 CWT-CNN-SVM模型结构
CWT-CNN-SVM模型的特征提取部分与CWT-CNN模型一致（卷积层1-池化层2），分类部分采用SVM替代全连接层，具体结构如下：
特征提取部分：与CWT-CNN模型的卷积层1、池化层1、卷积层2、池化层2一致，输出特征图尺寸为32×32×64；
扁平化层：将特征图转换为一维特征向量，维度为65536；
SVM分类器：采用RBF核函数，正则化参数C设置为10，核函数参数γ设置为0.001，输入为一维特征向量，输出为故障类别。
3.2.4 CWT-CNN-BiGRU模型结构
CWT-CNN-BiGRU模型的特征提取部分为CWT+CNN，时序特征捕捉与分类部分为BiGRU+全连接层，具体结构如下：
特征提取部分：与CWT-CNN模型的卷积层1-池化层2一致，输出特征图尺寸为32×32×64；
特征重构：将32×32×64的特征图转换为32×2048的特征序列（每一行作为一个时间步的特征）；
BiGRU层：隐藏层神经元数量为128， dropout系数为0.5，输出维度为256（正向GRU与反向GRU各128维）；
全连接层1：神经元数量为64，激活函数为ReLU；
全连接层2（输出层）：神经元数量为13，激活函数为Softmax，输出故障类别概率。
3.2.5 CWT-CNN-LSTM模型结构
CWT-CNN-LSTM模型的结构与CWT-CNN-BiGRU类似，仅将BiGRU层替换为LSTM层，具体结构如下：
特征提取部分：与CWT-CNN模型的卷积层1-池化层2一致，输出特征图尺寸为32×32×64；
特征重构：将32×32×64的特征图转换为32×2048的特征序列；
LSTM层：隐藏层神经元数量为128， dropout系数为0.5，输出维度为128；
全连接层1：神经元数量为64，激活函数为ReLU；
全连接层2（输出层）：神经元数量为13，激活函数为Softmax，输出故障类别概率。
3.3 模型训练参数设置
五种模型均采用PyTorch框架实现，训练环境为：CPU为Intel Core i7-12700H，GPU为NVIDIA RTX 3060，内存为16GB，操作系统为Windows 11，Python版本为3.9。训练参数统一设置如下，确保实验的公平性：
优化器：采用Adam优化器，学习率为0.001，权重衰减为1e-4，动量为0.9；
损失函数：CWT-CNN、CWT-ResNet、CWT-CNN-BiGRU、CWT-CNN-LSTM模型采用交叉熵损失函数，CWT-CNN-SVM模型采用SVM的默认损失函数；
训练轮次（Epoch）：50轮，采用早停策略（Early Stopping），当验证集准确率连续5轮不提升时，停止训练，避免过拟合；
批次大小（Batch Size）：32；
数据增强：训练过程中采用随机翻转、高斯噪声添加等数据增强方法，增强模型的泛化能力。
4 总结与展望
4.1 研究总结
本文以凯斯西储大学轴承故障数据集为研究对象，针对旋转机械故障诊断中传统方法的局限，提出基于连续小波变换（CWT）与五种深度学习及混合模型的故障诊断方案，通过系统的对比实验，得出以下结论：
连续小波变换（CWT）能够有效提取非平稳振动信号的时频特征，将原始振动信号转换为二维时频图，为模型输入提供高质量特征，显著提升了五种模型的故障诊断精度，解决了传统时域、频域分析分离的局限。
五种模型的诊断性能存在明显差异：CWT-CNN-LSTM模型的诊断准确率与抗干扰能力最强，CWT-ResNet模型的综合性能最优，CWT-CNN-SVM模型的训练效率最高，CWT-CNN模型结构最简单，CWT-CNN-BiGRU模型在双向时序捕捉上具有优势。
不同模型适用于不同的故障诊断场景，可根据实际需求（诊断精度、实时性、计算资源、故障类型）选择合适的模型，为实际工业故障诊断提供了针对性的参考。
4.2 研究不足与未来展望
4.2.1 研究不足
本文的研究仍存在一些不足：一是仅采用凯斯西储大学数据集进行实验验证，数据集的场景较为单一，未能在实际工业现场数据中验证模型的泛化能力；二是模型的超参数采用经验设置，未进行系统的优化，可能影响模型的性能；三是未考虑多源数据融合（如振动信号与温度、声音数据融合），未能充分利用多维度故障信息；四是模型的轻量化程度不足，难以适配嵌入式设备等资源有限的场景。
4.2.2 未来展望
针对上述不足，未来的研究方向主要包括：
扩大实验数据集，引入实际工业现场的轴承故障数据，验证模型的泛化能力，同时结合多源数据融合技术，提升故障诊断的鲁棒性。
采用智能优化算法（如粒子群优化、遗传算法）对模型的超参数进行系统优化，进一步提升模型的诊断性能与训练效率。
开展模型轻量化研究，采用知识蒸馏、模型剪枝等技术，压缩模型参数量，降低计算复杂度，实现模型在嵌入式设备上的部署，满足实时诊断需求。
探索迁移学习技术的应用，利用预训练模型（如ImageNet上的ResNet）进行跨工况、跨设备迁移，解决小样本故障诊断问题。
结合注意力机制，增强模型对关键故障特征的提取能力，进一步提升模型的诊断精度与抗干扰能力。
⛳️ 运行结果
🔗 参考文献
[1] 戢晓峰,郭雅诗,陈方,等.基于CNN-LSTM-Attention 组合模型的黄金周旅游客流预测——以大理州为例[J].干旱区资源与环境, 2025, 39(3):200-208.
[2] 谢醉冰,马龙涛,王吉利,等.基于LSTM-CNN特征融合和BO-SVM的GIS绝缘缺陷识别研究[J].电机与控制学报, 2025, 29(12):182-194.
📣 部分代码
🎈 部分理论引用网络文献，若有侵权联系博主删除
👇 关注我领取海量matlab电子书和数学建模资料
🏆团队擅长辅导定制多种科研领域MATLAB仿真，助力科研梦：
🌈 各类智能优化算法改进及应用
生产调度、经济调度、装配线调度、充电优化、车间调度、发车优化、水库调度、三维装箱、物流选址、货位优化、公交排班优化、充电桩布局优化、车间布局优化、集装箱船配载优化、水泵组合优化、解医疗资源分配优化、设施布局优化、可视域基站和无人机选址优化、背包问题、 风电场布局、时隙分配优化、 最佳分布式发电单元分配、多阶段管道维修、 工厂-中心-需求点三级选址问题、 应急生活物质配送中心选址、 基站选址、 道路灯柱布置、 枢纽节点部署、 输电线路台风监测装置、 集装箱调度、 机组优化、 投资优化组合、云服务器组合优化、 天线线性阵列分布优化、CVRP问题、VRPPD问题、多中心VRP问题、多层网络的VRP问题、多中心多车型的VRP问题、 动态VRP问题、双层车辆路径规划（2E-VRP）、充电车辆路径规划（EVRP）、油电混合车辆路径规划、混合流水车间问题、 订单拆分调度问题、 公交车的调度排班优化问题、航班摆渡车辆调度问题、选址路径规划问题、港口调度、港口岸桥调度、停机位分配、机场航班调度、泄漏源定位
🌈 机器学习和深度学习时序、回归、分类、聚类和降维
2.1 bp时序、回归预测和分类
2.2 ENS声神经网络时序、回归预测和分类
2.3 SVM/CNN-SVM/LSSVM/RVM支持向量机系列时序、回归预测和分类
2.4 CNN|TCN|GCN卷积神经网络系列时序、回归预测和分类
2.5 ELM/KELM/RELM/DELM极限学习机系列时序、回归预测和分类
2.6 GRU/Bi-GRU/CNN-GRU/CNN-BiGRU门控神经网络时序、回归预测和分类
2.7 ELMAN递归神经网络时序、回归\预测和分类
2.8 LSTM/BiLSTM/CNN-LSTM/CNN-BiLSTM/长短记忆神经网络系列时序、回归预测和分类
2.9 RBF径向基神经网络时序、回归预测和分类
2.10 DBN深度置信网络时序、回归预测和分类
2.11 FNN模糊神经网络时序、回归预测
2.12 RF随机森林时序、回归预测和分类
2.13 BLS宽度学习时序、回归预测和分类
2.14 PNN脉冲神经网络分类
2.15 模糊小波神经网络预测和分类
2.16 时序、回归预测和分类
2.17 时序、回归预测预测和分类
2.18 XGBOOST集成学习时序、回归预测预测和分类
2.19 Transform各类组合时序、回归预测预测和分类
方向涵盖风电预测、光伏预测、电池寿命预测、辐射源识别、交通流预测、负荷预测、股价预测、PM2.5浓度预测、电池健康状态预测、用电量预测、水体光学参数反演、NLOS信号识别、地铁停车精准预测、变压器故障诊断
🌈图像处理方面
图像识别、图像分割、图像检测、图像隐藏、图像配准、图像拼接、图像融合、图像增强、图像压缩感知
🌈 路径规划方面
旅行商问题（TSP）、车辆路径问题（VRP、MVRP、CVRP、VRPTW等）、无人机三维路径规划、无人机协同、无人机编队、机器人路径规划、栅格地图路径规划、多式联运运输问题、 充电车辆路径规划（EVRP）、 双层车辆路径规划（2E-VRP）、 油电混合车辆路径规划、 船舶航迹规划、 全路径规划规划、 仓储巡逻
🌈 无人机应用方面
无人机路径规划、无人机控制、无人机编队、无人机协同、无人机任务分配、无人机安全通信轨迹在线优化、车辆协同无人机路径规划
🌈 通信方面
传感器部署优化、通信协议优化、路由优化、目标定位优化、Dv-Hop定位优化、Leach协议优化、WSN覆盖优化、组播优化、RSSI定位优化、水声通信、通信上传下载分配
🌈 信号处理方面
信号识别、信号加密、信号去噪、信号增强、雷达信号处理、信号水印嵌入提取、肌电信号、脑电信号、信号配时优化、心电信号、DOA估计、编码译码、变分模态分解、管道泄漏、滤波器、数字信号处理+传输+分析+去噪、数字信号调制、误码率、信号估计、DTMF、信号检测
🌈电力系统方面
微电网优化、无功优化、配电网重构、储能配置、有序充电、MPPT优化、家庭用电
🌈 元胞自动机方面
交通流 人群疏散 病毒扩散 晶体生长 金属腐蚀
🌈 雷达方面
卡尔曼滤波跟踪、航迹关联、航迹融合、SOC估计、阵列优化、NLOS识别
🌈 车间调度
零等待流水车间调度问题NWFSP、置换流水车间调度问题PFSP、混合流水车间调度问题HFSP、零空闲流水车间调度问题NIFSP、分布式置换流水车间调度问题 DPFSP、阻塞流水车间调度问题BFSP
👇
