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

时空预测任务在神经科学、交通、气象等领域应用广泛.气温预测作为典型的时空预测任务,需要挖掘气温数据中固有的时空特征.针对现有气温预测算法存在预测误差大、空间特征提取不充分的问题,提出一种基于图卷积网络和门控循环单元的气温预测(GCN-GRU)模型.首先,使用重新分配权重和多阶近邻连接方式修正图卷积网络(GCN),以有效...

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

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.