【预测模型-ELM预测】基于蚁群算法优化极限学习机预测matlab代码

1 简介

针对变压器故障的特征,结合变压器油中气体分析法以及三比值法.提出了基于蚁群算法改进极限学习机的故障诊断方法.由于输入层与隐含层的权值和阈值是随机产生.传统的极限学习机可能会使隐含层节点过多,训练过程中容易产生过拟合现象.该方法运用蚁群算法对极限学习机的输入层与隐含层的权值与阈值进行优化,从而提高模型的稳定性和预测精度.将诊断结果与传统的基于极限学习机故障诊断进行对比,结果表明,基于蚁群算法改进极限学习机变压器故障诊断的精度更高.

基于 ACO 优化 ELM 的变压器故障诊断的具体步骤为：

1)确定 ELM 的拓扑结构。即输入层神经元个数，隐含层神经元个数以及输出层神经元个数；

2)对 ELM 中输入层到隐含层的权值以及阈值进行编码，得到初始种群；

3)解码得到权值和阈值，将权值和阈值带入到ELM 的训练网络中，使用训练样本进行训练；

4)训练完成后，使用测试样本进行测试，将测试样本的期望值和预测值的误差平方和作为适应度函数；

5)对种群进行选择，交叉，变异，得到新的种群，如果满足条件，则得出了误差平方和最小的网络权值和阈值，如果不满足条件，则返回步骤 2)；

6)将优化后的权值和阈值带入到训练网络中，计算隐含层输出矩阵 H， 并求解矩阵 H 的 MoorePenrose 广义逆 H+；

7)计算输出层权值β赞=H+T；

8)将测试样本带入到模型中进行预测。

2 部分代码

hile iter <= iter_max

     % 随机产生各个蚂蚁的起点城市

      start = zeros(m,1);

      for i = 1:m

          temp = randperm(n);

          start(i) = temp(1);

      end

      Table(:,1) = start; 

      citys_index = 1:n;

      % 逐个蚂蚁路径选择

      for i = 1:m

          % 逐个城市路径选择

         for j = 2:n

             tabu = Table(i,1:(j - 1));           % 已访问的城市集合(禁忌表)

             allow_index = ~ismember(citys_index,tabu);

             allow = citys_index(allow_index);  % 待访问的城市集合

             P = allow;

             % 计算城市间转移概率

             for k = 1:length(allow)

                 P(k) = Tau(tabu(end),allow(k))^alpha * Eta(tabu(end),allow(k))^beta;

             end

             P = P/sum(P);

             % 轮盘赌法选择下一个访问城市

             Pc = cumsum(P);     

            target_index = find(Pc >= rand); 

            target = allow(target_index(1));

            Table(i,j) = target;

         end

      end

      % 计算各个蚂蚁的路径距离

      Length = zeros(m,1);

      for i = 1:m

          Route = Table(i,:);

          for j = 1:(n - 1)

              Length(i) = Length(i) + D(Route(j),Route(j + 1));

          end

          Length(i) = Length(i) + D(Route(n),Route(1));

      end

      % 计算最短路径距离及平均距离

      if iter == 1

          [min_Length,min_index] = min(Length);

          Length_best(iter) = min_Length;  

          Length_ave(iter) = mean(Length);

          Route_best(iter,:) = Table(min_index,:);

      else

          [min_Length,min_index] = min(Length);

          Length_best(iter) = min(Length_best(iter - 1),min_Length);

          Length_ave(iter) = mean(Length);

          if Length_best(iter) == min_Length

              Route_best(iter,:) = Table(min_index,:);

          else

              Route_best(iter,:) = Route_best((iter-1),:);

          end

      end

3 仿真结果

4 参考文献

[1]郑嘉利, & 王哲. (2019). 一种基于蝗虫算法和极限学习机的RFID室内定位方法.  CN109598320A.