MATLAB粒子群优化算法实现（PSO）

mutougeda

- e+ a: E0 C) {) j. uPSO（PSO——Particle SwARM Optimization）（基于种群的随机优化技术算法）
粒子群算法模仿昆虫、兽群、鸟群和鱼群等的群集行为，这些群体按照一种合作的方式寻找食物，群体中的每个成员通过学习它自身的经验和其他成员的经验来不断改变其搜索模式。

MATLAB代码：

! B4 W2 k5 m; L1 e* v! e  b/ A%------初始格式化--------------------------------------------------
5 o3 M2 P% E# @: L+ S4 p/ N3 fclear all;
7 `: B& Y0 P+ d6 ]clc;
. R; O3 [  K' E9 y: W) o2 g- V& pformat long;
0 Z, w+ T( k7 J; o; G( i%------给定初始化条件----------------------------------------------
c1=2;             %学习因子1
c2=2;             %学习因子2
  H% \1 F+ S; J3 Lw=0.7298;              %惯性权重
MaxDT=200;            %最大迭代次数
% D=2;                  %搜索空间维数（未知数个数）
# C. k7 ^, C- l2 K6 k- E5 uN=20;                  %初始化群体个体数目
%eps=10^(-6);           %设置精度(在已知最小值时候用)
Vmax=1;
1 O1 P7 T  r. o" h* j* w3 ~Vmin=-1;
' b) n- a7 b9 \- Q4 ]+ S. m8 fpopmax=5;
popmin=-5;
%------初始化种群的个体(可以在这里限定位置和速度的范围)------------
; O5 i. y8 H8 P' ifor i=1:N
  S' ^9 j1 S7 r9 h% z' c0 u9 W; k        pop(i,: )=popmin+(popmax-popmin)*rand(1,2);  %随机初始化位置
5 ?& ~" q8 b6 C3 J6 g! N9 O
( @$ E9 h9 M6 ]- T3 E# h8 J        V(i,: )=rand(1,2); %随机初始化速度
% c, f. h' K3 R        fitness(i)=ackley(pop(i,: ));

    end
%------先计算各个粒子的适应度，并初始化Pi和Pg----------------------
[fitnessgbest bestindex]=min(fitness);
gbest=pop(bestindex,: );
pbest=pop;
fitnesspbest=fitness;

1 A2 W) M  s! t$ k5 j3 d: T6 E( m# Pfor i=1:MaxDT
" @7 T! ]5 P! K4 J    for j=1:N
        V(j,: )=w*V(j,: )+c1*rand*(pbest(j,: )-pop(j,: ))+c2*rand*(gbest-pop(j,: ));
        V(j,find(V(j,: )>Vmax))=Vmax;
        V(j,find(V(j,: )<Vmin))=Vmin;
        pop(j,: )=pop(j,: )+V(j,: );
        pop(j,find(pop(j,: )>popmax))=popmax;
        pop(j,find(pop(j,: )<popmin))=popmin;
5 p& b4 B; p( e
3 x4 K5 j. C5 `1 }%         if rand>0.8
3 V& o: J# \; @0 h%             k=ceil(2*rand);
8 N& f: }" ]6 y, S- n$ b$ \! Y' n%             pop(j,k)=rand;
, K9 E: I" W$ ]/ s8 [  F' E& S, N/ p%         end        
: _6 H! e8 b8 s& @' m        fitness(j)=ackley(pop(j,: ));
  m& _( A6 S4 {: {8 @6 ~3 e
; n' g' k3 K5 @
) b  S. ~- u1 b1 W       if fitness(j)<fitnesspbest(j)
            pbest(j,: )=pop(j,: );
            fitnesspbest(j)=fitness(j);
       end
1 u' ^, N1 s  X
$ H# C2 l9 h. U2 [2 O" y       if fitness(j)<fitnessgbest
           gbest=pop(j,: );
           fitnessgbest=fitness(j);  
       end
$ Y# Z) h7 s7 M
    end
   yy(i)=fitnessgbest;
2 \0 R' V/ s) P+ Y
& p8 ~* @7 [. Y  ]& wend
%------最后给出计算结果
: _8 q" y; G; U; }plot(yy)
2 f8 \6 ^! j. u, u) gtitle(['适应度曲线 ' '终止次数=' num2str(MaxDT)]);
" O' j$ E& v/ Q. Wxlabel('进化代数');
ylabel('适应度')
  e/ Z3 x" a: @6 V9 v%------算法结束---DreamSun GL & HF-----------------------------------
) t" J7 `( Q( v! ~! H& u

* o& x* R) ]) }& \0 R优化的函数为ackley函数：

% ackley.m
+ u: u; Z4 n) W0 L1 j8 d1 ~: A% Ackley's function
8 ~/ Z6 [4 f! Y  ~2 k9 p" C  ]& \7 j% and further shown at:
%
% commonly used to test optimization/global minimization problems
& B9 y1 X0 Z) N  X! A: n# L%
% f(x)= [ 20 + e ...
. g$ O! ]1 `7 R%        -20*exp(-0.2*sqrt((1/n)*sum(x.^2,2))) ...
%        -exp((1/n)*sum(cos(2*pi*x),2))];
%
8 o  h. f) j0 s" p2 H! }2 }% dimension n = # of columns of input, x1, x2, ..., xn
. ]# q8 b$ f  q2 E- V3 K4 k* B8 d% each row is processed independently,
% you can feed in matrices of timeXdim no prob
%
% example: cost = ackley([1,2,3;4,5,6])
5 i3 |- m7 {7 \; `- y/ ^0 A
function [out]=ackley(in)

% dimension is # of columns of input, x1, x2, ..., xn
5 \, ]4 X2 z  t1 s$ m( O% f n=length(in(1,: ));
6 b2 D+ r7 i5 J5 ~, t) o  W
x=in;
- E* p% L2 A* v" x2 t5 H e=exp(1);
2 Y. R$ C' q; B. s+ m- C! v1 u
4 \, G% r8 J! o out = (20 + e ...
       -20*exp(-0.2*sqrt((1/n).*sum(x.^2,2))) ...
! q5 E! H+ a+ G       -exp((1/n).*sum(cos(2*pi*x),2)));
9 S1 h" ]0 ^4 h return
4 X1 j9 M' E. i* b9 H6 C) \

6 h7 l/ a* v/ e: u

BillScot2