Persistence Initialization: a novel adaptation of the Transformer architecture for time series forecasting

被引:10
作者
Haugsdal, Espen [1 ]
Aune, Erlend [1 ,2 ]
Ruocco, Massimiliano [1 ,3 ]
机构
[1] Norwegian Univ Sci & Technol, Trondheim, Norway
[2] BI Norwegian Business Sch, Oslo, Norway
[3] Sintef Digital, Trondheim, Norway
关键词
Transformer; Time series forecasting; M4; competition; Deep neural networks;
D O I
10.1007/s10489-023-04927-4
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Time series forecasting is an important problem, with many real world applications. Transformer models have been successfully applied to natural language processing tasks, but have received relatively little attention for time series forecasting. Motivated by the differences between classification tasks and forecasting, we propose PI-Transformer, an adaptation of the Transformer architecture designed for time series forecasting, consisting of three parts: First, we propose a novel initialization method called Persistence Initialization, with the goal of increasing training stability of forecasting models by ensuring that the initial outputs of an untrained model are identical to the outputs of a simple baseline model. Second, we use ReZero normalization instead of Layer Normalization, in order to further tackle issues related to training stability. Third, we use Rotary positional encodings to provide a better inductive bias for forecasting. Multiple ablation studies show that the PI-Transformer is more accurate, learns faster, and scales better than regular Transformer models. Finally, PI-Transformer achieves competitive performance on the challenging M4 dataset, both when compared to the current state of the art, and to recently proposed Transformer models for time series forecasting.
引用
收藏
页码:26781 / 26796
页数:16
相关论文
共 29 条
  • [1] Ba Jimmy Lei, 2016, Layer normalization
  • [2] Bachlechner T, 2021, PR MACH LEARN RES, V161, P1352
  • [3] Baevski Alexei, 2019, IN TERNATIONAL C LE
  • [4] Brown TB, 2020, ARXIV
  • [5] Chang SY, 2017, ADV NEUR IN, V30
  • [6] Chen LL, 2021, ADV NEUR IN, V34
  • [7] Devlin J, 2019, 2019 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL HLT 2019), VOL. 1, P4171
  • [8] Glorot X., 13 INT C ART INT STA
  • [9] Goodfellow I, 2016, ADAPT COMPUT MACH LE, P1
  • [10] Reducing the dimensionality of data with neural networks
    Hinton, G. E.
    Salakhutdinov, R. R.
    [J]. SCIENCE, 2006, 313 (5786) : 504 - 507