Aiding Long-Term Investment Decisions with XGBoost Machine Learning Model

Ekaterina Zolotareva
DOI: https://doi.org/10.1007/978-3-030-87897-9_37
2021-01-01
Abstract:The ability to identify stock market trends has obvious advantages for investors. Buying stock on the upward trend (as well as selling it in case of downward movement) results in profit. Accordingly, the start and endpoints of the trend are the optimal points for entering and leaving the market. The research concentrates on recognizing stock market long-term upward and downward trends. The key results are obtained with the use of gradient boosting algorithms, XGBoost in particular. The raw data is represented by time series with basic stock market quotes with periods labelled by experts as «Trend» or «Flat». The features are then obtained via various data transformations, aiming to catch implicit factors resulting in the change of stock direction. Modelling is done in two stages: stage one aims to detect endpoints of tendencies (i.e. “sliding windows”), stage two recognizes the tendency itself inside the window. The research addresses such issues as imbalanced datasets and contradicting labels, as well as the need for specific quality metrics to keep up with practical applicability. The model can be used to design an investment strategy though further research in feature engineering and fine calibration is required. This is the reduced version of the research, full text can be found on arxiv.org (arXiv: 2104.09341).
What problem does this paper attempt to address?