AI-Driven Stock Market Prediction Model Shows Promising Results
Researchers at the University of Engineering and Technology have developed an AI-driven stock market prediction model that incorporates economic factors and demonstrates significant accuracy in predicting stock market behavior. This model, based on a long short-term memory (LSTM) model, outperforms traditional ARIMA and machine learning-based random forest models. The study's findings suggest that AI-driven techniques can be adapted to predict stock behavior with precision, providing valuable insights for investors.
Key Takeaways:
- The study constructed an LSTM model that incorporates economic factors influencing stock market direction, including interest rates, inflation, GDP, and unemployment rates.
- The model's performance was evaluated using root mean squared error, mean absolute error, and directional accuracy, with the LSTM model significantly outperforming ARIMA and RF models.
- The LSTM model effectively captures nonlinear patterns in financial time series and shows strong generalization capabilities.
- The study suggests that AI-driven techniques can be adapted to predict stock behavior with precision and provides valuable insights for investors.
- The model's ability to incorporate economic factors directly impacting stock market movements enhances the methodological rigor and applicability of the approach.
- The study proposes a practical AI-based stock market prediction model that can be adapted for other market indices displaying similar behavioral patterns.
Statistics:
- The LSTM model's performance varied across four different cross-validation folds, with the fourth fold appearing to have better generalization performance than the first fold, as both the mean training (0.0450687) and validation (0.01293) values were lower in the fourth fold than in the previous folds.
- The LSTM model significantly outperformed ARIMA and RF models, with a root mean squared error of 0.0450687 and a mean absolute error of 0.01293.
- The study highlights that the LSTM model effectively captures nonlinear patterns in financial time series, demonstrating strong generalization capabilities.