Matlab加速循环计算

今天在整理一些MATLAB的parfor加速循环计算的示例。也算是突发奇想,这里整理一些相关的示例,以供参考。

parfor简介

在 MATLAB 中,parfor(parallel for-loop)是一种用于并行执行循环迭代的工具,特别适用于可以独立并行处理的计算任务。使用 parfor 可以显著加速计算密集型程序,尤其是在多核 CPU 或集群环境中。

matlab 复制代码
parfor loopVar = initVal:endVal
    statements
end

与普通 for 循环类似,但要求:

(1)各次迭代之间不能有依赖关系 (即不能互相读写共享变量)。

(2)循环变量必须是整数(不能是向量或结构体等)。

性能提示

(1)避免小任务开销:parfor 有调度开销,适合计算量大的循环。

(2)预分配数组:避免在循环中动态增长数组。

(3)减少数据传输:只传递必要数据到 worker(大矩阵会复制多次)。

(4)使用 spmd 或 parfeval 替代复杂场景:如果 parfor 限制太多。

初步实践

确保 Parallel Computing Toolbox 已安装,并启动并行池:

matlab 复制代码
% 查看当前并行池状态
pool = gcp('nocreate'); % 获取当前池(不自动创建)
if isempty(pool)
    parpool; % 创建默认并行池(通常为本地配置,核心数由设置决定)
end

parfor 要求所有变量属于以下五类之一:

⚠️ 禁止:不同迭代之间通过变量通信(如 A(i) = A(i-1) + 1)。

示例一:蒙特卡洛计算(失败示例)

matlab 复制代码
clear,clc
N = 200;  % 迭代次数减少,但每次很重

tic;
result_for = zeros(N,1);
for i = 1:N
    % 每次做 100,000 次随机采样
    x = rand(1e5, 1);
    result_for(i) = mean(exp(-x.^2));  % 估算 ∫₀¹ e^{-x²} dx
end
time_for = toc;

% parfor
tic;
result_parfor = zeros(N,1);
parfor i = 1:N
    x = rand(1e5, 1);
    result_parfor(i) = mean(exp(-x.^2));
end
time_parfor = toc;

fprintf('for: %.3f s | parfor: %.3f s | 加速比: %.2fx\n', ...
        time_for, time_parfor, time_for/time_parfor);

输出结果

for: 0.187 s | parfor: 0.224 s | 加速比: 0.84x

普通for循环的速度就比parfor快,parfor没起到真正的加速作用。

示例二:真正适合 parfor 的蒙特卡洛示例

matlab 复制代码
%% 蒙特卡洛估算 ∫₀¹ e^{-x²} dx ≈ 0.746824
clear; clc;

% === 配置参数 ===
N = 20;                     % 迭代次数(每个迭代是一次独立蒙特卡洛实验)
samples_per_iter = 5e6;     % 每次采样数量(越大,单次估计越准)
REPRODUCIBLE = false;       % ← 设置为 true 可强制 for/parfor 结果完全一致(仅用于调试)

% 理论参考值
theoretical = 0.746824132812427;

% 启动并行池(如果需要)
if ~REPRODUCIBLE && isempty(gcp('nocreate'))
    parpool('local');  % 正常模式:让 MATLAB 自动管理随机流
end

%% --- 普通 for 循环 ---
fprintf('Running for loop...\n');
tic;
result_for = zeros(N,1);
if REPRODUCIBLE
    % 调试模式:每次用固定种子
    for i = 1:N
        rng(i);  % 关键:固定种子
        x = rand(samples_per_iter, 1);
        result_for(i) = mean(exp(-x.^2));
    end
else
    % 正常模式:使用全局流(顺序生成)
    for i = 1:N
        x = rand(samples_per_iter, 1);
        result_for(i) = mean(exp(-x.^2));
    end
end
time_for = toc;

%% --- parfor 循环 ---
fprintf('Running parfor loop...\n');
tic;
result_parfor = zeros(N,1);
if REPRODUCIBLE
    % 调试模式:每个 worker 用相同种子
    parfor i = 1:N
        rng(i);  % 与 for 循环相同种子
        x = rand(samples_per_iter, 1);
        result_parfor(i) = mean(exp(-x.^2));
    end
else
    % 正常模式:让每个 worker 使用独立随机流(MATLAB 自动处理)
    parfor i = 1:N
        x = rand(samples_per_iter, 1);
        result_parfor(i) = mean(exp(-x.^2));
    end
end
time_parfor = toc;

%% --- 结果分析 ---
% 1. 检查是否完全一致(仅当 REPRODUCIBLE=true 时应为 true)
is_identical = isequal(result_for, result_parfor);

% 2. 计算统计均值和误差
mean_for    = mean(result_for);
mean_parfor = mean(result_parfor);
err_for     = abs(mean_for - theoretical);
err_parfor  = abs(mean_parfor - theoretical);

% 3. 输出
fprintf('\n=== 结果报告 ===\n');
fprintf('REPRODUCIBLE 模式: %s\n', string(REPRODUCIBLE));
fprintf('结果完全一致: %s\n', string(is_identical));
fprintf('\n理论值: %.9f\n', theoretical);
fprintf('for 均值:    %.9f (误差: %.2e)\n', mean_for, err_for);
fprintf('parfor 均值: %.9f (误差: %.2e)\n', mean_parfor, err_parfor);
fprintf('\nfor 耗时: %.3f s | parfor 耗时: %.3f s | 加速比: %.2fx\n', ...
        time_for, time_parfor, time_for/time_parfor);

输出结果

理论值: 0.746824133

for 均值: 0.746846070 (误差: 2.19e-05)

parfor 均值: 0.746791349 (误差: 3.28e-05)

for 耗时: 0.841 s | parfor 耗时: 0.381 s | 加速比: 2.21x

这里验证了parfor在一些比较复杂的计算情况下,可以比普通for循环快一些。

以上示例仅供参考。

相关推荐
dracula0001 天前
Simulink建模助手系列-4【批量添加信号线】
matlab
dracula0001 天前
Simulink建模助手系列-5【批量对齐模块】
matlab
dracula0002 天前
Simulink建模助手系列-2【自动创建缺失的Goto】
matlab
Evand J2 天前
【UWB与IMU紧耦合定位,MATLAB例程】UWB的TOA定位方法,与IMU紧耦合,对目标轨迹定位并输出误差统计。适用于二维平面的高精度定位导航
开发语言·matlab·平面·uwb·组合导航
禁默2 天前
零基础全面掌握层次分析法(AHP):Python实现+论文加分全攻略
python·数学建模·matlab
软件算法开发2 天前
基于火烈鸟搜索算法的LSTM网络模型(FSA-LSTM)的一维时间序列预测matlab仿真
人工智能·rnn·matlab·lstm·一维时间序列预测·火烈鸟搜索算法·fsa-lstm
简简单单做算法3 天前
基于LSTM长短记忆网络模型的文本分类算法matlab仿真,对比GRU网络
matlab·分类·gru·lstm·文本分类
dracula0003 天前
通过get()函数获取Simulink模块的所有属性
matlab
dracula0003 天前
simulink建模助手系列-1【批量创建goto和from】
matlab
IT猿手4 天前
MOEA/D(基于分解的多目标进化算法)求解46个多目标函数及一个工程应用,包含四种评价指标,MATLAB代码
开发语言·算法·matlab·多目标算法