一、算法原理与流程
1. BP神经网络结构
- 输入层：故障特征维度（如振动频谱、电流时域特征等）
- 隐藏层：通常1-3层，节点数取输入层1.5-2倍
- 输出层：Softmax激活函数（多分类）或Sigmoid（二分类）
2. 核心流程
二、代码
%% 清空环境
clear; clc; warning off;
%% 数据加载（示例为电机故障数据）
load('fault_data.mat'); % 包含X(特征矩阵)和Y(标签向量)
%% 数据预处理
% 数据划分（70%训练，15%验证，15%测试）
cv = cvpartition(size(X,1),'HoldOut',0.3);
X_train = X(training(cv),:);
Y_train = Y(training(cv),:);
X_val = X(test(cv),:);
Y_val = Y(test(cv),:);
% 归一化处理
[X_train,norm_params] = mapminmax(X_train',0,1);
X_val = mapminmax('apply',X_val',norm_params);
X_test = mapminmax('apply',X_test',norm_params);
% 标签独热编码
Y_train = ind2vec(Y_train');
Y_val = ind2vec(Y_val');
Y_test = ind2vec(Y_test');
%% 网络结构设计
input_neurons = size(X_train,2); % 输入层节点数
hidden_neurons = 12; % 隐藏层节点数（经验公式：2*输入层）
output_neurons = size(Y_train,2); % 输出层节点数
net = feedforwardnet(hidden_neurons, 'traingdx'); % 动量梯度下降算法
net.trainParam.epochs = 1000; % 最大训练次数
net.trainParam.goal = 1e-5; % 目标误差
net.trainParam.lr = 0.01; % 学习率
net.trainParam.showWindow = 0; % 关闭训练窗口
%% 模型训练
[net,tr] = train(net,X_train',Y_train');
%% 性能评估
% 测试集预测
Y_pred = net(X_test');
[~,Y_pred_class] = max(Y_pred);
[~,Y_test_class] = max(Y_test);
% 混淆矩阵
C = confusionmat(Y_test_class,Y_pred_class);
accuracy = sum(diag(C))/sum(C(:));
disp(['分类准确率: ', num2str(accuracy*100, '%.2f'), '%']);
% ROC曲线
figure;
plotconfusion(Y_test_class,Y_pred_class);
title('混淆矩阵分析');
%% 特征可视化（可选）
figure;
ploterrhist(Y_test_class - Y_pred_class);
title('分类误差分布');
xlabel('预测误差'); ylabel('频数');
三、参数优化
1. 网络结构优化
隐藏层选择：通过交叉验证确定最佳节点数（常用范围8-20）
激活函数：
net.layers{ 1}.transferFcn = 'tansig'; % 隐藏层激活函数 net.layers{ 2}.transferFcn = 'softmax'; % 输出层激活函数
2. 训练参数调优
| 参数 | 推荐范围 | 优化方法 |
|---|---|---|
| 学习率(lr) | 0.001-0.1 | 自适应学习率（traingda） |
| 动量因子(mc) | 0.5-0.9 | 动量梯度下降(traingdm) |
| 正则化系数 | 0.0001-0.01 | L2正则化防止过拟合 |
3. 正则化实现
net.performFcn = 'mse'; % 均方误差
net.performParam.regularization = 0.001; % 正则化系数
四、常见问题解决
1. 过拟合问题
解决方法：
- 增加Dropout层
- 早停法（Early Stopping）
net.divideFcn = 'dividerand'; % 数据划分 net.divideParam.trainRatio = 0.7; net.divideParam.valRatio = 0.15; net.divideParam.testRatio = 0.15;
2. 收敛速度慢
优化方案：
- 使用Levenberg-Marquardt算法
net = feedforwardnet(hidden_neurons, 'trainlm');
参考代码 使用BP神经网络对故障数据实现分类以及matlab实现 www.youwenfan.com/contentali/81390.html
该方案通过特征工程、网络优化和正则化策略，在保证精度的同时提升模型泛化能力。
