Predicting Grain Losses and Waste Rate Along the Entire Chain: A Multitask Multigated Recurrent Unit Autoencoder Based Method

Jie Cao,Youquan Wang,Jing He,Weichao Liang,Haicheng Tao,Guixiang Zhu
DOI: https://doi.org/10.1109/tii.2020.3030709
IF: 12.3
2021-06-01
IEEE Transactions on Industrial Informatics
Abstract:Predicting grain losses and waste rate (LWR) is critical for agricultural planning and grain policy development. Capturing the stage interaction and generating robust features are the main challenges in grain LWR prediction. In this article, we propose MTGA, a Multitask Gated recurrent unit (GRU) Autoencoder, approach to 1) obtain the robust feature representation for the prediction task and 2) explore the time-ordered interactions among different stages of the grain chain. Specifically, we design multiple GRU encoderdecoder pairs to co-reconstruct the stage features in a common space for robust feature learning. Then, an attention mechanism is proposed better to fuse the reconstructed features from the GRU encoderdecoder pairs. Furthermore, we utilize the multitask for reconstructed loss and grain LWR prediction. We introduce the reconstructed loss task as an auxiliary task to help us to represent the robust features. Besides, we introduce the LWR prediction as main task to learn the parameters for prediction task. We collected the data with questionnaires, interviews, or data from grain management institutes for experiments. The evaluation results show that grain LWR prediction by our approach achieves the best results compared to several state-of-the-art prediction models. Moreover, our method gains overall performance decline of 12.518.3 on mean absolute error and root mean square error metrics.
automation & control systems,computer science, interdisciplinary applications,engineering, industrial
What problem does this paper attempt to address?