Vol.40, No.1, 2022, pp.237-246, doi:10.32604/csse.2022.017685
OPEN ACCESS
ARTICLE
Stock-Price Forecasting Based on XGBoost and LSTM
  • Pham Hoang Vuong1, Trinh Tan Dat1, Tieu Khoi Mai1, Pham Hoang Uyen2, Pham The Bao1,*
1 Information Science Faculty, Sai Gon University, Ho Chi Minh City, 700000, Vietnam
2 University of Economics and Law, Ho Chi Minh City, 700000, Vietnam
* Corresponding Author: Pham The Bao. Email:
(This article belongs to this Special Issue: Data Analytics in Industry 4.0)
Received 07 February 2021; Accepted 04 May 2021; Issue published 26 August 2021
Abstract
Using time-series data analysis for stock-price forecasting (SPF) is complex and challenging because many factors can influence stock prices (e.g., inflation, seasonality, economic policy, societal behaviors). Such factors can be analyzed over time for SPF. Machine learning and deep learning have been shown to obtain better forecasts of stock prices than traditional approaches. This study, therefore, proposed a method to enhance the performance of an SPF system based on advanced machine learning and deep learning approaches. First, we applied extreme gradient boosting as a feature-selection technique to extract important features from high-dimensional time-series data and remove redundant features. Then, we fed selected features into a deep long short-term memory (LSTM) network to forecast stock prices. The deep LSTM network was used to reflect the temporal nature of the input time series and fully exploit future contextual information. The complex structure enables this network to capture more stochasticity within the stock price. The method does not change when applied to stock data or Forex data. Experimental results based on a Forex dataset covering 2008–2018 showed that our approach outperformed the baseline autoregressive integrated moving average approach with regard to mean absolute error, mean squared error, and root-mean-square error.
Keywords
stock-price forecasting; ARIMA; XGBoost; LSTM; deep learning
Cite This Article
P. H. Vuong, T. T. Dat, T. K. Mai, P. H. Uyen and P. The Bao, "Stock-price forecasting based on xgboost and lstm," Computer Systems Science and Engineering, vol. 40, no.1, pp. 237–246, 2022.
Citations
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.