当前位置：首页 > news >正文

WOA-BP时间序列预测算法

news 2026/6/27 2:31:12

基于鲸鱼优化算法优化BP神经网络的(WOA-BP)的时间序列预测WOA-BP时间序列 matlab代码注：暂无Matlab版本要求--推荐2018B版本及以上

最近在搞时间序列预测的项目，发现传统BP神经网络总容易陷入局部最优，调参调到怀疑人生。偶然看到鲸鱼优化算法（WOA）这个黑科技，试了把两者的结合，效果居然意外不错。今天咱们不整虚的，直接上代码说人话。

先整点核心代码热热身，WOA的主循环长这样：

% WOA参数初始化 Max_iter=50; % 最大迭代次数 SearchAgents_no=30; % 种群数量 dim=4; % 优化参数个数（输入层节点数+隐含层节点数） % 主循环 for t=1:Max_iter a = 2 - t*(2/Max_iter); % 收缩系数 a2=-1+t*(-1/Max_iter); % 螺旋系数 for i=1:SearchAgents_no % 包围猎物机制 r1=rand(); r2=rand(); A=2*a*r1-a; C=2*r2; % 气泡网攻击 p=rand(); if p<0.5 % 收缩包围 if abs(A)<1 D_leader=abs(C*Leader_pos(t,:)-whales(i,:)); whales(i,:)=Leader_pos(t,:)-A*D_leader; else % 全局搜索 rand_index=randi([1,SearchAgents_no]); D_rand=abs(C*whales(rand_index,:)-whales(i,:)); whales(i,:)=whales(rand_index,:)-A*D_rand; end else % 螺旋更新 distance2Leader=abs(Leader_pos(t,:)-whales(i,:)); whales(i,:)=distance2Leader.*exp(b.*l).*cos(2*pi*l)+Leader_pos(t,:); end end end

这段代码最骚的操作在a系数的动态调整上。迭代初期a值较大（接近2），算法倾向于全局搜索；随着迭代进行a逐渐减小到0，转向局部开发。这种自适应机制比固定步长聪明多了，就像老司机开车知道什么时候该踩油门，什么时候要刹车。

接下来是BP网络的核心构建部分：

function net = create_BP(hiddenLayerSize) net = feedforwardnet(hiddenLayerSize); net.trainParam.epochs = 1000; % 别设太小，配合早停法 net.trainParam.max_fail = 6; % 验证失败次数阈值 net.divideFcn = 'dividerand'; net.divideParam.trainRatio = 0.7; net.divideParam.valRatio = 0.15; net.layers{1}.transferFcn = 'tansig'; % 隐藏层用双曲正切 net.layers{2}.transferFcn = 'purelin'; % 输出层线性 end

这里有个隐藏技巧：max_fail参数配合早停法可以有效防止过拟合。建议把epochs设大些，让网络有足够收敛时间，同时验证集误差连续6次不降就停止训练，这比固定训练次数科学多了。

重点来了，适应度函数怎么设计：

function fitness = get_fitness(position) input_num = round(position(1)); % 输入层节点数 hidden_num = round(position(2)); % 隐含层节点数 % 动态调整网络结构 net = create_BP(hidden_num); net = configure(net, inputData, targetData); % 训练并获取测试集误差 [net,tr] = train(net,inputData,targetData); testIndex = tr.testInd; testOutputs = net(inputData(:,testIndex)); testErrors = gsubtract(targetData(testIndex),testOutputs); fitness = mse(testErrors); % 以测试集MSE作为适应度 end

这里有个坑要注意：inputnum和hiddennum必须取整！之前用floor函数翻过车，后来发现round才是王道。适应度值用测试集的MSE，比用训练集误差靠谱，有效避免过拟合欺骗。

实际跑起来会发现个有趣现象：WOA初期会疯狂试探各种网络结构，后期逐渐收敛到某个最优配置。比如在某次电力负荷预测中，算法最终锁定了输入层8节点+隐藏层12节点的结构，比人工试错省事多了。

最后给个可视化对比图（伪代码）：

% 预测结果可视化 plot(testData,'b-o'); hold on; plot(WOA_BP_pred,'r-s'); plot(BP_pred,'g-*'); legend('真实值','WOA-BP预测','传统BP预测'); xlabel('时间点');ylabel('数值'); title('预测效果对比');

从实际运行结果看，WOA-BP的预测曲线明显更贴近真实数据波动，传统BP的预测线则像喝醉了一样忽上忽下。特别是在数据突变点，WOA-BP的反应速度比传统BP快至少2个时间步长。

几点实战建议：