A novel hierarchical feature selection with local shuffling and models reweighting for stock price forecasting

被引:7
作者
An, Zhiyon [1 ]
Wu, Yafei [1 ]
Hao, Fangjing [1 ]
Chen, Yuer [1 ]
He, Xuerui [2 ]
机构
[1] Shandong Technol & Business Univ, Sch Comp Sci & Technol, Yantai, Peoples R China
[2] Shandong Technol & Business Univ, Sch Law, Yantai, Peoples R China
关键词
Models reweighting; LSTM; Feature selection; Stock price forecasting; Local shuffling;
D O I
10.1016/j.eswa.2024.123482
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Stock price forecasting is a challenging task due to the complexity of financial markets and the high volatility of stocks. Because of the strong nonlinear representation ability of neural network models, such as longshort memory network (LSTM) and deep learning, they are used extensively in recent years for stock price forecasting. However, due to the volatility of financial data, neural network models often suffer from overfitting or instability problems. In addition, quantitative trading with feature mining tools can generate a growing number of features for financial data. Therefore, selecting effective features for financial data is an urgent problem. To address these problems, we propose a novel hierarchical feature selection with local shuffling (HFSLS) and models reweighting (MR) based on LSTM, named HFSLSMR-LSTM, for stock price forecasting. Specifically, for each layer, local shuffling perturbs each feature to re-predict, and its predicted value is compared with the true value to calculate the feature importance, and the important features are selected and returned to the next layer. Besides, a proximity reweighting scheme is presented to adjust the weight for each layer model that learns from hierarchical features. The HFSLSMR-LSTM model is still effective for financial data with multiple features and frequent fluctuations. Experimental results on stock index dataset and Dow Jones Industrial Average dataset demonstrate that the HFSLSMR-LSTM outperforms Informer, DoubleEnsemble, LSTM, GRU, BI-GRU and BI-LSTM on the metrics MSE, RMSE, MAE, MAPE and R2.
引用
收藏
页数:18
相关论文
共 39 条
[1]   Comparison of ARIMA and Artificial Neural Networks Models for Stock Price Prediction [J].
Adebiyi, Ayodele Ariyo ;
Adewumi, Aderemi Oluyinka ;
Ayo, Charles Korede .
JOURNAL OF APPLIED MATHEMATICS, 2014,
[2]   How to Handle Data Imbalance and Feature Selection Problems in CNN-Based Stock Price Forecasting [J].
Aksehir, Zinnet Duygu ;
Kilic, Erdal .
IEEE ACCESS, 2022, 10 :31297-31305
[3]   Predicting stock market index using LSTM [J].
Bhandari, Hum Nath ;
Rimal, Binod ;
Pokhrel, Nawa Raj ;
Rimal, Ramchandra ;
Dahal, Keshab R. ;
Khatri, Rajendra K. C. .
MACHINE LEARNING WITH APPLICATIONS, 2022, 9
[4]  
Bin Islam Sadman, 2021, 2021 IEEE 12th Annual Information Technology, Electronics and Mobile Communication Conference (IEMCON), P0479, DOI 10.1109/IEMCON53756.2021.9623206
[5]   A survey on feature selection methods [J].
Chandrashekar, Girish ;
Sahin, Ferat .
COMPUTERS & ELECTRICAL ENGINEERING, 2014, 40 (01) :16-28
[6]   StockGAN: robust stock price prediction using GAN algorithm [J].
Diqi M. ;
Hiswati M.E. ;
Nur A.S. .
International Journal of Information Technology, 2022, 14 (5) :2309-2315
[7]   Pattern recognition in financial surveillance with the ARMA-GARCH time series model using support vector machine [J].
Doroudyan, Mohammad Hadi ;
Niaki, Seyed Taghi Akhavan .
EXPERT SYSTEMS WITH APPLICATIONS, 2021, 182
[8]  
Goyal Vibhu, 2022, 2022 8th International Conference on Advanced Computing and Communication Systems (ICACCS), P918, DOI 10.1109/ICACCS54159.2022.9785150
[9]   Stock price prediction with optimized deep LSTM network with artificial rabbits optimization algorithm [J].
Gulmez, Burak .
EXPERT SYSTEMS WITH APPLICATIONS, 2023, 227
[10]   StockNet-GRU based stock index prediction [J].
Gupta, Umang ;
Bhattacharjee, Vandana ;
Bishnu, Partha Sarathi .
EXPERT SYSTEMS WITH APPLICATIONS, 2022, 207