HUBEI AGRICULTURAL SCIENCES ›› 2024, Vol. 63 ›› Issue (8): 109-115.doi: 10.14088/j.cnki.issn0439-8114.2024.08.019

• Production and Growth Model • Previous Articles     Next Articles

County level rice yield prediction model based on CNN-BiLSTM and residual attention

LIANG Ze1, CAO Shan-shan2a,2b,3, KONG Fan-tao2c, SUN Wei2a,2b,3   

  1. 1. College of Computer and Information Engineering, Xinjiang Agricultural University, Urumqi 830052, China;
    2a. Agricultural Information Institute; 2b. National Agriculture Science Data Center; 2c. Institute of Agricultural Economics and Development, Chinese Academy of Agricultural Sciences, Beijing 100081, China;
    3. National Nanfan Research Institute (Sanya), Chinese Academy of Agricultural Sciences, Sanya 572024, Hainan, China
  • Received:2024-02-27 Online:2024-08-25 Published:2024-09-05

Abstract: A county-level rice yield prediction model (CNN-BiLSTM-RA) was proposed, which integrated convolutional neural network (CNN), bidirectional long short term memory network (BiLSTM), and residual attention (RA) mechanism, key spatial features were effectively extracted from county-level rice meteorological data through CNN layers, the dynamic changes of time series data were deeply analyzed using BiLSTM layers, and RA mechanism was introduced to enhance the recognition and capture of key features in meteorological data. Using historical rice yield and meteorological data from 81 counties in Guangxi from 2015 to 2017 as samples, the prediction accuracy and effectiveness of the CNN-BiLSTM-RA model were compared with CNN, TRANSFORMER, BiLSTM, CNN-BiLSTM, and BiLSTM-RA models. The results showed that the R2, MAE, RMSE, and MAPE of the CNN-BiLSTM-RA model were 0.986 1, 0.121 9, 0.224 8, and 0.864 8, respectively, indicating a high degree of fit between the predicted and actual values of the model. The CNN-BiLSTM-RA model fully utilized the spatial feature extraction ability of CNN, the time series data analysis advantages of BiLSTM, and the RA mechanism’s ability to enhance key feature capture. It was a new method suitable for high-precision prediction of rice yield in counties.

Key words: rice yield prediction, convolutional neural network, bidirectional long short term memory network, residual attention

CLC Number: