基于鲸鱼优化算法优化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个时间步长。
几点实战建议:
- 种群数量别超过50,否则MATLAB会卡到怀疑人生
- 数据归一化用mapminmax比手动写归一化更稳
- 时间窗口滑动建议用buffer函数,比for循环优雅
- 并行计算可以加速WOA,用parfor替换for能省30%时间
这种混合算法最适合中短期预测,比如股票日内交易、电力小时级负荷预测。长期预测还是得结合其他方法,不过作为baseline模型已经足够惊艳。下次试试把Attention机制加进去,应该还能再提升一波。