An explainable deep learning approach for stock market trend prediction
Given the intricate nature of stock forecasting as well as the inherent risks and uncertainties, analysis of market trends is necessary to capitalize on optimal investment opportunities for profit maximization and timely disinvestment for loss minimization. In this work, we propose a deep learning model for predicting five distinct stock market trends: upward, downward, double top, rounded bottom, and rounded top. The proposed model surpasses common benchmarks, including support vector machine, random forest, and logistic regression, achieving an average accuracy of 94.9%, compared to 85.7% for random forest, 60.07% for support vector machine, and 52.45% for logistic regression. Furthermore the proposed model excels in F1-score, with a 94.85% performance, compared to 77.95% for random forest, 21.02% for support vector machine and 46.23% for logistic regression, across four real world diverse datasets. Additionally, we employ explainable AI (XAI) techniques, SHAP and LIME, to enhance interpretability, enabling stakeholders to understand the key factors driving predictions. The SHAP analysis reveals the top 10 most important/influential features, enabling feature reduction while maintaining performance. Interestingly, while accuracy slightly decreases with top 10 features, precision, recall, and F1-score improve, suggesting a trade-off between comprehensiveness and performance. These results demonstrate the potential for practical application in financial decision-making, providing a balance between interpretability and predictive power that can support investors in risk management and strategic planning.
Authors
- Iftikhar Ahmed (ORCID: https://orcid.org/0000-0001-8852-6936)
- Malika Bendechache (ORCID: https://orcid.org/0000-0003-0069-1860)
- Khwaja Naveed (ORCID: https://orcid.org/0000-0003-4972-6630)
- Dost Muhammad (ORCID: https://orcid.org/0000-0001-9186-6392)
Institutions
- Ollscoil na Gaillimhe – University of Galway (IE)
- University of Applied Sciences Potsdam (DE)
- Software (Germany) (DE)
- University of Gothenburg (SE)
Publication Details
- Journal
- Heliyon
- Published
- 2024-11-01
- DOI
- https://doi.org/10.1016/j.heliyon.2024.e40095
- Citations
- 37
- Primary Topic
- Stock Market Forecasting Methods
- Type
- article
- Field-Weighted Citation Impact
- 18.15
Funders
- Science Foundation Ireland
- University of Gujrat
- College of Science and Engineering, University of Minnesota
- ADAPT - Centre for Digital Content Technology