Wheat Yield Prediction Based on Continuous Wavelet Transform and
Machine Learning
FAN Jie-jie1, 2, QIU Chun-xia1, FAN Yi-guang2, CHEN Ri-qiang2, LIU Yang2, BIAN Ming-bo2, MA Yan-peng2, YANG Fu-qin4, FENG Hai-kuan2, 3*
1. School of Surveying and Mapping Science and Technology, Xi'an University of Science and Technology, Xi'an 710054, China
2. Key Laboratory of Quantitative Remote Sensing in Agriculture of Ministry of Agriculture and Rural Affairs, Information Technology Research Center, Beijing Academy of Agriculture and Forestry Sciences, Beijing 100097, China
3. National Engineering and Technology Center for Information Agriculture, Nanjing Agricultural University, Nanjing 210095, China
4. College of Civil Engineering, Henan University of Engineering, Zhengzhou 451191, China
Abstract：Timely and accurate crop yield estimation is crucial for making informed decisions regarding crop management and assessing food security. This study aims to develop a method that combines continuous wavelet transform (CWT) with machine learning to predict wheat yield accurately. This research is based on the spectral data of canopy height and yield data obtained from two-year field trials conducted during wheat growth's flowering and filling stages in 2020—2021. Initially, CWT is employed to extract three wavelet features (WFs), namely Bortua-WFs based on the Bortua method, 1%R2-WFs representing WFs along with the top 1% determination coefficient for wheat yield, and SS-WFs encompassing all WFs under a single decomposition scale. Subsequently, three machine learning algorithms Random Forest (RF), K-nearest neighbor (KNN), and extreme gradient Lift (XGBoost) are utilized to construct the yield prediction model. Finally, optimal spectral features are selected using the same methodology for modeling and comparison purposes. The results demonstrate that: (1) all three WFs models combined with machine learning methods perform well, with higher accuracy and stability observed in the model built based on Boruta-WFs. (2) Compared to the spectral characteristic model, improved accuracy was achieved by utilizing Bortua-WFs at each growth stage; specifically, an increase in R2 accuracy by 17.5%, 4%, and 39.6% during flowering stage, as well as an increase by 8.4%, 5.6%, and 16.9% during filling stage respectively were observed across different models.(3) The estimation model at the grouting stage outperformed that at the flowering stage; particularly noteworthy was the performance of XGBoost when combined with Bortua-WFs, which yielded an R2 value of 0.83 accompanied by an RMSE value of 0.78 t·ha-1. This study compared the performance of different characteristics and methods. It determined the best model accuracy under different schemes, which can provide technical references for the accurate wheat yield prediction by spectral technology.
樊杰杰，邱春霞，樊意广，陈日强，刘 杨，边明博，马彦鹏，杨福芹，冯海宽. 基于连续小波变换和机器学习的小麦产量预测[J]. 光谱学与光谱分析, 2024, 44(10): 2890-2899.
FAN Jie-jie, QIU Chun-xia, FAN Yi-guang, CHEN Ri-qiang, LIU Yang, BIAN Ming-bo, MA Yan-peng, YANG Fu-qin, FENG Hai-kuan. Wheat Yield Prediction Based on Continuous Wavelet Transform and
Machine Learning. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2024, 44(10): 2890-2899.
[1] Xue H, Xu X, Zhu Q, et al. Computers and Electronics in Agriculture, 2024, 218: 108731.
[2] Chaofa B, Hongtao S, Suqin W, et al. Remote Sensing, 2022, 14(6): 1474.
[3] Wei Y, Tyler N, Ziyuan H, et al. Computers and Electronics in Agriculture, 2021, 184: 106092.
[4] Fu Y, Yang G, Wang J, et al. Computers and Electronics in Agriculture, 2014, 100: 51.
[5] Messina G, Modica G. Remote Sensing, 2020, 12 (9): 1491.
[6] XIAO Lu-jie, YANG Wu-de, FENG Mei-chen, et al(肖璐洁，杨武德，冯美臣，等). Chinese Journal of Ecology(生态学杂志), 2022, 41(7): 1433.
[7] CUI Huai-yang, XU Hui, ZHANG Wei, et al(崔怀洋，徐 晖，张 伟，等). Journal of Triticeae Crops(麦类作物学报), 2015, 35(8): 1155.
[8] Liu Y, Sun L, Liu B, et al. Remote Sensing, 2023, 15(19): 4800.
[9] Yang L, Haikuan F, Jibo Y, et al. Remote Sensing, 2022, 14(21): 5449.
[10] Liu Y, Sun L, Liu B, et al. Remote Sensing, 2023, 15(19): 4158.
[11] Li Y, Zeng H, Zhang M, et al. International Journal of Applied Earth Observation and Geoinformation, 2023, 118: 103269.
[12] Joshi A, Pradhan B, Chakraborty S, et al. Ecological Informatics, 2023, 77: 102194.
[13] Xiaokai C, Fenling L, Qingrui C. Remote Sensing, 2023, 15(4): 997.
[14] Haojie L, Minzan L, Junyi Z, et al. International Journal of Agricultural and Biological Engineering, 2018, 11(6): 132.
[15] Cheng T, Riaño D, Ustin S L. Remote Sensing of Environment, 2014, 143: 39.
[16] Fitzgerald M A, Mccouch S R, Hall R D. Trends in Plant Science, 2008, 14(3): 133.
[17] Gitelson A A, Kaufman Y J, Merzlyak M N. Remote Sensing of Environment, 1996, 58(3): 289.
[18] Gamon J A, PeñUelas J, Field C B. Remote Sensing of Environment, 1992, 41(1): 35.
[19] Li Z, Li Z, Fairbairn D, et al. Computers and Electronics in Agriculture, 2019, 162: 174.
[20] Dash J, Curran P J. IEEE International Geoscience & Remote Sensing Symposium, 2004, 25(549): 151.
[21] Houborg R, Mccabe M F. ISPRS Journal of Photogrammetry and Remote Sensing, 2018, 135: 173.
[22] Han L, Yang G, Dai H, et al. Plant Methods, 2019, 15(1): 1.
[23] Hernandez-Clemente R, Navarro-Cerrillo R M, Zarco-Tejada P J. IEEE Transactions on Geoscience and Remote Sensing, 2014, 52(8): 5206.
[24] Yiguang F, Haikuan F, Jibo Y, et al. Remote Sensing, 2023, 15(3): 602.
[25] Liu Y, Liu S, Li J, et al. Computers and Electronics in Agriculture, 2019, 166: 105026.
[26] Yue J, Yang G, Tian Q, et al. ISPRS Journal of Photogrammetry and Remote Sensing, 2019, 150: 226.
[27] Zhang X, Zhao J, Yang G, et al. Remote Sensing, 2019, 11(23): 2752.
[28] Kaufman Y J, Tanre D. IEEE Trans. Geosci. Remote Sens., 1992, 30(2): 261.
[29] Penuelas J, Baret F, Filella I. Photosynthetica, 1995, 31(2): 221.
[30] Qi J, Chehbouni A, Huete A R, et al. Remote Sensing of Environment, 1994, 48(2): 119.
[31] Ebrahimi H, Rajaee T. Global and Planetary Change, 2017, 148: 181.
[32] Hongfei Y, Yurong Q, Feng Y, et al. Soil Science, 2012, 177(11): 674.
[33] Bruce L M, Morgan C, Larsen S. IEEE Trans. Geoscience and Remote Sensing, 2001, 39(10): 2217.
[34] Zhang N, Chen M, Yang F, et al. Remote Sensing, 2022, 14(18): 4434.
[35] Kursa M B, Jankowski A, Rudnicki W R. Fundamenta Informaticae, 2010, 101(4): 271.
[36] Degenhardt Frauke, Seifert Stephan, Szymczak Silke. Briefings in Bioinformatics, 2019, 20(2): 492.
[37] Hong Y, Chen S, Chen Y, et al. Soil & Tillage Research, 2020, 199: 104589.
[38] Tianao W, Wei Z, Xiyun J, et al. Computers and Electronics in Agriculture, 2021, 184: 106039.
[39] Maria B, Chris C, Aris K. Remote Sensing, 2022, 14(16): 3880.
