深圳幻海软件技术有限公司 欢迎您!

灰狼优化算法(GWO)(解决TSP问题,代码完整免费)

2023-04-02

算法背景灰狼优化算法(GWO),由澳大利亚格里菲斯大学学者Mirjalili等人于2014年提出来的一种群智能优化算法。灵感来自于灰狼群体捕食行为。优点:较强的收敛性能,结构简单、需要调节的参数少,容易实现,存在能够自适应调整的收敛因子以及信息反馈机制,能够在局部寻优与全局搜索之间实现平衡,因此在对

算法背景

灰狼优化算法(GWO),由澳大利亚格里菲斯大学学者 Mirjalili 等人于2014年提出来的一种群智能优化算法。灵感来自于灰狼群体捕食行为。

优点:较强的收敛性能,结构简单、需要调节的参数少,容易实现,存在能够自适应调整的收敛因子以及信息反馈机制,能够在局部寻优与全局搜索之间实现平衡,因此在对问题的求解精度和收敛速度方面都有良好的性能。

缺点:存在着易早熟收敛,面对复杂问题时收敛精度不高,收敛速度不够快

应用:车间调度、参数优化、图像分类、路径规划。


算法思想

灰狼群体中有严格的等级制度,一小部分拥有绝对话语权的灰狼带领一群灰狼向猎物前进。灰狼群一般分为4个等级:  (权利从大到小)模拟领导阶层。

集体狩猎是灰狼的一种社会行为,社会等级在集体狩猎过程中发挥着重要的作用,捕食的过程在α的带领下完成。主要包括三个步骤:

  • 跟踪和接近猎物
  • 骚扰、追捕和包围猎物,直到它停止移动
  • 攻击猎物

社会等级分层

构建灰狼社会等级层次模型,对灰狼的社会等级进行数学建模。

考虑一个搜索空间,现实中狼群由眼睛嗅觉等等知道猎物的位置,然而计算机中模拟无法做到这点,但是我们可以假设,猎物位置是由我们在搜索空间中发现的最佳解决方案提供,我们可以使用该解决方案去找到更好的解决方案,不断迭代优化,此最优解决方案会非常接近最好的解决方案。

将  作为最优解(个体的适应度最优),次优解   ,最佳解决方案  ,剩下的候选解命名为 , 狩猎过程由  引导, 跟随这三只狼。即我们总是去找到三个最佳解决方案,然后围绕该区域进行搜索,目的是找到更好的解决方案然后更新 。


包围猎物

在狩猎过程中,将灰狼围捕猎物的行为定义如下:

个体与猎间的距离公式:

        (1)

灰狼位置更新公式:

        (2)

系数向量:

        (3)

        (4)

其中:t 为迭代次数,D 是个体与猎间的距离向量, ‘’ 不是点积,是乘法, 是猎物位置向量,X 为灰狼位置向量,  是收敛因子(随迭代次数从 2 线性减小到 0 ),r1 和 r2 是随机向量,模取【0-1】之间的随机数。

从公式可以看出,经过移动后灰狼群向  移动,移动方向由自身位置和随机向量 C 决定,移动步长由隔离距离灰狼距离和系数向量 A 决定,即  线性减小意味着随机性和运动步长幅度,步长随迭代次数而减小越来越接近最优解。

狩猎

灰狼能够识别猎物的位置并包围它们。当灰狼识别出猎物的位置后,  在  的带领下指导狼群包围猎物。灰狼个体跟踪猎物位置的数学模型描述如下:

                        

                (5)

                        

其中,  , 和  分别表示   和  , 与其他个体间的距离; 分别代表   和  ,当前的位置; 是随机向量,  X 是当前灰狼的位置。

                        

                (6)

                        

                 (7)

式(6)分别定义了狼群中  个体朝向    和  ,  前进的步长和方向 ,式 (7) 定义了ω的最终位置。

攻击猎物

