Mamba Meets Financial Markets: A Graph-Mamba Approach for Stock Price Prediction
Fuente:
arXiv
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866916562238177280 |
|---|---|
| author | Mehrabian, Ali Hoseinzade, Ehsan Mazloum, Mahdi Chen, Xiaohong |
| author_facet | Mehrabian, Ali Hoseinzade, Ehsan Mazloum, Mahdi Chen, Xiaohong |
| contents | Stock markets play an important role in the global economy, where accurate stock price predictions can lead to significant financial returns. While existing transformer-based models have outperformed long short-term memory networks and convolutional neural networks in financial time series prediction, their high computational complexity and memory requirements limit their practicality for real-time trading and long-sequence data processing. To address these challenges, we propose SAMBA, an innovative framework for stock return prediction that builds on the Mamba architecture and integrates graph neural networks. SAMBA achieves near-linear computational complexity by utilizing a bidirectional Mamba block to capture long-term dependencies in historical price data and employing adaptive graph convolution to model dependencies between daily stock features. Our experimental results demonstrate that SAMBA significantly outperforms state-of-the-art baseline models in prediction accuracy, maintaining low computational complexity. The code and datasets are available at github.com/Ali-Meh619/SAMBA. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2410_03707 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Mamba Meets Financial Markets: A Graph-Mamba Approach for Stock Price Prediction Mehrabian, Ali Hoseinzade, Ehsan Mazloum, Mahdi Chen, Xiaohong Computational Finance Machine Learning Stock markets play an important role in the global economy, where accurate stock price predictions can lead to significant financial returns. While existing transformer-based models have outperformed long short-term memory networks and convolutional neural networks in financial time series prediction, their high computational complexity and memory requirements limit their practicality for real-time trading and long-sequence data processing. To address these challenges, we propose SAMBA, an innovative framework for stock return prediction that builds on the Mamba architecture and integrates graph neural networks. SAMBA achieves near-linear computational complexity by utilizing a bidirectional Mamba block to capture long-term dependencies in historical price data and employing adaptive graph convolution to model dependencies between daily stock features. Our experimental results demonstrate that SAMBA significantly outperforms state-of-the-art baseline models in prediction accuracy, maintaining low computational complexity. The code and datasets are available at github.com/Ali-Meh619/SAMBA. |
| title | Mamba Meets Financial Markets: A Graph-Mamba Approach for Stock Price Prediction |
| topic | Computational Finance Machine Learning |
| url | https://arxiv.org/abs/2410.03707 |