A Prediction Method of Seedling Transplanting Time with DCNN-LSTM Based on the Attention Mechanism

Abstract

To improve the production efficiency and reduce the labor cost of seedling operations, cabbage was selected as the research subject, and a novel approach based on the attention mechanism combining the deep convolutional neural network (DCNN) and long short-term memory (LSTM) is proposed. First, the cabbage growth data and environmental monitoring data were normalized, and input samples were obtained by sliding the time window. Then, the DCNN and the LSTM were used to extract the spatial feature information and temporal correlation of the samples, respectively. At the same time, the attention mechanism was used to set the weight coefficients of different feature information and highlight the role of the main features of the sample in the model, thereby improving the prediction accuracy. By analyzing the experimental data collected by the Shandong Seedling Plant, the DCNN-LSTM method based on the proposed attention mechanism achieved good prediction results, providing experience for the engineering application of decision-making regarding seedling transplanting time. The experimental data showed that the mean absolute error, root-mean-square error, mean absolute percentage error, and symmetric mean absolute percentage error of the prediction results of this method were 0.356, 0.507, 0.157, and 0.082, respectively. Compared with the CNN, LSTM, LSTM-Attention and CNN-LSTM models, this model showed higher prediction accuracy.