结合虚拟样本生成的油菜花期集成学习预测模型  

Ensemble learning prediction model for rapeseed flowering periods incorporating virtual sample generation

作　　者：谢乾伟 薛丰昌[1] 陈剑飞 XIE Qianwei;XUE Fengchang;CHEN Jianfei(Meteorological Disaster Geographic Information Engineering Laboratory,Nanjing University of Information Science&Technology,Nanjing 210044,China;Guangxi Zhuang Autonomous Region Lightning Protection Center,Nanning 530000,China)

机构地区：[1]南京信息工程大学气象灾害地理信息工程实验室,南京210044 [2]广西壮族自治区防雷中心,南宁530000

出　　处：《农业工程学报》2024年第19期159-167,共9页Transactions of the Chinese Society of Agricultural Engineering

基　　金：广西重点研发计划项目(桂科AB22080101);南昌市农业气象重点实验室开放基金项目(2019NNZS102)。

摘　　要：针对统计和线性回归模型难以完全揭示花期影响因子与花期之间的复杂非线性关系及油菜花期样本稀少的问题,提出了一种结合虚拟样本生成的集成学习算法来实现油菜花期的预测。该研究利用浙江省衢州市龙游县1999—2023年油菜盛花期与1998—2023年气象数据,通过基于高斯混合模型的虚拟样本生成(GMM-based virtual sample generation,GMM-VSG)算法与三次样条插值法(cubic spline interpolation)分别对原始样本进行扩充,采用8种机器学习算法建模并基于贝叶斯优化器进行超参数优化,最后通过Stacking集成学习方法,对8种算法进行不同的组合,建立了油菜花期预测模型。研究结果表明:相较于原始数据集,通过三次样条插值法与高斯混合模型生成的两个扩展数据集在各种机器学习算法中的性能显著提升,其中通过三次样条插值法生成的数据集表现最为优异。通过Stacking思想能提升模型的精度,其中以核岭回归(kernel ridge regression,KRR)、支持向量回归(support vector regression,SVR)、极端梯度提升树(extreme gradient boosting,XGBoost)这3种算法作为基模型,线性回归作为元模型的SRX_L模型表现最优,其平均绝对误差、均方根误差和决定系数,分别为0.1056 d、0.1227 d和0.9997。该研究结果可为油菜花期的准确预测提供有效方法。Linear regression cannot fully reveal the complex non-linear relationships among influencing factors and scarce samples in the flowering period.In this study,ensemble learning was proposed to predict the flowering periods of rapeseed.The generation of virtual samples was also incorporated.The rapeseed in full bloom and meteorological data was utilized in Longyou County,Quzhou City,Zhejiang Province,China from 1998 to 2023.The original samples were expanded using Gaussian Mixture Model-based Virtual Sample Generation and Cubic Spline Interpolation.Two new datasets were obtained,each of which contained 985 samples.The models were established using eight machine learning methods:Random Forest(RF),Kernel Ridge Regression(KRR),Ridge Regression(RR),Least Absolute Shrinkage and Selection Operator(Lasso),Support Vector Regression(SVR),Extreme Gradient Boosting(XGBoost),Light Gradient Boosting Machine(LightGBM),and Gradient Boosting Decision Tree(GBDT).Hyperparameter optimization was conducted using a Bayesian optimizer.Finally,a prediction model was established for the rapeseed flowering period using stacking ensemble learning.The vast majority of models demonstrated superior performance on the Cubic interpolation dataset,compared with the original and GMM-VSG dataset.Specifically,the RF model was achieved in an RMSE of 0.679 d,an MAE of 0.351 d,and an R2 of 0.990,indicating significant improvements,compared with the original dataset with an RMSE of 6.286 d,an MAE of 5.028 d,and an R2 of 0.201,as well as the GMM-VSG dataset with an RMSE of 2.680 d,an MAE of 1.588 d,and an R2 of 0.881.Additionally,the SVR model also performed better on the Cubic dataset,with an RMSE of 0.849 d,an MAE of 0.333 d,and an R2 of 0.984,indicating a better performance than before.LightGBM as an ensemble learning was performed the best on the Cubic dataset,with the lowest RMSE of 0.613 d MAE of 0.336 d,and the highest R2 of 0.992.The strong feature learning and noise resistance were verified to capture the complex relationships within the dataset

关 键 词：集成学习 虚拟样本生成 花期预测 油菜 STACKING 

分 类 号：S565.4[农业科学—作物学]