4 个月前

PRformer:用于多变量时间序列预测的金字塔递归变换器

PRformer:用于多变量时间序列预测的金字塔递归变换器

摘要

在Transformer架构中,自注意力机制对序列顺序不变,因此需要位置嵌入来编码时间序列预测中的时间顺序。我们认为,这种依赖于位置嵌入的方法限制了Transformer有效表示时间序列的能力,尤其是在使用较长的历史窗口时。为了解决这一问题,我们提出了一种创新方法,将金字塔RNN嵌入(Pyramid RNN Embeddings, PRE)与Transformer建模多变量依赖关系的能力相结合。PRE通过使用金字塔一维卷积层构建多尺度卷积特征,从而保留时间顺序。此外,堆叠在这些特征之上的RNN学习到对序列顺序敏感的多尺度时间序列表示。将这种方法与具有注意力机制的Transformer模型结合,可以显著提升性能。我们介绍了PRformer模型,该模型将PRE与标准的Transformer编码器集成在一起,在多个实际数据集上展示了最先进的性能。这一性能表现突显了我们的方法在利用较长历史窗口方面的有效性,并强调了稳健的时间表示在最大化Transformer预测任务潜力中的关键作用。代码可在以下仓库获取:https://github.com/usualheart/PRformer。

代码仓库

usualheart/prformer
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
multivariate-time-series-forecasting-on-etth1PRformer
MAE: 0.383
MSE: 0.354
multivariate-time-series-forecasting-on-etth1-1PRformer
MAE: 0.410
MSE: 0.397
time-series-forecasting-on-electricity-192PRformer
MSE: 0.148
time-series-forecasting-on-electricity-336PRformer
MSE: 0.161
time-series-forecasting-on-electricity-720PRformer
MSE: 0.185
time-series-forecasting-on-electricity-96PRformer
MSE: 0.127
time-series-forecasting-on-etth1-192-1PRformer
MSE: 0.397
time-series-forecasting-on-etth1-336-1PRformer
MSE: 0.427
time-series-forecasting-on-etth1-720-1PRformer
MSE: 0.489
time-series-forecasting-on-etth1-96-1PRformer
MSE: 0.354
time-series-forecasting-on-etth2-192-1PRformer
MSE: 0.332
time-series-forecasting-on-etth2-336-1PRformer
MSE: 0.361
time-series-forecasting-on-etth2-720-1PRformer
MSE: 0.396
time-series-forecasting-on-etth2-96-1PRformer
MSE: 0.268
time-series-forecasting-on-ettm1-192-1PRformer
MSE: 0.324
time-series-forecasting-on-ettm1-336-1PRformer
MSE: 0.362
time-series-forecasting-on-ettm1-720-1PRformer
MSE: 0.426
time-series-forecasting-on-ettm1-96-1PRformer
MSE: 0.278
time-series-forecasting-on-ettm2-192-1PRformer
MSE: 0.219
time-series-forecasting-on-ettm2-336-1PRformer
MSE: 0.272
time-series-forecasting-on-ettm2-720-1PRformer
MSE: 0.359
time-series-forecasting-on-ettm2-96-1PRformer
MSE: 0.162
time-series-forecasting-on-traffic-192PRformer
MSE: 0.372
time-series-forecasting-on-traffic-336PRformer
MSE: 0.385
time-series-forecasting-on-traffic-720PRformer
MSE: 0.421
time-series-forecasting-on-traffic-96PRformer
MSE: 0.353
time-series-forecasting-on-weather-192PRformer
MSE: 0.188
time-series-forecasting-on-weather-336PRformer
MSE: 0.241
time-series-forecasting-on-weather-720PRformer
MSE: 0.326
time-series-forecasting-on-weather-96PRformer
MSE: 0.144

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
PRformer:用于多变量时间序列预测的金字塔递归变换器 | 论文 | HyperAI超神经