当猎物停止移动时,灰狼通过攻击来完成狩猎过程.为了模拟逼近猎物,  的值被逐渐减小,因此  A的波动范围也随之减小。换句话说,在迭代过程中,当   的值从2线性下降到0时,其对应的 A 的值也在区间 [ - ] 内变化。

如图所示,当  A 的值位于区间内时,灰狼的下一位置可以位于其当前位置和猎物位置之间的任意位置(由于随机),当 -1<A<1 时,狼群向猎物发起攻击(陷入局部最优)。当 A>1 或 A<-1 时,灰狼远离猎物,探索其他区域(为了找到全局最优解)。

GWO 另一个有利于全局探索的组件 C(随机权重) ,C 是[0,2]之间的随机向量。表示狼所在的位置对猎物影响的随机权重,C>1表示影响权重大,反之,表示影响权重小。在算法陷入了局部最优并且不易跳出时, C 的随机性在避免局部最优方面发挥了非常重要的作用。


算法流程

伪代码

  1. Initialize the grey wolf population Xi (i=1,2,,,,n)
  2. Initialize a,A,C,t=0
  3. Calculate the fitness of each search agent
  4. X(alpha) = the best search agent
  5. X(bete) = the second best search agent
  6. X(delta) = the third best search agent
  7. while (t< Max number of iterations)
  8. for each search agent
  9. Update the position of the current search agent
  10. end for
  11. Update a,A,and C
  12. Calculate the fitness of all search agents
  13. Update X(alpha),X(bete),and X(delta)
  14. t=t+1
  15. end while
  16. return X(alpha)

算法测试

求函数最大值(一元)

  1. % 灰狼优化算法(求函数极值)
  2. clc;
  3. clear;
  4. close all;
  5. %% 目标函数
  6. f= @(x) - (x - 10) .^ 2 + x .* sin(x) .* cos(2 * x) - 5 * x .* sin(3 * x) ; % 适应度函数表达式(求这个函数的最大值)
  7. figure(1);
  8. fplot(f, [0 20], 'b-'); % 画出初始图像
  9. title('初始图像');
  10. hold on;
  11. %% 初始化参数
  12. N=30; % 灰狼个数
  13. dim=1; % 维度
  14. Iter=50; % 最大迭代次数
  15. a=2; % 收敛因子
  16. ub=20; % 最大值限制
  17. lb=0; % 最小值限制
  18. % 初始化alpha,beta,delta
  19. Alpha_pos=zeros(1,dim);
  20. Alpha_score=-inf; %求最大值改为inf
  21. Beta_pos=zeros(1,dim);
  22. Beta_score=-inf;
  23. Delta_pos=zeros(1,dim);
  24. Delta_score=-inf;
  25. Positions=rand(N,dim).*(ub-lb)+lb; % 初始化个体位置
  26. Convergence_curve=zeros(1,Iter); % 收敛曲线
  27. l=0; %循环次数记录
  28. %% 迭代求解
  29. while l<Iter
  30. for i=1:size(Positions,1)
  31. % 超出边界处理
  32. Flag4ub=Positions(i,:)>ub;
  33. Flag4lb=Positions(i,:)<lb;
  34. Positions(i,:)=(Positions(i,:).*(~(Flag4ub+Flag4lb)))+ub.*Flag4ub+lb.*Flag4lb;
  35. % 计算个体适应度函数
  36. fitness=f(Positions(i,:));
  37. % 更新 Alpha, Beta, and Delta
  38. if fitness>Alpha_score
  39. Alpha_score=fitness;
  40. Alpha_pos=Positions(i,:);
  41. end
  42. if fitness<Alpha_score && fitness>Beta_score
  43. Beta_score=fitness;
  44. Beta_pos=Positions(i,:);
  45. end
  46. if fitness<Alpha_score && fitness<Beta_score && fitness>Delta_score
  47. Delta_score=fitness;
  48. Delta_pos=Positions(i,:);
  49. end
  50. end
  51. a=2-l*((2)/Iter); % 收敛因子从2线性递减到0
  52. % 更新灰狼个体的位置
  53. for i=1:size(Positions,1)
  54. for j=1:size(Positions,2)
  55. r1=rand(); % r1 is a random number in [0,1]
  56. r2=rand(); % r2 is a random number in [0,1]
  57. A1=2*a*r1-a;
  58. C1=2*r2;
  59. D_alpha=abs(C1*Alpha_pos(j)-Positions(i,j));
  60. X1=Alpha_pos(j)-A1*D_alpha;
  61. r1=rand();
  62. r2=rand();
  63. A2=2*a*r1-a;
  64. C2=2*r2;
  65. D_beta=abs(C2*Beta_pos(j)-Positions(i,j));
  66. X2=Beta_pos(j)-A2*D_beta;
  67. r1=rand();
  68. r2=rand();
  69. A3=2*a*r1-a;
  70. C3=2*r2;
  71. D_delta=abs(C3*Delta_pos(j)-Positions(i,j));
  72. X3=Delta_pos(j)-A3*D_delta;
  73. Positions(i,j)=(X1+X2+X3)/3;% Equation (3.7)
  74. end
  75. end
  76. l=l+1;
  77. Convergence_curve(l)=Alpha_score;
  78. end
  79. plot(Alpha_pos, f(Alpha_pos), '*r');
  80. figure(2);
  81. plot(Convergence_curve);
  82. title('收敛过程');
  83. display(['The best solution obtained by GWO is : ', num2str(Alpha_pos)]);
  84. display(['The best optimal value of the objective funciton found by GWO is : ', num2str(Alpha_score)]);

