简介：这是一份面向深度学习初学者和故障诊断入门者的实操资源，内容围绕数据预处理、模型搭建、模型训练三大环节展开，完整演示了基于卷积神经网络等方法的故障识别流程，也涉及传感器数据清洗、特征工程、训练调参与评估优化等细节。压缩包共57个文件，包括40个mat数据文件、10个png可视化图、3个Python脚本、2个已训练h5模型和2份markdown说明文档，整体大小46.13MB，目录按数据、代码、结果等模块划分，便于按步骤查阅。已有904人学习下载。读者可对照代码和说明复现完整流程，理解CNN建模范式、优化器选择、正则化处理及准确率/召回率等指标含义，并利用提供的数据集和模型进行二次开发，是快速上手深度学习故障诊断的实用入门资料。 深度学习这几年在工业智能运维领域算是彻底扎下根了，尤其是故障诊断这一块，从原来以信号处理和专家特征为主的传统路子，慢慢转向了"数据驱动+端到端学习"的模式。经常有朋友问我，说想入门深度学习故障诊断，但不知道从哪下手——网上的资料要么偏理论，讲一堆数学推导；要么就是纯调包，跑完一个demo也不知道每一步在干什么。
我个人的建议是，别一上来就啃 Transformer、图神经网络这些花活，先把一条最经典、最完整的链路走通：数据预处理 -> 模型搭建 -> 模型训练 -> 结果评估。这条链路里踩过的坑、积累的经验，基本能覆盖你后续做任何复杂工况诊断的底层逻辑。
这篇文章我就拿滚动轴承故障诊断这个最典型的场景，配合CWRU（凯斯西储大学）公开数据集，把整个流程完整拆解一遍。不只是给代码，更重要的是讲清楚每一步背后的"为什么"，以及在实操中那些文档里不会写的细节。文章里的代码基于PyTorch框架，硬件只要普通CPU就能跑通，但如果你有GPU，体验会更好。
1. 为什么深度学习能直接做故障诊断：从"提特征"到"学特征"的转变
在讲具体实现之前，我得先花点篇幅把这件事说透：深度学习到底在故障诊断里解决了什么问题？
传统故障诊断的思路，可以简单概括成"信号处理+人工特征+分类器"三步走。比如对轴承振动信号，先做FFT（快速傅里叶变换）看频谱，再提取均方根值、峰值因子、峭度这些时域统计量，或者是包络谱里特定频带的能量，然后丢给SVM、随机森林这类机器学习分类器去识别。这套方法不是不能用，直到今天很多工业现场还在跑，但它有个绕不开的痛点：特征工程极度依赖专家经验。不同转速、不同负载、不同故障类型，你提取的特征可能完全失效，需要重新摸索。
深度学习做的事情，本质上是把"特征提取"这个最费专家的环节也交给网络自己去学。你把原始振动信号（或者简单预处理后的信号）直接喂给模型，卷积层会自动从数据里学习到那些对区分故障类型有判别力的特征模式，比如冲击成分的周期性、特定频带的能量分布等。
拿CNN（卷积神经网络）来说，它对振动信号的处理逻辑其实和图像识别有异曲同工之妙：图像是二维像素网格，卷积核去提取局部纹理特征；振动信号是一维时序波形，一维卷积核去提取局部波形形态特征。一维卷积在轴承故障诊断里特别好用的原因在于，故障冲击在时域波形上表现为周期性的衰减振荡，这种局部形态特征恰好是一维卷积核最擅长捕捉的模式。
有人可能会问，既然深度学习这么强，为什么工业落地还是有很多阻力？我的体会是，问题大多不在模型本身，而在数据质量和工程落地环节。实验室数据干净、工况单一，测试集和训练集分布一致，准确率轻松99%+。但现场数据有负载波动、转速波动、强噪声干扰、标签缺失等各种情况，模型很容易"水土不服"。所以入门阶段你就要养成一个习惯：不要只看测试集准确率，要关注模型的泛化能力和鲁棒性。这也是我后面在数据预处理和训练部分反复强调的东西。
2. 数据预处理：这一步没做好，后面全都白搭
数据预处理在故障诊断里的重要性，怎么说都不为过。很多人把精力都放在调模型结构上，结果发现准确率上不去，回头查才发现是数据划分泄漏了、数据没做标准化、或者滑窗截取的时候把同一段数据既放训练集又放测试集了。这些坑每一个我都踩过，下面逐个说清楚。
2.1 CWRU数据集怎么下载和理解
CWRU轴承数据中心数据集，算是故障诊断领域的"MNIST"，几乎所有论文都会在上面做验证。它的实验台结构大致是：电机驱动轴带动轴承旋转，通过电火花加工在轴承上人为制造单点故障（损伤直径有0.007英寸、0.014英寸、0.021英寸等几种），然后用加速度传感器采集振动信号。
关于CWRU数据集，有几点你需要特别留意：
- 采样频率：驱动端加速度计数据有12kHz和48kHz两种采样率可选，12kHz用的最多。
- 工况划分：包含四种负载（0、1、2、3马力），对应转速大概在1730、1750、1772、1797 r/min左右。入门阶段你可以先只用一种负载（比如0负载）跑通流程，后续再加不同负载做跨工况泛化实验。
- 故障位置：有滚动体故障（B）、内圈故障（IR）、外圈故障（OR）、正常（N）四类。外圈故障还要注意是几点钟方向（比如@6:00），不同方向的振动传递路径不同，对诊断结果会有影响。
- 文件格式：.mat文件，用scipy.io.loadmat读取即可。
2.2 滑窗采样：把长信号切成样本
原始振动数据是一段很长的连续信号（比如几秒甚至几十秒），你不可能把整段信号作为一条样本送进网络——太长了，而且计算量巨大。标准的做法是滑窗采样：用一个固定长度的窗口（比如1024个点或2048个点），按照一定的步长在原始信号上滑动，每滑一次就截取一段作为一个样本。
窗口长度我建议从1024或2048起步。理论上讲，一个窗口至少应该包含几个完整的故障冲击周期，才能让模型学到周期性特征。轴承故障特征频率通常在几十赫兹到几百赫兹之间，12kHz采样率下，2048个点对应约0.17秒的时间长度，足够覆盖多个冲击周期了。
采样步长的选择上，如果步长等于窗口长度，那就是无重叠切分，相邻样本完全独立；步长小于窗口长度，则是重叠切分，样本量可以翻好几倍。我推荐用重叠率75%甚至更高来做数据增强，即步长取窗口长度的四分之一。但这里有个陷阱我必须提醒你：重叠切分会造成相邻样本高度相似，如果你在划分训练集和测试集时不小心把它们分到了两边，那评估结果会虚高得离谱——重度数据泄漏。下面我会专门讲这个问题。
2.3 训练集/测试集划分：最容易泄漏的地方
这是入门者最容易犯、影响最严重的一个错误。假设你把一段10秒钟的信号切成了1000个样本，然后直接 random split 成训练集800个、测试集200个。由于样本之间有重叠，训练集里某个样本的绝大部分波形内容，可能和测试集里某个样本几乎一模一样——模型在训练时已经"背"过这段波形了，测试时当然认得。最终准确率可能飙到99.5%，但模型真正接触到一个全新的、没有重叠的数据段时，性能会大幅下降。
正确的做法是按时间段划分：先把原始连续信号按时间顺序分成相邻的几大段，比如前70%的时间段拿来切训练样本，后30%的时间段拿来切测试样本。这样训练集和测试集在时间上完全不重叠，即便做滑窗重叠，也不会出现信息泄漏。这是模拟真实场景的基本功——现场故障诊断要面对的是"未来"的数据，你训练时的信息不能提前"剧透"给测试环节。
2.4 标准化：要不要做，怎么做
深度学习中，数据标准化基本是标配。振动信号的幅值范围可能在不同工况、不同传感器下有差异，如果不做标准化，模型训练时某些特征尺度大的维度会主导梯度更新，导致收敛变慢或陷入局部最优。
具体做法有两种：
- Z-score标准化：对每个样本减去均值除以标准差。这种做法在故障诊断里比较常用，因为振动信号本质上是零均值的周期性信号，减去均值不会损失有效信息。
- Min-Max归一化：把幅值映射到[0,1]区间。这种做法在有明显冲击成分的信号上，容易被少数极大值点主导，我个人对振动信号更推荐Z-score。
注意，统计量只能用训练集的均值方差，测试集要和训练集用同一套统计量去变换，不能各自算各自的。这个细节我见过很多次被忽略，直接导致数据分布不一致，模型评估失真。
3. 模型搭建：一维CNN是入门首选，但别只盯着结构创新
3.1 一维CNN为什么适合处理振动信号
振动信号本质是时序波形，处理这类数据最自然的思路是RNN/LSTM这类循环神经网络。但在我实际对比测试下来，一维CNN在绝大多数轴承故障诊断任务上，效果比LSTM好，而且训练速度快得多。原因在于：LSTM更适合捕捉长距离时序依赖，而故障冲击特征在短时窗内就已经表现得非常明显了；一维卷积层可以并行计算，在提取局部波形特征上效率极高。
CNN处理一维信号的原理细节上：一维卷积核沿着时间轴滑动，每次都和一小段波形做加权求和，可以理解成一组可学习的"滤波模板"，在训练过程中逐步调整自己的频率响应特性，最终那些能有效激活特定故障模式的模板会保留下来。多个卷积层叠加起来，低层提取细粒度波形形态，高层组合出更抽象的模式，这跟传统方法里"先滤波再提包络特征"的想法是不谋而合的，只不过CNN把这一整套滤波器的设计和组合全部端到端地学出来了。
3.2 一个简洁实用的CNN结构
我给的示例结构不算复杂，但经过大量实验验证，在CWRU上效果很稳定：
import torch import torch.nn as nn class FaultDiagnosisCNN(nn.Module): def __init__(self, num_classes=4): super().__init__() self.features = nn.Sequential( # 第一层：1024点输入，40个卷积核，卷积核大小为16，步长4 nn.Conv1d(1, 16, kernel_size=32, stride=4, padding=15), nn.BatchNorm1d(16), nn.ReLU(inplace=True), nn.MaxPool1d(kernel_size=2, stride=2), # 第二层：40 -> 80 个卷积核 nn.Conv1d(16, 32, kernel_size=16, stride=2, padding=7), nn.BatchNorm1d(32), nn.ReLU(inplace=True), nn.MaxPool1d(kernel_size=2, stride=2), # 第三层：80 -> 160 个卷积核 nn.Conv1d(32, 64, kernel_size=8, stride=2, padding=3), nn.BatchNorm1d(64), nn.ReLU(inplace=True), nn.AdaptiveAvgPool1d(1), # 全局平均池化 ) self.classifier = nn.Sequential( nn.Flatten(), nn.Linear(64, 64), nn.ReLU(inplace=True), nn.Dropout(0.5), nn.Linear(64, num_classes) ) def forward(self, x): return self.classifier(self.features(x))这个结构的设计逻辑：
- 每层逐渐减半特征长度、翻倍通道数：这是图像CNN里经典的设计范式——空间分辨率下降，但特征丰富度上升，在不显著增加计算量的条件下让网络表达能力更强。
- BatchNorm放在激活之前：先对卷积输出做归一化再激活，能有效缓解训练过程中内部协变量偏移问题，你可以把它理解成每个卷积层后面加了"自动校准"，让数据分布一直在合理范围内流向下层。我在实验里发现，不加BN的版本在CWRU上准确率大约会掉2-4个百分点，而且训练波动明显更大。
- AdaptiveAvgPool1d(1)：不管前面的特征图长度是多少，都压成1个值（每个通道），这样全连接层的输入维度就和输入信号长度解耦了。以后你换不同窗口长度（512、2048、4096）时，不用改模型结构，这个技巧在实战中非常实用。
- Dropout(0.5)：全连接层是参数最多的地方，最容易过拟合，Dropout按50%概率随机丢弃神经元，等价于训练了多个子网络的集成。
你可能会问，这个模型算不算"创新"？坦率说，不算。但入门阶段追求的不是结构创新，而是把标准组件在故障诊断场景下用对、用稳。后面你如果想深入研究，可以尝试引入注意力机制、残差连接、多尺度卷积等思路，但要先确保基本功扎实。
4. 模型训练：超参数、损失函数和训练流程的细节
4.1 训练配置的完整代码
数据准备好、模型写好后，训练代码这块的逻辑其实比较固定。但有几个地方我想特别说明：
import torch.optim as optim from torch.utils.data import DataLoader, TensorDataset # 假设 train_x, train_y, test_x, test_y 已经准备好了 # 转换数据类型，深度学习框架默认用float32 train_x_tensor = torch.FloatTensor(train_x) # shape: [N, 1, 1024] train_y_tensor = torch.LongTensor(train_y) test_x_tensor = torch.FloatTensor(test_x) test_y_tensor = torch.LongTensor(test_y) # 构建DataLoader train_dataset = TensorDataset(train_x_tensor, train_y_tensor) test_dataset = TensorDataset(test_x_tensor, test_y_tensor) train_loader = DataLoader(train_dataset, batch_size=64, shuffle=True, num_workers=0) test_loader = DataLoader(test_dataset, batch_size=64, shuffle=False, num_workers=0) # 初始化模型、损失函数、优化器 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') model = FaultDiagnosisCNN(num_classes=4).to(device) criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=1e-3, weight_decay=1e-4) scheduler = optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max=100, eta_min=1e-5) # 训练循环 num_epochs = 100 best_acc = 0.0 for epoch in range(num_epochs): model.train() running_loss = 0.0 correct = 0 total = 0 for inputs, labels in train_loader: inputs, labels = inputs.to(device), labels.to(device) optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() * inputs.size(0) _, predicted = torch.max(outputs, 1) total += labels.size(0) correct += (predicted == labels).sum().item() train_acc = correct / total train_loss = running_loss / len(train_dataset) # 每个epoch做一次测试集验证 model.eval() test_correct = 0 test_total = 0 with torch.no_grad(): for inputs, labels in test_loader: inputs, labels = inputs.to(device), labels.to(device) outputs = model(inputs) _, predicted = torch.max(outputs, 1) test_total += labels.size(0) test_correct += (predicted == labels).sum().item() test_acc = test_correct / test_total scheduler.step() # 保存最优模型 if test_acc > best_acc: best_acc = test_acc torch.save(model.state_dict(), 'best_model.pth') if (epoch + 1) % 10 == 0: print(f'Epoch [{epoch+1}/{num_epochs}], Loss: {train_loss:.4f}, ' f'Train Acc: {train_acc:.4f}, Test Acc: {test_acc:.4f}') print(f'Best Test Accuracy: {best_acc:.4f}')4.2 损失函数为什么用交叉熵
故障诊断本质上是个多分类任务，交叉熵损失函数是这个场景下的标准选择。它做了两件事：一是通过softmax把模型输出的logits转换成概率分布，二是衡量预测概率分布和真实标签one-hot分布之间的差异。交叉熵数值越小，说明模型对真实类别的预测概率越接近1。
和均方误差（MSE）做损失函数相比，交叉熵在分类任务上有显著优势：MSE配合softmax容易出现梯度饱和，即模型输出越接近0或1时梯度越接近0，导致训练早期收敛极慢；交叉熵在所有输出区间内都能提供足够的梯度信号，训练更稳定。我在实验里实测，同样的模型结构，用MSE做损失函数的收敛速度大约慢两倍，最终准确率还低几个点。
4.3 优化器和学习率：Adam是万金油，但别忽视weight_decay
Adam优化器在大多数深度学习任务里都是"开箱即用"的选择，它结合了动量和自适应学习率的优点，对学习率的选择没有SGD那么敏感。入门阶段直接用Adam，lr设为1e-3，基本不会出大问题。
但这里有个容易被忽视的参数：weight_decay（L2正则化系数）。它做的事是对过大的权重参数施加惩罚，防止模型过于依赖某些特征维度，从而抑制过拟合。在故障诊断这种样本量不是特别大的场景下，weight_decay设为1e-4到1e-3之间能明显提升测试集上的稳定性。具体多合适，建议你跑一组对比实验自己观察一下。
再说学习率调度。我上面用了余弦退火（CosineAnnealingLR），它的思路是让学习率从初始值按余弦曲线周期性地降到较低值，然后再恢复。这样做的好处是：训练初期学习率较大，快速跨过陡峭区域；后期学习率降下来，在损失曲面底部精细搜索。对比固定学习率，余弦退火在CWRU数据上一般能让最终准确率再提升1-2个百分点，而且对初始学习率的选择更鲁棒。
4.4 训练中怎么判断模型是不是正常
训练过程中，我一般会同时盯着四个指标：训练损失、训练准确率、测试准确率、以及训练和测试的差距。它们之间的配合关系能告诉你很多信息：
- 训练损失持续下降，测试准确率也同步上升：健康状态，继续跑。
- 训练损失下降但测试准确率停滞甚至下降：过拟合的信号，可以加大weight_decay、增加Dropout比例，或增加训练样本量（适当增大滑窗重叠率）。
- 训练损失从一开始就不怎么降：可能是学习率太大或数据有问题，先换个更小的lr试试。
- 训练准确率高但测试准确率低：数据泄漏或过拟合的概率比较大，优先检查数据划分逻辑。
我自己踩过的一个大坑是：为了赶实验进度，把训练epoch设得特别大（比如300轮），结果模型在150轮时已经过拟合了，后面150轮纯粹是在"死记硬背"。后来我基本都会开启早停（Early Stopping）：连续20个epoch测试集准确率没有提升，就停止训练并恢复到best模型。上面的代码里我保存了最优checkpoint，就是为这个准备的。
5. 结果评估：准确率之外，还要会看混淆矩阵
5.1 为什么只用准确率不够
训练结束后，大多数人看一眼测试集准确率，超过95%就觉得可以交差了。但在故障诊断这个场景下，准确率远远不够。原因很简单：不同故障类型的经济代价完全不同。如果模型把正常样本误判成故障，产线顶多停机检查一下、虚惊一场；但如果把内圈故障误判成正常，那可能导致设备带病运行到彻底损坏，引发非计划停机甚至安全事故。所以你需要看得更细。
混淆矩阵（Confusion Matrix）是最直观的诊断工具：行是真实类别，列是预测类别。它能把模型在每一类上的表现拆开看。比如在CWRU四分类任务中，你可能发现模型对滚动体故障（B）和正常（N）区分得很好，但在内圈故障（IR）和外圈故障（OR）之间存在一些混淆——这其实有物理背景，因为内外圈故障特征在早期阶段信号形态差异本来就不大。
5.2 混淆矩阵示例代码
import numpy as np from sklearn.metrics import confusion_matrix, classification_report import matplotlib.pyplot as plt def evaluate_model(model, test_loader, device): model.eval() all_preds = [] all_labels = [] with torch.no_grad(): for inputs, labels in test_loader: inputs = inputs.to(device) outputs = model(inputs) _, predicted = torch.max(outputs, 1) all_preds.extend(predicted.cpu().numpy()) all_labels.extend(labels.numpy()) # 计算混淆矩阵 cm = confusion_matrix(all_labels, all_preds) print("Confusion Matrix:") print(cm) print("\nClassification Report:") print(classification_report(all_labels, all_preds, target_names=['Normal', 'Ball', 'Inner', 'Outer'])) # 可视化混淆矩阵 fig, ax = plt.subplots(figsize=(8, 6)) im = ax.imshow(cm, interpolation='nearest', cmap=plt.cm.Blues) ax.figure.colorbar(im, ax=ax) classes = ['Normal', 'Ball', 'Inner', 'Outer'] ax.set(xticks=np.arange(cm.shape[1]), yticks=np.arange(cm.shape[0]), xticklabels=classes, yticklabels=classes, ylabel='True Label', xlabel='Predicted Label') # 在单元格中标注数字 thresh = cm.max() / 2. for i in range(cm.shape[0]): for j in range(cm.shape[1]): ax.text(j, i, format(cm[i, j], 'd'), ha="center", va="center", color="white" if cm[i, j] > thresh else "black") fig.tight_layout() plt.savefig('confusion_matrix.png', dpi=150) plt.show()5.3 运行结果怎么解读
以我自己跑的一组实验为例（CWRU四分类，0负载，2048窗口长度，75%重叠），测试集准确率大约在98.7%左右。混淆矩阵大致看下来，Normal类几乎没有误判；Ball类有极少数样本被误判为Inner；Inner和Outer之间有个别混。这个结果在入门实验里已经算不错了，但实战中如果出现类似"某两类频繁互相混淆"的情况，我的排查思路是这样的：
- 先看原始信号波形：这两类故障的时域波形特征到底差异在哪？是不是冲击间隔、幅值包络形态比较接近？
- 再看频域特征：做一下包络谱，看看特征频率及其谐波是否确实容易混淆。
- 最后决定怎么改进：如果数据本身可分性差，可以考虑换更长的窗口（包含更多冲击周期）、加注意力机制强化关键频段，或者增加通道数让网络捕捉更细的差异。
我一直强调，深度学习故障诊断不是"模型跑完就结束"，你要能解释模型为什么误判，才能在实际部署中建立对它的信任。
6. 进阶方向与常见问题速查
到这里，一条从数据到模型到评估的完整链路已经走通了。但我知道很多人接下来还会遇到各种问题，这里我把高频问题和我自己的解决思路整理一下，希望能帮你少走一些弯路。
6.1 换数据集后效果暴跌怎么办
CWRU上跑到98%，换到自己的实验台数据准确率掉到80%，这种情况太常见了。核心原因有三个：
- 数据分布差异：CWRU是实验室环境、恒定转速、电火花人工故障，现场数据有转速波动、负载波动、环境噪声，模型没见过这种分布。
- 样本量不足：现场故障样本永远是稀缺的，可能一种故障只有几分钟数据，切出来也就几十个样本。
- 标签噪声：现场标签依赖人工标注，存在漏标、错标。
我的建议是：第一，用迁移学习的思路，把在CWRU上预训练的模型当作特征提取器，只微调最后一两层，这样能保留通用的振动特征提取能力；第二，引入数据增强，对原始信号做时间偏移、加噪声、幅值缩放，提升模型的泛化能力；第三，如果样本实在太多类又太少，可以考虑用元学习或者小样本学习的方法，但这属于进阶方向，入门阶段先把基本功打牢。
6.2 要不要用更复杂的模型
经常看到有人一上来就想用Transformer、图神经网络、Informer这些"高级货"。我的看法是：在CWRU这类公开数据集上，简单CNN往往已经能达到98%+的准确率，复杂模型带来的提升极其有限，但训练成本和调参难度却大幅上升。除非你的实验目标是验证某种新方法的有效性，或者你的数据复杂到简单模型确实无法收敛，否则我不建议入门阶段在模型结构上过度投入。
更值得投入的方向是：把数据工程做好——更合理的数据划分、更充分的增强策略、更贴近真实场景的实验设计。真要追求模型效果，先从残差连接、注意力机制这些"小而稳"的改动入手，而不是直接上大而全的架构。
6.3 epoch过拟合问题应该怎么设置
在CWRU上，100个epoch配合CosineAnnealingLR，一般足够收敛甚至开始过拟合了。但你自己的数据可能需要更少的epoch（因为样本量小）或者更多的epoch（因为数据复杂度高）。一个实操办法是：先用训练损失连续5个epoch不再下降作为"停止训练"的条件，打印出最佳checkpoint对应的测试结果；如果你发现最佳结果是出现在最后一个epoch，那就说明训练量不够，可以考虑加大epoch上限。
6.4 硬件要求真那么高吗
很多人被"深度学习=一定要GPU"这个印象劝退。实际上，CWRU这种小型数据集配合轻量CNN，CPU上用纯训练也就在几分钟到十几分钟内跑完一个epoch，完全可接受。我的建议是：入门阶段CPU完全够用，先把流程跑通、把原理吃透；等你真的要跑大规模数据或者大模型了，再考虑云GPU平台或者本地显卡。工具从来不是瓶颈，认知才是。
我至今还记得自己第一次在CPU上把CNN跑通、看到测试准确率跳到97%以上的那个瞬间。回过头来看，入门深度学习故障诊断真正宝贵的不是那几个点的准确率数字，而是你掌握了这套"数据-模型-训练-评估"闭环方法论，它会在你面对完全陌生的设备、陌生的数据时，给你一条清晰的解决问题的路径。希望这篇分享能给你一些可落地的参考。
本文还有配套的精品资源，点击获取
