当前位置：首页 > news >正文

基于MATLAB实现多变量高斯过程回归（GPR）

news 2026/5/12 9:00:43

一、数据生成与预处理

%% 生成多变量输入输出数据
rng(0); % 固定随机种子
n_samples = 200;% 输入特征（3维）
X = [linspace(-5,5,n_samples)', randn(n_samples,2)*2];% 输出目标（2维非线性函数）
Y(:,1) = sin(X(:,1)).*exp(-0.1*X(:,1).^2) + 0.5*X(:,2);
Y(:,2) = cos(X(:,1)).*exp(-0.1*X(:,1).^2) + 0.3*X(:,3) + 0.1*randn(n_samples,1);%% 数据可视化
figure;
subplot(2,1,1);
scatter3(X(:,1),X(:,2),X(:,3),50,Y(:,1),'filled');
xlabel('X1'); ylabel('X2'); zlabel('X3');
title('输入数据分布（3D）');subplot(2,1,2);
plot(Y(:,1),'r',Y(:,2),'b');
legend('输出1','输出2');
title('输出目标分布');

二、模型训练与预测

%% 数据划分（70%训练，30%测试）
cv = cvpartition(size(X,1),'HoldOut',0.3);
X_train = X(training(cv),:);
Y_train = Y(training(cv),:);
X_test = X(test(cv),:);
Y_test = Y(test(cv),:);%% 数据归一化（参考）
[yp_train, ps_input] = mapminmax(X_train',0,1);
yp_test = mapminmax('apply',X_test',ps_input);
[yt_train, ps_output] = mapminmax(Y_train',0,1);
yt_test = mapminmax('apply',Y_test',ps_output);%% 训练GPR模型（多输出）
kernel = {'ardsquaredexponential','ardsquaredexponential'}; % 自动相关尺度核
gpr_model = fitrgp(yp_train', yt_train', 'Kernel', kernel, ...'Basis', 'constant', 'FitMethod', 'exact', ...'PredictMethod', 'exact', 'Standardize', 1);%% 预测
[ypred,ypredci] = predict(gpr_model, yp_test');
ypred = mapminmax('reverse',ypred',ps_output)';
ypredci = mapminmax('reverse',ypredci',ps_output)';

三、性能评估（参考）

%% 计算评价指标
R2 = 1 - sum((Y_test - ypred).^2)/sum((Y_test - mean(Y_test)).^2);
MAE = mean(abs(Y_test - ypred));
MSE = mean((Y_test - ypred).^2);
RMSE = sqrt(MSE);%% 输出结果
fprintf('===== 模型评估 =====
');
fprintf('R²: %.4f\n', R2);
fprintf('MAE: %.4f\n', MAE);
fprintf('RMSE: %.4f\n', RMSE);

四、可视化分析

%% 三维预测结果可视化
figure;
scatter3(X_test(:,1),X_test(:,2),X_test(:,3),50,Y_test(:,1),'filled');
hold on;
plot3(X_test(:,1),X_test(:,2),X_test(:,3),...'o', 'MarkerSize', 8, 'MarkerFaceColor', 'r');
xlabel('X1'); ylabel('X2'); zlabel('X3');
title('输出1预测结果（红点为真实值，曲面为预测值）');%% 置信区间可视化（以输出2为例）
figure;
plot(Y_test(:,2),'b', 'LineWidth',1.5);
hold on;
plot(ypred(:,2),'r--', 'LineWidth',1.5);
fill([1:length(ypred)], [ypred(:,2); flipud(ypredci(:,2))], ...[0.8 0.8 1], 'FaceAlpha',0.3, 'EdgeColor','none');
xlabel('样本序号'); ylabel('输出值');
legend('真实值','预测均值','95%置信区间');
title('输出2预测置信区间');

五、关键参数说明

核函数选择 ardsquaredexponential：自动相关尺度平方指数核（推荐多变量场景） matern52：Matérn 5/2核（适合有导数连续性的数据）

超参数优化

可通过最大似然估计自动优化：

gpr_model = fitrgp(yp_train', yt_train', ...'Basis', 'constant', 'FitMethod', 'exact', ...'OptimizeHyperparameters', 'all', ...'HyperparameterOptimizationOptions', struct('AcquisitionFunctionName','expected-improvement-plus'));

参考代码多变量高斯过程回归示例 www.youwenfan.com/contentcnp/97722.html

六、扩展应用场景

时序预测

将时间序列数据转换为监督学习格式：

X_t = [X(1:end-1,:) X(2:end,:)]; % 输入为时间步特征
Y_t = X(2:end,:); % 输出为下一时刻状态

不确定性量化

通过预测方差分析不确定性：

[~, S] = predict(gpr_model, yp_test');
figure;
semilogy(Y_test(:,1), 'b', 'LineWidth',1.5);
hold on;
plot(ypred(:,1), 'r--', 'LineWidth',1.5);
errorbar(1:length(ypred), ypred(:,1), 2*sqrt(S(:,1)), 'r.');