
MATLAB 代码实现了一个基于粒子群优化PSO的LightGBM 回归预测模型并结合核密度估计KDE进行概率性区间预测。1. 研究背景在工程与科学领域中许多实际问题如能源消耗、金融预测、环境监测需要建立高精度、可解释的回归模型。LightGBM作为一种高效梯度提升框架在非线性建模中表现优异但其性能受超参数如叶子节点数、学习率、树深度影响较大。粒子群优化PSO可用于自动搜索最优超参数组合提高模型泛化能力。此外点预测往往不足以反映预测的不确定性因此引入概率区间预测如基于核密度估计的置信区间有助于量化预测风险。2. 主要功能数据预处理划分训练/测试集、归一化超参数优化使用 PSO 自动优化 LightGBM 的三个关键参数num_leaves、learning_rate、max_depth模型训练与预测利用最优参数训练 LightGBM 回归模型输出点预测结果误差分析与核密度估计对训练误差进行自适应带宽核密度估计ABKDE构建误差概率分布区间预测基于误差分布生成多个置信水平下的预测区间性能评估计算点预测指标R²、MAE、RMSE、MAPE和区间预测指标PINAW、PICP、CRPS、CWC3. 算法步骤数据导入与划分读取 Excel 数据按比例划分训练集与测试集数据归一化使用mapminmax将特征与目标值映射到 [0,1] 区间PSO 超参数寻优定义目标函数getObjValue以 LightGBM 在验证集上的 RMSE 为优化目标PSO 搜索叶子节点数、学习率、树深度的最优组合LightGBM 模型训练使用最优参数构建 LightGBM 模型设置早停机制防止过拟合点预测与误差计算在测试集上预测并反归一化计算 RMSE、R² 等指标误差分布建模与区间预测对训练误差进行自适应核密度估计ABKDE得到误差概率密度函数根据分位数计算不同置信度下的误差上下界叠加到点预测上形成预测区间结果可视化与指标输出绘制特征重要性、误差概率密度、预测区间图输出点预测与区间预测的评价指标4. 技术路线数据加载 → 数据划分与归一化 → PSO优化LightGBM超参数 → 训练最优LightGBM模型 → 点预测与误差分析 → 自适应核密度估计误差分布 → 生成多置信度预测区间 → 指标评估与可视化5. 公式原理PSO 速度与位置更新vi,jt1wvi,jtc1r1(pBesti,j−xi,jt)c2r2(gBestj−xi,jt) v_{i,j}^{t1} w v_{i,j}^t c_1 r_1 (pBest_{i,j} - x_{i,j}^t) c_2 r_2 (gBest_j - x_{i,j}^t)vi,jt1wvi,jtc1r1(pBesti,j−xi,jt)c2r2(gBestj−xi,jt)xi,jt1xi,jtvi,jt1 x_{i,j}^{t1} x_{i,j}^t v_{i,j}^{t1}xi,jt1xi,jtvi,jt1其中www为惯性权重线性递减c1,c2c_1, c_2c1,c2为加速因子r1,r2r_1, r_2r1,r2为随机数。LightGBM 目标函数回归任务采用均方误差MSE作为损失函数通过梯度提升迭代优化。自适应带宽核密度估计ABKDE根据误差分布局部密度调整带宽使估计的概率密度更贴近真实分布用于准确提取误差分位数。区间预测指标PICP区间覆盖率实际值落在预测区间内的比例PINAW区间平均宽度衡量区间宽度的归一化值CRPS连续排序概率分数综合评价预测分布的准确性CWC覆盖率宽度准则结合 PICP 与 PINAW 的复合指标6. 参数设定参数类别参数名称取值数据训练集比例0.7PSO种群数20最大迭代次数20寻优维度3参数范围num_leaves ∈ [2,64]learning_rate ∈ [0.8,1]max_depth ∈ [2,10]LightGBM迭代次数20最大早停轮次5特征采样比例0.9样本采样比例0.8区间预测置信度97.5%、95%、87.5%、75%、62.5%、55%、52.5%7. 运行环境软件MATLAB20208. 应用场景该方法适用于需要点预测且同时量化预测不确定性的回归任务典型场景包括电力负荷预测结合置信区间便于调度决策空气质量指数AQI预测提供概率性预警金融时间序列预测如股票收益率区间预测工业过程控制如设备剩余寿命预测完整代码私信回复PSO-LightGBM-ABKDE粒子群算法优化轻量级梯度提升机自适应带宽核密度估计多变量回归区间预测Matlab实现