简介：面向设备运维与故障诊断场景的BP神经网络齿轮箱故障诊断源码包，以Matlab（m文件）形式呈现完整网络构建、训练与诊断流程。文件围绕齿轮箱振动信号处理展开，涵盖数据特征提取、网络层结构设计、反向传播权重更新到最终故障模式判别等核心环节，可通过调整隐藏层节点数与训练参数适配不同工况数据，适合机械工程、自动化或数据挖掘方向学生及工业检测人员入门与二次开发。包体轻量，仅1个m脚本文件，大小1KB，便于直接阅读修改，无需复杂环境配置即可运行。资源目前已吸引196人学习，说明故障诊断方向上有一定参考热度。通过这份程序，读者可直观掌握BP网络在齿轮磨损、轴承异常等典型故障识别中的实际配置思路，也能以此为基础替换自己的实测数据，快速验证诊断效果，进一步拓展到同类旋转机械的状态监测与故障预警任务。
1. 齿轮箱故障诊断为什么选BP神经网络：从振动信号到故障标签的闭环
车间里那台减速机连续一周夜班振动超标，老师傅拆开一看，第二级齿轮已经断了两颗齿。这种“坏了才知道”的被动维修，每次停机都在烧钱。BP神经网络在齿轮箱故障诊断里解决的就是这件事：让设备自己“说出”哪里出了问题，把振动信号自动归类到磨损、断齿、轴承损伤等故障类型。这套资源的核心是一个可直接运行的 MATLAB 实现，BP_net.m 从数据载入、特征提取到网络训练、诊断输出，完整走完整个链路。适合手里已有一批振动数据、想快速验证 BP 方案可行性的工程师和研究生；如果你是第一次接触神经网络，照流程跑通代码，也能把这条诊断链路看明白。这类 BP 故障诊断代码并不是什么黑匣子，它的上限取决于特征工程，下限才是网络调参。
2. 数据准备与特征提取：把原始振动信号变成网络能吃的特征向量
BP 网络不认原始波形，它吃的是特征向量。这一步决定了诊断效果的上限，后面网络调得再好，也只是在下限附近做优化。数据准备的基本流程是：采集振动信号→截断成样本→提取特征→打标签→划分数据集，中间任何一步出了问题，模型都白练。
2.1 数据采集的基本配置：传感器、采样率与工况记录
常见的采集方案是加速度传感器吸附在齿轮箱轴承座或箱体表面，测点离啮合区越近，信号里包含的故障信息越完整。传感器一般用 IEPE 型加速度计，灵敏度 100 mV/g 左右，通过数据采集卡接入上位机。采样率需要覆盖齿轮啮合频率的 5～10 倍。齿轮啮合频率通常等于齿数乘以转频，例如一个 23 齿的高速齿轮在 1500 r/min 下，啮合频率约 575 Hz，考虑到谐波和边频带，采样率取 12.8 kHz 以上比较稳妥；我常用 20 kHz。
采样时长方面，单次记录至少包含 10 个轴转周期，这样提取出的特征才有统计意义。特别提醒：采集时同步记录转速、负载、油温等工况参数，这些信息不参与训练，但用于数据分组——不同工况下的振动特征分布差异很大，混在一起训练容易让网络学到“工况”而不是“故障”。
| 项目 | 建议值 | 说明 |
|---|---|---|
| 传感器 | IEPE 加速度计 | 吸附式安装，100 mV/g 灵敏度 |
| 采样率 | 12.8～20 kHz | 覆盖啮合频率及其谐波 |
| 单段时长 | ≥10 个轴转周期 | 保证特征统计稳定 |
| 工况记录 | 转速、负载、油温 | 用于数据分组和异常排查 |
数据采集这块没什么玄学，但漏记工况是我见过最多的低级失误。没有工况信息，训练时模型把转速变化当成故障变化，部署到现场一测一个不准。
2.2 时域特征提取：均值、峰值、峭度到底在看什么
时域特征计算成本低，适合作为第一组输入。常用特征包括均值、均方根、峰值、峭度和偏度。均值反映信号直流分量，实际振动信号均值接近零，但传感器漂移时会有变化；均方根反映振动能量总体大小，齿轮磨损后能量通常会上升；峰值对冲击信号敏感，齿面剥落、断齿都会产生瞬态冲击；峭度对正常信号接近 3，出现冲击类故障时数值显著升高，是判断早期故障的敏感指标；偏度反映波形不对称性，对不均匀磨损有一定指示作用。
MATLAB 实现如下：
% 时域特征提取：输入一段振动信号x，输出特征向量feat function feat = extract_time_features(x) N = length(x); feat(1) = mean(x); % 均值：对传感器偏置敏感 feat(2) = sqrt(mean(x.^2)); % 均方根：振动能量总体水平 feat(3) = max(abs(x)); % 峰值：对冲击类故障敏感 feat(4) = kurtosis(x); % 峭度：正常约等于3，故障时显著升高 feat(5) = skewness(x); % 偏度：波形的非对称程度 end说明：kurtosis 和 skewness 在 MATLAB 统计工具箱里直接可用，Octave 下也能调用。每个特征的物理意义已经写在注释里，这是后面调参时判断“为什么这个特征有效”的依据。比如峭度升高首先要怀疑冲击类故障，如果某个故障类别的峭度和其他类别完全重叠，那这个特征对它就没有区分度。
2.3 频域特征提取：傅里叶变换与边频带计算逻辑
齿轮箱故障在振动频谱上的特征非常明显：啮合频率及其谐波处会有较大的谱峰；故障会导致啮合频率两侧出现以转频为间隔的边频带，边频带能量与故障程度密切相关。所以频域特征是齿轮箱诊断里最重要的特征来源。
傅里叶变换提取频域特征的代码：
% 频域特征提取：输入振动信号x、采样率fs、啮合频率fm、转频fr function feat = extract_freq_features(x, fs, fm, fr) X = fft(x); L = length(x); P = abs(X(1:floor(L/2)+1)) / L; % 单边幅值谱 f = (0:floor(L/2)) * fs / L; % 频率坐标 % 啮合频率附近的峰值幅值（频带宽度取±5 Hz） idx = find(f >= fm-5 & f <= fm+5); feat(1) = max(P(idx)); % 边频带能量：啮合频率±1倍转频区间内的谱线和 idx_side = find(f >= fm-fr-5 & f <= fm+fr+5); feat(2) = sum(P(idx_side)); % 2倍啮合频率及其边频带能量 idx2 = find(f >= 2*fm-5 & f <= 2*fm+5); feat(3) = max(P(idx2)); end说明：fft 后取单边幅值谱，除以点数是为了把幅值归一化到真实振动幅值量纲。啮合频率附近的峰值幅值是齿轮状态的直接体现；边频带能量由故障调制产生，能量越高表示故障越明显。如果你有高阶故障，可以继续取 3 倍、4 倍啮合频率附近的特征，但要注意高频段衰减快，特征稳定性会下降。频域特征提取最怕频率分辨率不够，采样时长太短时谱线间隔大，边频带的细节根本看不出。
2.4 样本构建与数据集划分：一个决定评估真实性的关键步骤
把原始信号截成固定长度，每段算一组特征，加上标签构成一个样本。标签用 one-hot 编码：比如 [1 0 0 0] 代表正常，[0 1 0 0] 代表均匀磨损，[0 0 1 0] 代表断齿，[0 0 0 1] 代表轴承故障。
数据集划分是最容易翻车的一步。很多人直接 randperm 随机打乱，这样做会让同一段连续数据同时出现在训练集和测试集，网络相当于见过测试答案，评估结果虚高，到了现场立刻打回原形。我一般按以下原则划分：
- 按时间先后顺序，前 70% 作训练集，后 30% 作测试集；
- 如果数据来自不同工况或不同转速段，按工况分段后再抽取，避免同类工况被拆到两边；
- 划分索引固定保存，训练和测试用同一份索引。
| 划分方式 | 做法 | 风险评估 |
|---|---|---|
| 随机打乱 | randperm 直接切 | 数据泄漏，评估乐观 |
| 按时间切分 | 前 70% 训练后 30% 测试 | 更接近现场场景 |
| 按工况切分 | 每种工况各取 70% | 适合多工况数据 |
随机打乱不是完全不能用，如果每个样本都来自独立采集批次，泄漏风险确实小；但齿轮箱振动是连续时间序列，相邻两个样本高度相关，随机切大概率把相关样本分进两边。宁可保守一点按时间段切，评估结果才有参考价值。
3. 构建与训练 BP 网络：三层结构设计与反向传播参数详解
特征向量准备好了，下一步是把特征映射到故障类别。BP 网络做这件事靠的是多层神经元的权重叠加和反向传播的误差修正。如果按最常见的 BP 神经网络结构图画出来，就是“输入层—隐藏层—输出层”三层结构，核心逻辑不复杂，但参数选择上有大量细节。
3.1 网络结构怎么定：输入层、隐藏层、输出层的节点数量
BP 网络一般用三层结构：输入层、单隐藏层、输出层。输入层节点数等于特征向量维数，比如 2.3 节的时域 5 维加频域 3 维共 8 维，那输入层就是 8 个节点。输出层节点数等于故障类别数，4 类故障就是 4 个节点，用 one-hot 编码作为目标输出。
隐藏层节点数没有解析解，工具书里给过经验公式：H = sqrt(M+N)+a，M 是输入节点数，N 是输出节点数，a 取 1～10。对 8 输入 4 输出的问题，H 在 11～20 之间。经验上从下限开始试更稳妥：隐藏层节点太少，网络欠拟合；节点太多，参数数量爆炸式增长，小样本下过拟合风险非常高。还有个额外原则：隐藏层节点数不要超过训练样本数的十分之一，否则网络能直接记住样本，失去泛化能力。
需要注意的是，输出层 4 个节点和 4 类故障之间不是“第 1 个节点强、第 2 个节点弱”的关系，而是经过 logsig 激活后输出 0～1 之间的概率值，哪个节点最接近 1，就判成哪一类。训练时目标也是 one-hot 向量，这个匹配关系必须一致。
3.2 训练参数：学习率、动量、迭代次数与目标误差
训练参数直接影响收敛速度和最终效果。下表是我在 BP 故障诊断里常用的参数范围：
| 参数 | 常用范围 | 作用与选择说明 |
|---|---|---|
| 学习率 lr | 0.01～0.1 | 过大训练震荡，过小收敛慢 |
| 动量因子 mc | 0.9～0.95 | 平滑梯度方向，减少局部极小值停滞 |
| 最大迭代次数 | 1000～5000 | 防止无限训练，配合早停使用 |
| 目标误差 goal | 1e-5～1e-3 | 达到即停，通常由验证集决定 |
| 隐藏层激活 | logsig / tansig | 分类任务默认 logsig 即可 |
| 输出层激活 | logsig | 输出范围 0～1，匹配 one-hot 编码 |
这些参数不是拍脑袋定的。学习率 0.05 配合动量 0.9 是很多工业故障诊断案例的起点配置；如果你的样本量很大、特征噪声高，可以把学习率降到 0.01 起步，多观察前 500 代的误差曲线再决定调大还是调小。目标误差不要设得太激进，1e-5 对大多数诊断场景足够，设成 1e-8 只会让训练时间无限拉长，收益却趋近于零。
3.3 训练代码与收敛判据：BP_net.m 的核心逻辑
资源包里的 BP_net.m 走的流程和我下面这段代码是一致的：构建网络、固定划分、设参、训练、输出。
% 构建单隐藏层BP网络 hiddenSize = 15; % 隐藏层节点数，按经验公式下限 net = feedforwardnet(hiddenSize); net.trainFcn = 'traingdx'; % 带动量的梯度下降 net.layers{1}.transferFcn = 'logsig'; % 隐藏层激活 net.layers{2}.transferFcn = 'logsig'; % 输出层激活 net.performFcn = 'mse'; % 误差函数：均方误差 % 固定数据划分，避免随机泄漏 net.divideFcn = 'divideind'; net.divideParam.trainInd = trainIdx; % 训练集索引 net.divideParam.valInd = valIdx; % 验证集索引 net.divideParam.testInd = testIdx; % 测试集索引 % 训练参数 net.trainParam.epochs = 3000; net.trainParam.lr = 0.05; net.trainParam.mc = 0.9; net.trainParam.goal = 1e-5; net.trainParam.showWindow = true; % 打开训练窗口 % 训练：X_train是特征矩阵，T_train是对应one-hot标签 % 注意feedforwardnet要求输入是“样本数×特征数”，所以转置 [net, tr] = train(net, X_train', T_train');逻辑说明：feedforwardnet(hiddenSize) 创建的是单隐藏层前馈网络，hiddenSize 是隐藏层节点数。trainFcn 设为 traingdx，表示带动量项的梯度下降加自适应学习率，这是故障诊断里最常见的配置，既有动量项的稳定，又有自适应学习率的加速。divideFcn 设为 divideind，按给定索引切分数据，这一步是防止数据泄漏的关键，也是我反复强调的一个习惯。
参数说明：trainIdx/valIdx/testIdx 需要自行生成并保存，保证每次训练复用同一份划分。performFcn 默认是 mse，如果想把误分类的惩罚加大，可以换成 crossentropy，但要确认工具箱支持。训练结束后，tr 结构体里记录了每一轮的误差曲线，我一般把它画出来，观察验证集误差的走势。
提示：训练集误差和验证集误差同步下降是正常状态；验证集误差降到最低点后开始回升，就是过拟合启动的信号，这时不用等全部迭代跑完，直接取最低点对应的网络作为最终模型。
训练过程中还有个细节：样本量不大的时候，train 函数内部的增量训练和批处理模式对结果影响很小，我习惯用默认的批量梯度下降，收敛稳定，调试时也更可控。
4. 模型评估与调优：从训练误差到诊断准确率的落差
训练完不等于能用。很多模型训练阶段好看，测试阶段掉链子，问题出在评估方式和调参路径上。这一章先讲清楚怎么评估，再给出一套可执行的调参顺序。
4.1 评估指标：混淆矩阵、准确率与召回率之间的优先级
故障诊断中只报一个准确率远远不够。设备故障往往是小概率事件，如果正常样本占 90%，模型全部猜“正常”也能有 90% 准确率，但这对诊断毫无意义。正确的做法是看每一类故障的召回率。召回率低意味着这类故障经常被漏报，漏报的代价是设备带病运行，严重时整个产线停机。
% 测试集推理与评估 Y_pred = sim(net, X_test'); % 网络输出 [~, pred] = max(Y_pred, [], 1); % 取最大输出作为预测类别 [~, true] = max(T_test', [], 1); % one-hot标签还原为类别编号 cm = confusionmat(true, pred); % 混淆矩阵 acc = sum(diag(cm)) / sum(cm(:)); % 整体准确率 recall = diag(cm) ./ sum(cm, 2); % 每个类别的召回率逻辑说明：sim 是 MATLAB 的推理接口，新版直接用 net(X_test') 效果一样。confusionmat 把真实标签和预测标签交叉统计成矩阵，第 i 行第 j 列表示“真实 i 类被判成 j 类的样本数”。recall 输出的是每个类别的漏报情况，比整体准确率更值得关注。举个例子，如果断齿的召回率只有 60%，说明 10 次断齿里会有 4 次被当成正常或磨损，这绝对是不能接受的。
4.2 欠拟合与过拟合：拿训练集本身做个测试就能定位
判断模型状态有个很笨但有效的方法：把训练集再喂给网络算一遍准确率，然后和测试集对比：
- 训练集准确率低（比如低于 85%）→ 欠拟合，模型容量不够或特征不够区分；
- 训练集准确率很高、测试集明显更低 → 过拟合，模型记住了训练样本；
- 两者都接近且高 → 状态正常。
欠拟合的解决方向是增加隐藏层节点数，或补充更有效的特征，尤其频域的啮合频率和边频带特征。过拟合的解决方向是减少隐藏层节点数、提前停止训练，或调大正则化系数。一个我踩过多次的误区：一看到测试集差，立刻加训练数据或调网络结构，但真正原因往往在数据划分——训练集和测试集若来自同一时段的高相关样本，评估结果虚高；划分正确之后，欠拟合和过拟合的问题才会真实暴露出来。
4.3 一条可行的调参路径：先结构后参数
我习惯的调参顺序是固定的，不随意跳步：
- 固定隐藏层节点数在下限（经验公式的 a 取 1），学习率 0.05，跑通全流程，记录测试准确率；
- 隐藏层节点数以 5 为步长递增，试到上限（a 取 10），画出准确率对比曲线，这一步基本能确定结构范围；
- 锁定表现最好的结构，再调学习率：0.1、0.05、0.01、0.005 四档，在验证集上比较，选验证误差最小的档位；
- 最后回头看各类召回率，若某类故障仍偏低，回到特征层面分析，而不是继续动网络。
这套路径的核心逻辑是：网络结构对容量的影响是决定性的，学习率只是微调；特征决定上限，结构决定逼近上限的能力。调参有时候确实像玄学，但按这个顺序走，至少每次改动都能定位到具体因素。血泪经验就是：先结构后参数，比漫无目的地瞎试省一半时间。
5. 避坑指南：BP_net 实战中五个典型翻车场景
这一章记录我在复现和部署 BP 类诊断程序时踩过的真实坑，每一条都对应具体的错误现象、原因和排查路径。
5.1 现象：训练误差震荡不收敛
现象：训练曲线像锯齿一样上下跳动，迭代了上千次误差仍不下降，有时误差值直接从几万起步。
原因：最常见的是两个。第一，输入特征没有归一化，轴承振动幅值、峭度、温度等量纲差异巨大，梯度更新被大幅值特征主导，网络在误差曲面里来回震荡；第二，学习率设得过大，在梯度陡峭的区域一步跨出最优区间。偶尔还有激活函数用成 purelin 导致输出无界的情况。
解决：训练前统一归一到 [0,1] 或 [-1,1]，保存归一化参数用于测试阶段，这是最有效的一步。然后学习率降到 0.01 以下重跑，观察前几次迭代的误差变化，如果下降稳定，再逐步往上调。输出层激活函数务必用 logsig，不要用线性函数。
5.2 现象：测试集准确率比训练集低一大截
现象：训练集准确率 95% 以上，测试集只有 70% 左右。你以为是过拟合，把网络结构缩到很小、加正则项，测试结果却没有根本改善。
原因：大概率不是过拟合，而是数据划分产生了数据泄漏。连续采样得到的振动信号，相邻样本高度相关，随机打乱后划分，相关样本被分到训练和测试两边，模型在训练时已经见过测试样本的“影子”，评估结果虚高。一旦模型部署到现场，精度立刻崩溃。
解决：按时间段划分数据集，前 70% 训练后 30% 测试，并在创建网络时用 divideind 固定索引。多工况数据先按工况分组，再在各组内按时间切分，保证任何一段连续采集数据不会同时出现在两个集合里。
5.3 现象：某些故障类别几乎识别不出来
现象：正常样本召回率 90% 以上，某种故障召回率只有 40%，大量样本被分到正常类。
原因：第一是类别不平衡，齿轮箱故障样本采集成本高，数据集中正常样本占 60% 以上，网络梯度更新被多数类主导；第二是这类故障与正常状态的时域、频域特征重叠度高，所选特征本身不能有效区分。
解决：少数类用过采样加小幅噪声的方式扩增到与多数类同一量级；也可以给损失函数按类别比例加权。如果扩增后依然区分不开，回到特征层面，补加该类故障特有的频域特征（比如磨损类故障的边频带能量变化），比继续调网络参数有效得多。
5.4 现象：特征算了一大堆，诊断效果反而更差
现象：从 5 个特征一路加到 30 多个，测试准确率不升反降，训练时间变长，网络输出还不稳定。
原因：特征维度增长但样本量没变，小样本在高维空间里变得极度稀疏，这就是经典的维度灾难。同时很多特征之间高度相关，冗余信息被网络当成噪声吸收，干扰了权重学习。
解决：特征先做相关性筛选，保留与故障标签相关性高、彼此相关性低的组合；或先用主成分分析把特征压缩回 10 维左右再进网络。特征不是越多越好，能区分故障的才是好特征。
5.5 现象：每次运行结果都不一样，复现不了
现象：同一个脚本、同一批数据，今天跑准确率 88%，明天跑变成 89.5%，三次跑结果各有差异，想在报告里给出唯一结论都难。
原因：两个随机源。一是网络权重初始化随机，不同初始点可能收敛到不同局部极小值；二是数据划分若用了随机函数，每次划分不同，测试集都换了。两个因素叠加，结果自然不一致。
解决：脚本开头固定随机种子，MATLAB 新版用 rng(2024)，老版本用 rand('state', 2024)。数据划分用固定索引。如果追求稳健，可以用多次重复训练取平均值的方式报告结果，但每次训练之间也要固定随机种子来保证可复现。这条看起来小，却是工程交付里最容易丢分的细节。
6. 从离线诊断到在线预警：模型部署与增量更新的一个可行套路
模型在实验台上准确率高，不等于现场能用。部署时最容易漏的一件事：把训练阶段的归一化参数、网络权重、阈值一起打包，绝不能在新数据上重新计算归一化，否则在线数据分布一变，输出立刻飘。
6.1 保存与加载：一个简单的部署结构
% 训练完成后统一保存 save('bp_model.mat', 'net', 'ps_in', 'ps_out', 'fs', 'f_mesh', 'thresh');ps_in 和 ps_out 是 mapminmax 生成的归一化结构体，在线推理时用 mapminmax('apply', x_new, ps_in) 保持同一套处理流程。加载后推理流程如下：
% 在线推理：新来一段振动信号x_new feat_new = extract_features(x_new, fs, f_mesh); feat_norm = mapminmax('apply', feat_new, ps_in); y = sim(net, feat_norm'); [~, label] = max(y);6.2 滑动窗口、阈值报警与误报抑制
现场实时数据是无限流，常见套路是开一个固定长度滑动窗口（比如 1 秒数据），滑一步处理一步，窗口重叠率 50% 可以平滑预测波动。连续 N 个窗口判为同一故障才触发报警，N 取 3～5，能压掉大部分单窗口误报。阈值不要拍脑袋定，用正常工况下输出的最大值留出 20% 余量作为报警线，比凭感觉设 0.8 可靠得多。
6.3 模型漂移与定期重训
设备磨损是渐变过程，特征分布会缓慢漂移，半年后老模型效果明显变差。我的习惯是每月把新采集且已人工确认的故障样本追加进训练集，用旧权重作初始值继续训练几个 epoch 做增量微调，同时用最近一个月的验证集监控准确率，跌破 90% 就触发全面重训。
从那以后，我每次部署诊断模型都强制自己走完这一套：数据分割不留泄漏、归一化参数随模型一起持久化、随机种子写进脚本、部署时阈值和模型一起交付。这些步骤看起来琐碎，每一条都是踩坑换来的。希望帮到你。
本文还有配套的精品资源，点击获取
