An interpretable model for stock price movement prediction based on the hierarchical belief rule base

Xiuxian Yin,Xin Zhang,Hongyu Li,Yujia Chen,Wei He
DOI: https://doi.org/10.1016/j.heliyon.2023.e16589
IF: 3.776
2023-05-26
Heliyon
Abstract:Stock price movement prediction is the basis for decision-making to maintain the stability and security of stock markets. It is important to generate predictions in an interpretable manner. The Belief Rule Base (BRB) has certain interpretability based on IF-THEN rule semantics. However, the interpretability of BRB in the whole process of stock prediction modeling may be weakened or lost. Therefore, this paper proposes an interpretable model for stock price movement prediction based on the hierarchical Belief Rule Base (HBRB-I). The interpretability of the model is considered, and several criteria are constructed based on the BRB expert system. First, the hierarchical structure of BRB is constructed to ensure the interpretability of the initial modeling. Second, the interpretability of the inference process is ensured by the Evidential Reasoning (ER) method as a transparent inference engine. Third, a new Projection Covariance Matrix Adaptive Evolution Strategy (P-CMA-ES) algorithm with interpretability criteria is designed to ensure the interpretability of the optimization process. The final mean squared error value of 1.69E-04 was obtained with similar accuracy to the initial BRB and enhanced in terms of interpretability. This paper is for short-term stock forecasting, and more data will be collected in the future to update the rules to enhance the forecasting capability of the rule base.
What problem does this paper attempt to address?