二元函数求最大值

  1. % 灰狼优化算法(求函数极值)
  2. clc;
  3. clear;
  4. close all;
  5. %% 目标函数
  6. [ZuoBiao_x, ZuoBiao_y] = meshgrid(-10:0.1:10,-5:0.1:5); % 产生二维坐标
  7. ZuoBiao_z = f_xy(ZuoBiao_x, ZuoBiao_y);
  8. figure(1);
  9. s = mesh(ZuoBiao_x, ZuoBiao_y, ZuoBiao_z); % 画出初始图像
  10. title('初始图像');
  11. %% 初始化参数
  12. N=30; % 灰狼个数
  13. dim=2; % 维度
  14. Iter=200; % 最大迭代次数
  15. a=2; % 收敛因子
  16. ub=[10 5]; % 最大值限制
  17. lb=[-10 -5]; % 最小值限制
  18. % 初始化alpha,beta,delta
  19. Alpha_pos=zeros(1,dim);
  20. Alpha_score=-inf; %求最大值改为inf
  21. Beta_pos=zeros(1,dim);
  22. Beta_score=-inf;
  23. Delta_pos=zeros(1,dim);
  24. Delta_score=-inf;
  25. Positions=rand(N,dim).*(ub-lb)+lb; % 初始化个体位置
  26. Convergence_curve=zeros(1,Iter); % 收敛曲线
  27. l=0; %循环次数记录
  28. %% 迭代求解
  29. while l<Iter
  30. for i=1:size(Positions,1)
  31. % 超出边界处理
  32. Flag4ub=Positions(i,:)>ub;
  33. Flag4lb=Positions(i,:)<lb;
  34. Positions(i,:)=(Positions(i,:).*(~(Flag4ub+Flag4lb)))+ub.*Flag4ub+lb.*Flag4lb;
  35. % 计算个体适应度函数
  36. fitness=f_xy(Positions(i,1), Positions(i,2));
  37. % 更新 Alpha, Beta, and Delta
  38. if fitness>Alpha_score
  39. Alpha_score=fitness;
  40. Alpha_pos=Positions(i,:);
  41. end
  42. if fitness<Alpha_score && fitness>Beta_score
  43. Beta_score=fitness;
  44. Beta_pos=Positions(i,:);
  45. end
  46. if fitness<Alpha_score && fitness<Beta_score && fitness>Delta_score
  47. Delta_score=fitness;
  48. Delta_pos=Positions(i,:);
  49. end
  50. end
  51. a=2-l*((2)/Iter); % 收敛因子从2线性递减到0
  52. % 更新灰狼个体的位置
  53. for i=1:size(Positions,1)
  54. for j=1:size(Positions,2)
  55. r1=rand(); % r1 is a random number in [0,1]
  56. r2=rand(); % r2 is a random number in [0,1]
  57. A1=2*a*r1-a;
  58. C1=2*r2;
  59. D_alpha=abs(C1*Alpha_pos(j)-Positions(i,j));
  60. X1=Alpha_pos(j)-A1*D_alpha;
  61. r1=rand();
  62. r2=rand();
  63. A2=2*a*r1-a;
  64. C2=2*r2;
  65. D_beta=abs(C2*Beta_pos(j)-Positions(i,j));
  66. X2=Beta_pos(j)-A2*D_beta;
  67. r1=rand();
  68. r2=rand();
  69. A3=2*a*r1-a;
  70. C3=2*r2;
  71. D_delta=abs(C3*Delta_pos(j)-Positions(i,j));
  72. X3=Delta_pos(j)-A3*D_delta;
  73. Positions(i,j)=(X1+X2+X3)/3;% Equation (3.7)
  74. end
  75. end
  76. l=l+1;
  77. Convergence_curve(l)=Alpha_score;
  78. end
  79. figure(2);
  80. plot(Convergence_curve);
  81. title('收敛过程');
  82. display(['The best solution obtained by GWO is : ', num2str(Alpha_pos)]);
  83. display(['The best optimal value of the objective funciton found by GWO is : ', num2str(Alpha_score)]);

