Yin X, Zhang X, Li H, Chen Y, He W. An interpretable model for stock price movement prediction based on the hierarchical belief rule base.
Heliyon 2023;
9:e16589. [PMID:
37260876 PMCID:
PMC10227350 DOI:
10.1016/j.heliyon.2023.e16589]
[Citation(s) in RCA: 1] [Impact Index Per Article: 1.0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 03/01/2023] [Revised: 05/18/2023] [Accepted: 05/22/2023] [Indexed: 06/02/2023] Open
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.
Collapse