Mamba Meets Financial Markets: A Graph-Mamba Approach for Stock Price Prediction

ArXiv ID: 2410.03707 “View on arXiv”

Authors: Unknown

Abstract

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.

Keywords: SAMBA, Mamba architecture, graph neural networks, stock return prediction, computational complexity, Equities (Stocks)

Complexity vs Empirical Score

• Math Complexity: 7.0/10
• Empirical Rigor: 8.0/10
• Quadrant: Holy Grail
• Why: The paper uses advanced mathematics including state space models, discretization, and bidirectional Mamba blocks, while also providing specific empirical results with comparisons to state-of-the-art baselines and publicly available code and datasets for reproducibility.

  flowchart TD
    A["Research Goal:<br>Accurate Stock Price Prediction<br>Low Computational Complexity"] --> B["Input Data:<br>Historical Stock Price Data"]
    B --> C["Methodology 1:<br>Bidirectional Mamba Block<br>for Long-Term Dependencies"]
    B --> D["Methodology 2:<br>Adaptive Graph Convolution<br>for Feature Dependencies"]
    C & D --> E["Computational Process:<br>Integrated Graph-Mamba Framework"]
    E --> F["Key Finding 1:<br>SOTA Prediction Accuracy"]
    E --> G["Key Finding 2:<br>Near-Linear Computational Complexity"]