MASTER: Market-Guided Stock Transformer for Stock Price Forecasting (2312.15235v1)
Abstract: Stock price forecasting has remained an extremely challenging problem for many decades due to the high volatility of the stock market. Recent efforts have been devoted to modeling complex stock correlations toward joint stock price forecasting. Existing works share a common neural architecture that learns temporal patterns from individual stock series and then mixes up temporal representations to establish stock correlations. However, they only consider time-aligned stock correlations stemming from all the input stock features, which suffer from two limitations. First, stock correlations often occur momentarily and in a cross-time manner. Second, the feature effectiveness is dynamic with market variation, which affects both the stock sequential patterns and their correlations. To address the limitations, this paper introduces MASTER, a MArkert-Guided Stock TransformER, which models the momentary and cross-time stock correlation and leverages market information for automatic feature selection. MASTER elegantly tackles the complex stock correlation by alternatively engaging in intra-stock and inter-stock information aggregation. Experiments show the superiority of MASTER compared with previous works and visualize the captured realistic stock correlation to provide valuable insights.
- An empirical evaluation of generic convolutional and recurrent networks for sequence modeling. arXiv preprint arXiv:1803.01271.
- Lead–lag detection and network clustering for multivariate time series with an application to the US equity market. Machine Learning, 111(12): 4497–4538.
- Space-time mixing attention for video transformer. Advances in neural information processing systems, 34: 19594–19607.
- Xgboost: A scalable tree boosting system. In Proceedings of the 22nd acm sigkdd international conference on knowledge discovery and data mining, 785–794.
- Learning phrase representations using RNN encoder-decoder for statistical machine translation. arXiv preprint arXiv:1406.1078.
- Spatial-temporal transformer for dynamic scene graph generation. In Proceedings of the IEEE/CVF international conference on computer vision, 16372–16382.
- Hierarchical Multi-Scale Gaussian Transformer for Stock Movement Prediction. In IJCAI, 4640–4646.
- Enhancing stock movement prediction with adversarial training. arXiv preprint arXiv:1810.09936.
- Temporal relational ranking for stock prediction. ACM Transactions on Information Systems (TOIS), 37(2): 1–30.
- Long short-term memory. Supervised sequence labelling with recurrent neural networks, 37–45.
- Efficient integration of multi-order dynamics and internal dynamics in stock movement prediction. In Proceedings of the Sixteenth ACM International Conference on Web Search and Data Mining, 850–858.
- Kamble, R. A. 2017. Short and long term stock trend prediction using decision tree. In 2017 International Conference on Intelligent Computing and Control Systems (ICICCS), 1371–1375. IEEE.
- Reformer: The efficient transformer. arXiv preprint arXiv:2001.04451.
- Memory-Enhanced Transformer for Representation Learning on Temporal Heterogeneous Graphs. Data Science and Engineering, 8(2): 98–111.
- Tensor-based learning for predicting stock movements. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 29.
- Transformer-based capsule network for stock movement prediction. In Proceedings of the first workshop on financial technology and natural language processing, 66–73.
- A Time Series is Worth 64 Words: Long-term Forecasting with Transformers. In The Eleventh International Conference on Learning Representations.
- Decision support system for stock trading using multiple indicators decision tree. In 2014 The 1st International Conference on Information Technology, Computer, and Electrical Engineering, 291–296. IEEE.
- Piccolo, D. 1990. A distance measure for classifying ARIMA models. Journal of time series analysis, 11(2): 153–164.
- Stock selection via spatiotemporal hypergraph attention network: A learning to rank approach. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 35, 497–504.
- Spatiotemporal hypergraph convolution network for stock movement forecasting. In 2020 IEEE International Conference on Data Mining (ICDM), 482–491. IEEE.
- Attention is all you need. Advances in neural information processing systems, 30.
- Graph attention networks. arXiv preprint arXiv:1710.10903.
- Hierarchical Adaptive Temporal-Relational Modeling for Stock Trend Prediction. In IJCAI, 3691–3698.
- Adaptive long-short pattern transformer for stock investment selection. In Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence, 3970–3977.
- Review of graph construction and graph learning in stock price prediction. Procedia Computer Science, 214: 771–778.
- Temporal and Heterogeneous Graph Neural Network for Financial Time Series Prediction. In Proceedings of the 31st ACM International Conference on Information & Knowledge Management, 3584–3593.
- Semantic frames to predict stock price movement. In Proceedings of the 51st annual meeting of the association for computational linguistics, 873–883.
- Spatial-temporal transformer networks for traffic flow forecasting. arXiv preprint arXiv:2001.02908.
- Hist: A graph-based framework for stock trend forecasting via mining concept-oriented shared information. arXiv preprint arXiv:2110.13716.
- Qlib: An ai-oriented quantitative investment platform. arXiv preprint arXiv:2009.11189.
- Accurate multivariate stock movement prediction via data-axis transformer with multi-level contexts. In Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining, 2037–2045.
- Crossformer: Transformer utilizing cross-dimension dependency for multivariate time series forecasting. In The Eleventh International Conference on Learning Representations.
- Tong Li (197 papers)
- Zhaoyang Liu (42 papers)
- Yanyan Shen (54 papers)
- Xue Wang (69 papers)
- Haokun Chen (26 papers)
- Sen Huang (9 papers)