基于MATLAB的粒子群算法(PSO)优化libsvm参数完整实现

一、优化原理与流程

1. 优化目标

  • 参数空间:SVM关键参数为惩罚系数C和RBF核参数gamma
  • 目标函数:最大化交叉验证准确率(分类)或最小化均方误差(回归)

2. PSO算法流程

graph TD A[初始化粒子群] --> B{适应度评估} B --> C[更新个体最优] C --> D[更新全局最优] D --> E{终止条件?} E -->|是| F[输出最优参数] E -->|否| B

二、代码实现

1. 数据准备与预处理

% 加载数据(示例使用鸢尾花数据集)
load fisheriris
X = meas';  % 特征矩阵 (4维)
Y = grp2idx(species);  % 类别标签

% 数据归一化
[X, ps_input] = mapminmax(X', 0, 1);
X = X';

% 划分训练集/测试集
cv = cvpartition(Y,'HoldOut',0.3);
X_train = X(cv.training,:);
Y_train = Y(cv.training);
X_test = X(cv.test,:);
Y_test = Y(cv.test);

2. PSO参数设置

% PSO参数
pso_option = struct(...
    'maxgen', 100,    % 最大迭代次数
    'sizepop', 30,    % 粒子数量
    'c1', 1.5,        % 个体学习因子
    'c2', 1.7,        % 群体学习因子
    'w', 0.7,         % 惯性权重
    'vmax', 0.5,      % 速度上限
    'vmin', -0.5,     % 速度下限
    'pc1', 0.1,       % 参数c1变异概率
    'pc2', 0.1);     % 参数c2变异概率

% SVM参数范围
param_range = [0.01, 100;   % C的范围
               0.001, 10];  % gamma的范围

3. 适应度函数定义

function fitness = svmFitness(params, X_train, Y_train)
    % 参数解码
    C = 10^params(1);
    gamma = 10^params(2);
    
    % 交叉验证参数
    cmd = sprintf('-v 5 -c %f -g %f', C, gamma);
    
    % 训练SVM并返回交叉验证准确率
    accuracy = svmtrain(Y_train, X_train, cmd);
    fitness = -accuracy;  % 最小化问题
end

4. PSO优化过程

% 初始化粒子群
num_particles = pso_option.sizepop;
dim = 2;  % C和gamma两个参数
particles = rand(num_particles, dim) .* (param_range(:,2)' - param_range(:,1)') + param_range(:,1)';
velocities = rand(num_particles, dim) .* (pso_option.vmax - pso_option.vmin) + pso_option.vmin;

% 初始化个体最优
pbest = particles;
pbest_fitness = inf(num_particles,1);

% 初始化全局最优
[gbest_fitness, gbest_idx] = min(pbest_fitness);
gbest = pbest(gbest_idx,:);

% 迭代优化
for iter = 1:pso_option.maxgen
    for i = 1:num_particles
        % 适应度评估
        current_fitness = svmFitness(particles(i,:), X_train, Y_train);
        
        % 更新个体最优
        if current_fitness < pbest_fitness(i)
            pbest(i,:) = particles(i,:);
            pbest_fitness(i) = current_fitness;
        end
        
        % 更新全局最优
        if current_fitness < gbest_fitness
            gbest = particles(i,:);
            gbest_fitness = current_fitness;
        end
    end
    
    % 速度更新
    r1 = rand(num_particles,dim);
    r2 = rand(num_particles,dim);
    velocities = pso_option.w*velocities ...
        + pso_option.c1*r1.*(pbest - particles) ...
        + pso_option.c2*r2.*(gbest - particles);
    
    % 速度限制
    velocities = max(min(velocities, pso_option.vmax), pso_option.vmin);
    
    % 位置更新
    particles = particles + velocities;
    
    % 参数范围限制
    particles(:,1) = max(min(particles(:,1), param_range(1,2)), param_range(1,1));
    particles(:,2) = max(min(particles(:,2), param_range(2,2)), param_range(2,1));
    
    % 自适应变异
    if rand < pso_option.pc1
        particles(:,1) = 10^randn(num_particles,1);
    end
    if rand < pso_option.pc2
        particles(:,2) = 10^randn(num_particles,1);
    end
    
    % 显示迭代信息
    fprintf('Iter %d: Best Fitness=%.4f (C=%.4f, gamma=%.4f)\n',...
        iter, -gbest_fitness, 10^gbest(1), 10^gbest(2));
end

5. 模型训练与评估

% 使用最优参数训练SVM
best_C = 10^gbest(1);
best_gamma = 10^gbest(2);
cmd = sprintf('-c %f -g %f', best_C, best_gamma);
model = svmtrain(Y_train, X_train, cmd);

% 测试集预测
[predict_label, accuracy, dec_values] = svmpredict(Y_test, X_test, model);

% 结果输出
fprintf('测试集准确率: %.2f%%
', accuracy(1));

三、改进

1. 自适应参数调整

  • 惯性权重动态调整

    w = pso_option.w * (1 - iter/pso_option.maxgen);  % 线性递减
    
  • 学习因子自适应

    c1 = pso_option.c1 * (1 - iter/pso_option.maxgen);
    c2 = pso_option.c2 * (1 + iter/pso_option.maxgen);
    

2. 多目标优化扩展

% 同时优化准确率和模型复杂度
fitness = -accuracy + 0.1*(sum(W.^2));  % W为支持向量权重

3. 并行计算加速

parfor i = 1:num_particles
    % 并行计算适应度
end

参考代码 小波神经网络预测代码 www.youwenfan.com/contentcni/64776.html

四、可视化分析

1. 参数收敛曲线

figure;
plot(1:pso_option.maxgen, -gbest_fitness*ones(1,pso_option.maxgen), 'r--');
hold on;
plot(1:pso_option.maxgen, mean(pbest_fitness,1), 'b-.');
legend('全局最优', '平均适应度');
xlabel('迭代次数'); ylabel('适应度值');
title('PSO收敛曲线');

2. 参数分布热力图

figure;
histogram2(pbest(:,1), pbest(:,2), 'DisplayStyle','tile');
xlabel('log10(C)'); ylabel('log10(gamma)');
title('最优参数分布');
colorbar;
posted @ 2025-10-11 10:37  yijg9998  阅读(84)  评论(0)    收藏  举报