English

中国农机化学报

中国农机化学报 ›› 2024, Vol. 45 ›› Issue (12): 133-139.DOI: 10.13733/j.jcam.issn.20955553.2024.12.020

• 农业信息化工程 • 上一篇    下一篇

基于TCN-BiGRU结合自注意力机制的储粮温度预测研究

祝玉华1, 2,张钰涵1, 2,李智慧1, 2,甄彤1, 2   

  1. (1. 河南工业大学信息科学与工程学院,郑州市,450001;2. 河南工业大学粮食信息处理与控制教育部重点实验室,郑州市,450001)
  • 出版日期:2024-12-15 发布日期:2024-12-02
  • 基金资助:
    国家重点研发计划(2022YFD2100202);河南省高等学校重点科研项目计划(24A520013)

Research on temperature prediction of grain storage based on TCN-BiGRU combined with self-attention mechanism

Zhu Yuhua1, 2, Zhang Yuhan1, 2, Li Zhihui1, 2, Zhen Tong1, 2   

  1. (1. College of Information Science and Engineering, Henan University of Technology, Zhengzhou, 450001, China;
    2. Key Laboratory of Ministry of Education Grain Information Processing and Control, Henan University of Technology, Zhengzhou, 450001, China)

  • Online:2024-12-15 Published:2024-12-02

摘要:

粮食仓储管理对于国家具有重要意义,储粮温度是判断粮食仓储安全的重要指标之一。准确地预测储粮温度并及时做出相应的防护措施能够有效降低粮食仓储损耗。针对传统储粮温度预测模型预测准确度较低的问题,提出一种融合时域卷积网络(TCN)、自注意力机制(Self-Attention)和双向门控循环单元(BiGRU)的网络模型。首先通过TCN提取储粮温度数据的局部特征,并根据储粮温度数据的时序特征将自注意力机制加入网络为不同粮情特征分配权重,突出对储粮温度预测影响更大的特征,之后利用BiGRU网络学习粮情序列的双向依赖关系来获取序列中的更多信息,实现对储粮温度的预测。结果表明,所提出的模型均方根误差RMSE为0.389 5,平均绝对误差MAE为0.328 1,确定系数R2为0.991 2,与其他模型相比误差小,预测精度高,能够为粮仓的温度管控提供决策依据。

关键词: 储粮温度预测, 时域卷积网络, 自注意力机制, 门控循环单元网络

Abstract:

Efficient grain storage management is of great significance to the country, and grain storage temperature is one of the key indicators to judge grain storage security. Accurately predicting the temperature of stored grain and making timely and appropriate protective measures can effectively reduce grain storage losses. In order to address the limitations of traditional prediction models, this paper introduces a novel network model integrating a time-domain convolutional network (TCN), a self-attention mechanism, and a bi-directional gated recursive unit (BiGRU). First of all, the local features of grain storage temperature data are obtained by TCN, and the SelfAttention mechanism is incorporated into the network according to the temporal characteristics of grain storage temperature to assign weights to different grain features, highlighting the features that have a greater impact on the prediction of grain storage temperature. In addition, BiGRU network is used to learn the bi-directional dependency of grain condition sequences to obtain more information in the sequences and achieve the prediction of grain storage temperature. The experimental results show that the RMSE of the model is 0.389 5, the MAE is 0.328 1, and the R2 is 0.9912. Compared with other models, the proposed method reduces the error and improves the prediction accuracy, thus providing a solid foundation for decision-making in grain storage temperature control.

Key words: grain storage temperature prediction, time-domain convolutional network (TCN), self-attention mechanism, gated recursive unit , (GRU)

中图分类号: