作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2023, Vol. 49 ›› Issue (1): 121-129. doi: 10.19678/j.issn.1000-3428.0063718

• 人工智能与模式识别 • 上一篇    下一篇

基于混合网络模型的多维时间序列预测

刘杭2, 殷歆1, 陈杰1, 罗恒1   

  1. 1. 苏州科技大学 电子与信息工程学院, 江苏 苏州 215009;
    2. 苏州科技大学 江苏省建筑智慧节能重点实验室, 江苏 苏州 215009
  • 收稿日期:2022-01-07 修回日期:2022-02-19 发布日期:2022-03-22
  • 作者简介:刘杭(1997-),男,硕士研究生,主研方向为深度学习、时间序列预测;殷歆、陈杰,硕士研究生;罗恒(通信作者),副教授、博士。
  • 基金资助:
    国家自然科学基金(51874205,51973109)。

Multi-Dimensional Time-Series Prediction Based on Hybrid Network Models

LIU Hang2, YIN Xin1, CHEN Jie1, LUO Heng1   

  1. 1. School of Electronic & Information Engineering, Suzhou University of Science and Technology, Suzhou, Jiangsu 215009, China;
    2. Jiangsu Province Key Laboratory of Intelligent Building Energy Efficiency, Suzhou University of Science and Technology, Suzhou, Jiangsu 215009, China
  • Received:2022-01-07 Revised:2022-02-19 Published:2022-03-22

摘要: 为捕捉时间序列中潜在的特征依赖关系并实现高维时序数据的快速模糊预测,构建基于时间卷积网络(TCN)与自注意力机制的两种混合网络模型:TSANet和TSANet-MF。TSANet模型通过全局和局部两个并行卷积分量结构提取特征后,利用自注意力机制增强特征点关联程度,并结合并行的TCN增大卷积的感受野范围,最大程度地捕捉多维时序数据的周期性特征。TSANet-MF模型将TSANet作为矩阵分解算法的正则化项,使高维数据转化为具有更多时序特征的低维数据,减少计算复杂度,实现高维数据的快速模糊预测。在4种不同领域的时间序列数据集上的实验结果表明,TSANet模型在3种数据集上的预测性能均优于基准模型,尤其在高维Traffic数据集上相对平方根误差降低了19.52%~56.37%,TSANet-MF模型在Electricity和Traffic高维数据集上的训练时间相比于基准模型明显减少。上述实验结果验证了两种混合网络模型均具有较好的多维时间序列预测性能。

关键词: 多维时间序列, 时间卷积网络, 自注意力机制, 卷积神经网络, 矩阵分解, 正则化

Abstract: To capture the potential feature dependencies in times-series and demonstrate the fast fuzzy prediction of high-dimensional time-series data, this study proposes two hybrid network models, TSANet and TSANet-MF, based on the Temporal Convolutional Network(TCN) and self-attention mechanism.TSANet uses global and local parallel convolution component structures to extract features, and enhance the correlation degree of feature points with self-attention.The parallel TCN is also used to increase the receptive field range of convolution, and effectively capture periodic features of multi-dimensional time-series data.The TSANet-MF model uses TSANet as the regularization term of the Matrix Factorization(MF) algorithm, which converts high-dimensional data into low-dimensional data with more temporal characteristics, reduces computational complexity, and realizes the fast fuzzy prediction of high-dimensional data.Experiments are carried out on real-world multivariate time-series datasets in four different fields.Results show that TSANet outperforms the benchmark models in terms of prediction performance, especially on the high-dimensional Traffic data set, where TSANet decreases Root Relative Squared Error(RRSE) by 19.52%-56.37%.Compared with the benchmark models, TSANet-MF also provides a significant reduction in training time on the Electricity and Traffic high-dimensional data sets.These experimental results help verify that both hybrid network models have good performance in multi-dimensional time-series forecasting.

Key words: multi-dimensional time-series, Temporal Convolutional Network(TCN), self-attention mechanism, Convolutional Neural Network(CNN), Matrix Factorization(MF), regularization

中图分类号: