An interpretable intuitionistic fuzzy inference model for stock prediction

Abstract

Stock price prediction and modeling demonstrate high economic value in the financial market. Due to the non-linearity and volatility of stock prices and the unique nature of financial transactions, it is essential for the prediction method to ensure high prediction performance and interpretability. However, existing methods fail to achieve both the two goals simultaneously. To fill this gap, this paper presents an interpretable intuitionistic fuzzy inference model, dubbed as IIFI. While retaining the prediction accuracy, the interpretable module in IIFI can automatically calculate the feature contribution based on the intuitionistic fuzzy set, which provides high interpretability of the model. Also, most of the existing training algorithms, such as LightGBM, XGBoost, DNN, Stacking, etc, can be embedded in the inference module of our proposed model and achieve better prediction results. The back-test experiment on China’s A-share market shows that IIFI achieves superior performance — the stock profitability can be increased by more than 20% over the baseline methods. Meanwhile, interpretable results show that IIFI can effectively distinguish between important and redundant features via rating corresponding scores to each feature. As a byproduct of our interpretable methods, the scores over features can be used to further optimize the investment strategy.