Stock Price Prediction Based on Temporal Fusion Transformer

被引:14
作者
Hu, Xiaokang [1 ]
机构
[1] Univ Sydney, Fac Informat Technol, Sydney, NSW 2017, Australia
来源
2021 3RD INTERNATIONAL CONFERENCE ON MACHINE LEARNING, BIG DATA AND BUSINESS INTELLIGENCE (MLBDBI 2021) | 2021年
关键词
stock market prediction; machine learning; deep learning; regression analysis; support vector regression; ISM transformer; Temporal Fusion Transformer;
D O I
10.1109/MLBDBI54094.2021.00019
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Stock price prediction has been an important financial problem which receives increasing attention in the past decades. Existing literature focusing on stock markets forecasting considers the high volatility of stock prices caused by multiple factors. During past years, a number of deep neural networks have been gaining much attention, achieving more accurate results compared to the traditional linear and non-linear approaches. However, most of these neural networks only take time-series features into consideration, ignoring static metadata potentially affecting stock markets. This paper utilizes the state-of-art Temporal Fusion Transformer (TFT) for stock price prediction compared with Support Vector Regression (SVR) and Long Short-Term Memory (LSTM). The comparison for each model is evaluated based on two metrics: Mean Square Error (MSE) and Symmetric Mean Absolute Percentage Error (SMAPE). The results document that TFT achieves the lowest errors.
引用
收藏
页码:60 / 66
页数:7
相关论文
共 12 条
  • [1] Feature normalization and likelihood-based similarity measures for image retrieval
    Aksoy, S
    Haralick, RM
    [J]. PATTERN RECOGNITION LETTERS, 2001, 22 (05) : 563 - 582
  • [2] Stock Return Predictability and Variance Risk Premia: Statistical Inference and International Evidence
    Bollerslev, Tim
    Marrone, James
    Xu, Lai
    Zhou, Hao
    [J]. JOURNAL OF FINANCIAL AND QUANTITATIVE ANALYSIS, 2014, 49 (03) : 633 - 661
  • [3] Stock return forecasting: Some new evidence
    Dinh Hoang Bach Phan
    Sharma, Susan Sunila
    Narayan, Paresh Kumar
    [J]. INTERNATIONAL REVIEW OF FINANCIAL ANALYSIS, 2015, 40 : 38 - 51
  • [4] STANDARDIZATION, COMPATIBILITY, AND INNOVATION
    FARRELL, J
    SALONER, G
    [J]. RAND JOURNAL OF ECONOMICS, 1985, 16 (01) : 70 - 83
  • [5] Graves A, 2012, STUD COMPUT INTELL, V385, P1, DOI [10.1162/neco.1997.9.1.1, 10.1007/978-3-642-24797-2]
  • [6] Hu H., 2017, INT C MACHINE LEARNI, V70, P1568
  • [7] Hutchinson M. L, 2017, Overcoming data scarcity with transfer learning. Arxiv
  • [9] Temporal Fusion Transformers for interpretable multi-horizon time series forecasting
    Lim, Bryan
    Arik, Sercan O.
    Loeff, Nicolas
    Pfister, Tomas
    [J]. INTERNATIONAL JOURNAL OF FORECASTING, 2021, 37 (04) : 1748 - 1764
  • [10] Lundberg SM, 2017, ADV NEUR IN, V30