《计算机应用》唯一官方网站 ›› 2022, Vol. 42 ›› Issue (1): 287-293.DOI: 10.11772/j.issn.1001-9081.2021010099

• 前沿与综合应用 • 上一篇    

基于图卷积网络和门控循环单元的多站点气温预测模型

马栋林, 马司周(), 王伟杰   

  1. 兰州理工大学 计算机与通信学院,兰州 730050
  • 收稿日期:2021-01-18 修回日期:2021-03-01 接受日期:2021-03-30 发布日期:2021-04-15 出版日期:2022-01-10
  • 通讯作者: 马司周
  • 作者简介:马栋林(1971—),男,甘肃兰州人,副教授,主要研究方向:智能信息处理、模式识别、深度学习
    马司周(1995—),男,甘肃通渭人,硕士研究生,主要研究方向:图神经网络、深度学习
    王伟杰(1994—),女,黑龙江齐齐哈尔人,博士研究生,主要研究方向:深度学习、声纹识别。
  • 基金资助:
    国家自然科学基金资助项目(51668043)

Multi-site temperature prediction model based on graph convolutional network and gated recurrent unit

Donglin MA, Sizhou MA(), Weijie WANG   

  1. School of Computer and Communication,Lanzhou University of Technology,Lanzhou Gansu 730050,China
  • Received:2021-01-18 Revised:2021-03-01 Accepted:2021-03-30 Online:2021-04-15 Published:2022-01-10
  • Contact: Sizhou MA
  • About author:MA Donglin, born in 1971, associate professor. His research interests include intelligent information processing, pattern recognition, deep learning.
    MA Sizhou, born in 1995, M. S. candidate. His research interests include graph neural network, deep learning.
    WANG Weijie, born in 1994, Ph. D. candidate. Her research interests include deep learning, voiceprint recognition.
  • Supported by:
    National Natural Science Foundation of China(51668043)

摘要:

时空预测任务在神经科学、交通、气象等领域应用广泛。气温预测作为典型的时空预测任务,需要挖掘气温数据中固有的时空特征。针对现有气温预测算法存在预测误差大、空间特征提取不充分的问题,提出一种基于图卷积网络和门控循环单元的气温预测(GCN-GRU)模型。首先,使用重新分配权重和多阶近邻连接方式修正图卷积网络(GCN),以有效挖掘气象数据独特的空间特征;然后,将门控循环单元(GRU)中每个循环单元的矩阵乘法替换成图卷积操作,并将所有的循环单元串联起来构成图卷积门控层;接着,使用图卷积门控层搭建网络主体结构来提取数据的时空特征;最后,通过一个全连接的输出层输出气温预测结果。通过与GRU和长短期记忆网络(LSTM)等单一模型对比,GCN-GRU模型的平均绝对误差(MAE)分别减小了0.67和0.83;与切比雪夫图卷积和长短期记忆网络结合的预测模型(Cheb-LSTM)、图卷积网络和长短期记忆网络结合的预测模型(GCN-LSTM)对比,平均绝对误差分别减小了0.36和0.23。

关键词: 时空预测, 气温预测, 多站点, 时空特征, 图卷积网络, 门控循环单元

Abstract:

Spatio-temporal prediction task is widely applied in neuroscience, transportation, meteorology and other fields. As a typical spatio-temporal prediction task, temperature prediction needs to dig out the inherent spatio-temporal characteristics of temperature data. Aiming at the problems of large prediction error and insufficient spatial feature extraction in the existing temperature prediction algorithms, a temperature prediction model based on Graph Convolutional Network and Gated Recurrent Unit (GCN-GRU) was proposed. Firstly, the methods of weight redistribution and multi-order neighbor connection were used to modify Graph Convolutional Network (GCN) in order to effectively mine the unique spatial characteristics of the meteorological data. Secondly, the matrix multiplication of each recurrent unit in the Gated Recurrent Unit (GRU) was replaced by graph convolution operation, and all the recurrent units were connected in series to form a graph convolutional gating layer. Then, the graph convolutional gating layer was used to build the main network structure to extract the spatio-temporal characteristics of the data. Finally, the temperature prediction results were output through a fully connected output layer. Compared with the single models such as Gated Recurrent Unit (GRU) and Long Short-Term Memory (LSTM), GCN-GRU had the Mean Absolute Error (MAE)reduced by 0.67 and 0.83 respectively; compared with the prediction model combined with Chebyshev graph convolution and Long Short-Term Memory (Cheb-LSTM) and the prediction model combined with Graph Convolutional Network and Long Short-Term Memory (GCN-LSTM), the proposed model had the MAE reduced by 0.36 and 0.23 respectively.

Key words: spatio-temporal prediction, temperature prediction, multi-site, spatio-temporal characteristic, Graph Convolutional Network (GCN), Gated Recurrent Unit (GRU)

中图分类号: