论文标题

基于注意的CNN-LSTM和XGBOOST HYBRID模型用于库存预测

Attention-based CNN-LSTM and XGBoost hybrid model for stock prediction

论文作者

Shi, Zhuangwei, Hu, Yang, Mo, Guangliang, Wu, Jian

论文摘要

股市在经济发展中起着重要作用。由于股票市场的复杂性波动,对股票价格变化的研究和预测可以避免投资者的风险。传统的时间序列模型Arima无法描述非线性,也无法在库存预测中获得令人满意的结果。由于神经网络具有强大的非线性泛化能力,因此本文提出了一个基于注意力的CNN-LSTM和XGBoost Hybrid模型,以预测股票价格。本文构建的模型将时间序列模型,具有注意机制的卷积神经网络,长期的短期存储网络和XGBoost回归器在非线性关系中集成在一起,并提高了预测准确性。该模型可以在多个时期内完全挖掘股票市场的历史信息。库存数据首先通过Arima进行预处理。然后,采用了预训练框架中形成的深度学习体系结构。训练前模型是基于序列到序列框架的基于注意力的CNN-LSTM模型。该模型首先使用卷积来提取原始库存数据的深度特征,然后使用长期的短期内存网络来挖掘长期时间序列的特征。最后,采用XGBoost模型进行微调。结果表明,混合模型更有效,预测准确性相对较高,这可以帮助投资者或机构做出决策并实现扩大回报并避免风险的目的。源代码可在https://github.com/zshicode/astention-clx-stock-prediction上找到。

Stock market plays an important role in the economic development. Due to the complex volatility of the stock market, the research and prediction on the change of the stock price, can avoid the risk for the investors. The traditional time series model ARIMA can not describe the nonlinearity, and can not achieve satisfactory results in the stock prediction. As neural networks are with strong nonlinear generalization ability, this paper proposes an attention-based CNN-LSTM and XGBoost hybrid model to predict the stock price. The model constructed in this paper integrates the time series model, the Convolutional Neural Networks with Attention mechanism, the Long Short-Term Memory network, and XGBoost regressor in a non-linear relationship, and improves the prediction accuracy. The model can fully mine the historical information of the stock market in multiple periods. The stock data is first preprocessed through ARIMA. Then, the deep learning architecture formed in pretraining-finetuning framework is adopted. The pre-training model is the Attention-based CNN-LSTM model based on sequence-to-sequence framework. The model first uses convolution to extract the deep features of the original stock data, and then uses the Long Short-Term Memory networks to mine the long-term time series features. Finally, the XGBoost model is adopted for fine-tuning. The results show that the hybrid model is more effective and the prediction accuracy is relatively high, which can help investors or institutions to make decisions and achieve the purpose of expanding return and avoiding risk. Source code is available at https://github.com/zshicode/Attention-CLX-stock-prediction.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源