多元函书求最小值(代码,24个测试函数,感兴趣下载)

链接:https://pan.baidu.com/s/1sQxYMXhwF1xhRqhXB1kJpQ 
提取码:1234 

GWO解TSP问题(网上找了好久,都不全,自己补充一下,简单注释一下,有问题私聊哦)

  1. % 灰狼优化算法GWO (求TSP问题)
  2. clc;
  3. clear;
  4. %% 初始化参数
  5. N=100; % 灰狼个数
  6. Max_iter=1000; % 最大迭代次数
  7. load('city.mat'); % 导入城市数据
  8. City=city; % 保存城市位置用于计算距离
  9. M=size(city,1); % 得到TSP问题的规模,即城市数目
  10. dim=M; % 维度
  11. figure('name','灰狼优化算法');
  12. plot(city(:,1),city(:,2),'o'); %描点 city--(302
  13. for i=1:M
  14. text(city(i,1)+0.5,city(i,2),num2str(i)); %标号
  15. end
  16. title('城市初始位置');
  17. % 初始化alpha,beta,delta
  18. Alpha_pos=zeros(1,dim);
  19. Alpha_score=inf;
  20. Beta_pos=zeros(1,dim);
  21. Beta_score=inf;
  22. Delta_pos=zeros(1,dim);
  23. Delta_score=inf;
  24. % 初始化种群位置
  25. Positions=zeros(N,dim);
  26. for i=1:N
  27. Positions(i,:)=randperm(dim); %随机排列,比如[2 4 5 6 1 3]
  28. end
  29. Length_best = zeros(1, Max_iter); % 定义每次迭代的最短距离
  30. Length_ave = zeros(1, Max_iter); % 定义每次迭代的平均距离
  31. l = 1; % 迭代计数器
  32. %% 迭代寻优
  33. while l < Max_iter+1
  34. for i = 1:N
  35. % 按行升序排列产生城市序列
  36. [~, sol] = sort(Positions, 2);
  37. % 计算目标函数值(即路径距离)
  38. fitness = Fun(sol(i, :),City,M);
  39. Length_ave(l) = Length_ave(l)+fitness;
  40. % 更新Alpha, Beta, and Delta
  41. if fitness < Alpha_score
  42. Alpha_score = fitness;
  43. Alpha_pos = Positions(i, :);
  44. end
  45. if fitness > Alpha_score && fitness < Beta_score
  46. Beta_score = fitness;
  47. Beta_pos = Positions(i, :);
  48. end
  49. if fitness > Alpha_score && fitness > Beta_score && fitness < Delta_score
  50. Delta_score = fitness;
  51. Delta_pos = Positions(i, :);
  52. end
  53. end
  54. a = 2-l*((2)/Max_iter); % a从2线性减到0
  55. % 更新所有个体位置
  56. for i = 1:N
  57. for j = 1:dim
  58. r1 = rand(); % r1 is a random number in [0,1]
  59. r2 = rand(); % r2 is a random number in [0,1]
  60. A1 = 2*a*r1-a;
  61. C1 = 2*r2;
  62. D_alpha = abs(C1*Alpha_pos(j)-Positions(i, j));
  63. X1 = Alpha_pos(j)-A1*D_alpha;
  64. r1 = rand();
  65. r2 = rand();
  66. A2 = 2*a*r1-a;
  67. C2 = 2*r2;
  68. D_beta = abs(C2*Beta_pos(j)-Positions(i, j));
  69. X2 = Beta_pos(j)-A2*D_beta;
  70. r1 = rand();
  71. r2 = rand();
  72. A3 = 2*a*r1-a;
  73. C3 = 2*r2;
  74. D_delta = abs(C3*Delta_pos(j)-Positions(i, j));
  75. X3 = Delta_pos(j)-A3*D_delta;
  76. Positions(i, j) = (X1+X2+X3)/3;
  77. end
  78. end
  79. Length_best(l) = Alpha_score; % 最短距离
  80. Length_ave(l) = Length_ave(l)/dim; % 平均距离
  81. disp(['Iteration ' num2str(l) ': Best Fitness = ' num2str(Alpha_score)]); % 命令行窗口展示
  82. l = l + 1; % 更新迭代次数
  83. [~, BestSol] = sort(Alpha_pos); % 得到最优解
  84. end
  85. figure(2);
  86. for i=1:M-1
  87. plot([City(BestSol(i),1),City(BestSol(i+1),1)],[City(BestSol(i),2),City(BestSol(i+1),2)],'bo-');
  88. hold on;
  89. end
  90. plot([City(BestSol(M),1),City(BestSol(1),1)],[City(BestSol(M),2),City(BestSol(1),2)],'ro-');
  91. for i=1:M
  92. text(City(i,1)+0.5,City(i,2),num2str(i)); %标号
  93. end
  94. text(City(BestSol(1),1),City(BestSol(1),2),' 起点');
  95. text(City(BestSol(M),1),City(BestSol(M),2),' 终点');
  96. title('最终路线图');
  97. figure(3);
  98. t = 1:Max_iter;
  99. plot(t, Length_best, 'r', t, Length_ave, 'b--', 'LineWidth', 2);
  100. xlabel('Iteration');
  101. ylabel('Best Cost');
  102. legend('最短距离','平均距离')
  103. xlabel('迭代次数')
  104. ylabel('距离')
  105. title('各代最短距离与平均距离对比')
  106. P=zeros(1, 30);
  107. function len=Fun(sol,City,M) % 距离计算函数
  108. len=0;
  109. for k=1:M-1
  110. len=len+sqrt(sum((City(sol(1,k),:)-City(sol(1,k+1),:)).^2));
  111. end
  112. len=len+sqrt(sum((City(sol(1,M),:)-City(sol(1,1),:)).^2));
  113. end

测试数据依旧用我们以前用的:

链接:https://pan.baidu.com/s/1j7omLCzx1uIJS0VdBbR-eA 
提取码:1234 

测试效果:


[1] S. Mirjalili, S. M. Mirjalili, and A. Lewis, "Grey Wolf Optimizer," Advances in Engineering Software, vol. 69, pp. 46-61, 2014.

文章知识点与官方知识档案匹配,可进一步学习相关知识
算法技能树首页概览42750 人正在系统学习中