手把手教你:基于深度学习的滚动轴承故障诊断|第 6 篇 · 一维卷积与残差模块:83% 到 95% 的原理拆解

本文是《滚动轴承故障诊断》系列的第 6 篇,也是全系列最有含金量的一篇 (算法核心)。

系列完整目录见文末【专栏导航】。本篇讲解的两个模型结构代码见公众号。

一、从"拍平"到"看波形":深度学习赢在哪

第 5 篇讲到,随机森林把样本"拍平"成 2000 个数字,它看不见波形的形状------这就是它卡在 73% 的根因。

神经网络走的是完全不同的路:它让模型自己去看波形。

一维卷积神经网络(1D-CNN)不把样本摊平,而是让一个"滑动的小窗口"(卷积核)顺着时间轴一路扫过去 ,每扫到一个局部,就提取这里的"形状特征"------比如某个故障的冲击脉冲长什么样。这就是自动特征工程:不用人设计特征,模型自己学。

所以深度学习相对传统方法的本质区别就一句话:随机森林看"一串数",CNN 看"一段波形"。

二、一维卷积是怎么"看波形"的

拿本项目第一层 Conv1D(64, 3) 举例:

  • 3 是卷积核长度:每次看相邻 3 个采样点;
  • 64 是核的个数:同一段波形用 64 种不同的"眼光"去看,各提取一种局部模式;
  • 窗口一路滑过整条信号,生成 64 组"局部响应"------这就是特征图。

再经过池化(下采样、压缩信息)、继续卷积,网络就一层层把"细小的局部模式"组合成"更高层的整体判断"。CNN 对波形这类有局部结构的数据,天生合适。

三、本项目 CNN 的网络结构

python 复制代码
Input(1000,2) → Conv1D(64,3) → MaxPooling1D(16) → 全局平均池化 → Dense(32) → Dropout(0.5) → softmax(4)

一层卷积 + 一次池化 + 一个分类头,非常浅。

浅到学不好------实测准确率只有 约 83% ,而且内圈故障这一类的召回率明显偏低:一部分内圈样本被误判成了滚动体。

为什么?因为内圈故障的冲击频率高、和滚动体故障的波形存在相似性,单层卷积的特征提取能力太弱,分不清它们。这正是引出残差网络的最佳铺垫。

四、残差模块:那条著名的"捷径"

残差网络(ResNet)的改动很小,却很重要:在网络里加了"残差连接"(Residual Block)。

看它的内部结构(本项目残差块):

复制代码
输入 x
   ↓
Conv1D(relu) → BN → Conv1D(无激活) → BN
   ↓
F(x)  ← 卷积分支算出来的"增量"
   ↓
   F(x) + x   ← 把原始输入直接加回来(捷径!)
   ↓
   ReLU → 输出

核心就一句:输出 = F(x) + x ,其中 x 是块的输入,F(x) 是两层卷积的变换。

多出来的这个 + x 就是那条"捷径"(shortcut)------它不学任何参数,只是把输入原样"抄近道"叠回来。

五、为什么一条捷径能救回 12 个点

这要从"深层网络训不动"说起。网络越深,反向传播时梯度逐层衰减,传到浅层几乎为 0------浅层学不动,整个网络就"退化"了(这叫梯度消失/网络退化)。

残差连接的作用恰好相反:

  • +x 给梯度提供了一条无损的捷径通道 ,浅层也能直接收到误差信号,训得动了;
  • 网络因此敢加深(本项目用 5 个残差块),特征表达能力上去了,就能分得清内圈和滚动体了。

所以道理一句话:残差连接解决了"深网络训不动"的问题,而更深的网络才有更强的特征提取能力------这就是本项目从 83% 涨到 95% 的根本机制,也是你答辩时最能讲清楚的一个点。

六、两个模型的结构对比

层 1D-CNN CNN+ResNet
输入 (1000, 2) (1000, 2)
卷积/池化 Conv1D(64) + MaxPool Conv1D(32)→Conv1D(64)→MaxPool
残差块 无 5 个残差块
分类头 Dense(32)→Dropout→softmax Dense(256)→Dropout→softmax
参数量 少 多
测试准确率 ~83% 95%+

注意:同样数据、同样切分、同样评估,唯一的系统差别就是"加不加残差"。控制变量做干净了,结论才有说服力------这也是论文实验设计最该学会的一课。

七、本篇小结

  • 深度学习的优势:让模型自己从波形里"看特征",不靠人工特征工程;
  • CNN 浅了分不清内圈/滚动体(83%);ResNet 用残差连接让网络加深而不退化,提到 95%+;
  • 核心知识点:输出 = F(x) + x,捷径解决梯度消失;
  • 两个模型完整结构与可运行脚本见同名公众号。

专栏导航

篇次 主题 状态
1 项目总览与三模型对比 ✅
2 环境部署 ✅
3 CWRU 数据集深度解析 ✅
4 数据预处理源码走读 ✅
5 随机森林实战 ✅
6 一维卷积与残差模块(本文) ✅
7 模型训练与评估 排期中
8 可视化诊断系统 排期中
9 论文报告与答辩话术 排期中

下一篇预告:《模型训练与评估:复现全部实验结果、读懂混淆矩阵》,手把手带你跑完三个模型,并把结果变成论文里的对比表。


版权说明:CWRU 数据集版权归凯斯西储大学所有,仅限学习研究使用。