粒子群算法

粒子群算法主函数particle_swarm_optimization.m

clear;
close all;
clc; 
%%%%%%%%%%%%%%%%%%%%%JA模型参数（试验测得的数据Bw）%%%%%%%%%%
Ms=6.2e5;
a=1.5;
alpha=2e-6;
k=2;
c=0.5;
Bw=JA_opt(Ms,a,alpha,k,c);
%%%%%%%%%%%%%%%%%%%%%初始化粒子群参数%%%%%
N = 100;    %粒子群规模
D = 5;      %粒子维度（也就是位移的坐标，即需要优化的参数个数）
T = 1000;    %迭代次数
c1 = 1.5;   %个体学习因子
c2 = 1.5;   %群体学习因子
w = 0.8;    %惯性权重
Xmax = 20;  %位移限制
Xmin = -20; 
Vmax = 10; %速度限制
Vmin = -10; 
%%%%%%%%%%%%%%%%%%%%%%%%随机初始化每个例子的位置和速度%%%%%%%%%%%
Ms=6.2e5+1e5*rand(N,1);
a=1.5+1*rand(N,1);
alpha=2e-6+1e-6*rand(N,1);
k=2+1*rand(N,1);
c=0.5+1*rand(N,1);
x = [Ms a alpha k c];
v = rand(N,D) * (Vmax-Vmin)+Vmin;
%%%%%%%%%%%初始化每个粒子的速度和位置，初始化每个粒子的适应值%%%%%%%%%%%%%%%
p = x;
pbest = [];
pbest_Bt=[];
for i = 1:1:N
    pbest_Bt(:,i) = JA_opt(x(i,1),x(i,2),x(i,3),x(i,4),x(i,5));
    pbest(i) =fitness_function(Bw,pbest_Bt(:,i));
end
%%%%%%%%%%%%%%初始化群体的历史最优适应值和位置%%%%%%%%%%%

g = ones(1,D);
gbest = inf;
%gbest_bt = inf;
for i = 1:1:N
    if(pbest(i) < gbest)
        g = p(i,:);
        gbest = pbest(i);
    end
end

gb = ones(1,T);
%%%%%%%%%%%%%%%%%更新迭代%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
for i = 1:T
    for j = 1:N
        %%%%%%%%%%更新每个粒子的速度和位置，计算每个粒子的适应值%%%%%%%%%%%%
        if (fitness_function(Bw,pbest_Bt(:,j)) < pbest(j))
            p(j,:) = x(j,:);
            pbest(j) = fitness_function(Bw,pbest_Bt(:,j));
        end
        %%%%%%%%%%%更新群体的历史最优适应值和位置%%%%%%%%%%%
        if(pbest(j) < gbest)
            g = p(j,:);
            gbest = pbest(j);
        end
        %%%%%%%%%%%%%速度更新公式 位置更新公式%%%%%%%%%%%%%
        v(j,:) = w*v(j,:)+c1*rand*(p(j,:)-x(j,:))...
            +c2*rand*(g-x(j,:));
        x(j,:) = x(j,:)+v(j,:);
        %%%%%%%%%%%%%%位置限制 和 速度限制%%%%%%%%%%%%%%%
        for ii = 1:D
            if (v(j,ii) > Vmax) || (v(j,ii) < Vmin)
                v(j,ii) = rand * (Vmax-Vmin)+Vmin;
            end

            if (x(j,ii) > Xmax) || (x(j,ii) < Xmin)
                x(j,ii) = rand * (Xmax-Xmin)+Xmin;
            end
        end
    end
    %%%%%%%%%%%%%%迭代结束%%%%%%%%%%%%%
    gb(i) = gbest;
end
g;       %群体最优位置（最优参数）
gb(end); %群体最优适应度大小

%%%%%%%%%%%%%%%%%%%%%%由试验测量数据 解JA模型得到的%%%%%%%%%%%%%%%%
Ms0=6.2e5;
a0=1.5;
alpha0=2e-6;
k0=2;
c0=0.5;
[Hw,Bt]=JA_opt1(Ms0,a0,alpha0,k0,c0) ;
figure(1)
plot(Hw,Bt,'g-','LineWidth',1)
xlabel('磁感应强度（T）')
ylabel('磁场强度(A/m)')

hold on
%%%%%%%%%%%%%%%%%%%%由PSO算法优化的参数 解JA模型得到%%%%%%%%%%%%%%%%%%
[Hw1,Bt1]=JA_opt1(g(1),g(2),g(3),g(4),g(5)) ;
plot(Hw1,Bt1,'r-')

legend('试验数据解','PSO优化参数解')

适应度函数fitness_function.m

function e = fitness_function(Bw,Bt)
e=(sum(((Bw(2:length(Bw))-Bt(2:length(Bt))).^2)./(Bw(2:length(Bw))))/503)^0.5;
end