Winter Wheat Yield Prediction Based on the ASTGNN Model Coupled with Multi-Source Data

Abstract

Timely and accurate prediction of winter wheat yields, which is crucial for optimizing production management, maintaining supply–demand balance, and ensuring food security, depends on interactions among numerous factors, such as climate, surface characteristics, and soil quality. Despite the extensive application of deep learning models in this field, few studies have analyzed the effect of the large-scale geospatial characteristics of neighboring regions on crop yields. Therefore, we present an attention-based spatio-temporal Graph Neural Network (ASTGNN) model coupled with geospatial characteristics and multi-source data for improved accuracy of winter wheat yield estimation. The datasets used in this study included multiple types of remote sensing, meteorological, soil, crop yield, and planting area data for Anhui, China, from 2005 to 2020. The results showed that multi-source data led to higher prediction performance than single-source data, and enabled accurate prediction of winter wheat yields three months prior to harvest. Furthermore, the ASTGNN model provided better prediction performance than two traditional crop yield prediction models (R2 = 0.70, RMSE = 0.21 t/ha, MAE = 0.17 t/ha). Therefore, ASTGNN enhances the accuracy of crop yield prediction by incorporating geospatial characteristics. This research has implications for improving agricultural production management, promoting the development of digital agriculture, and addressing climate change in agriculture.