Optimizing Stock Trend Prediction in the Chinese Market: A Comparative Study of Machine Learning Models with Bayesian Hyperparameter Tuning
Yiming Xiang ()
Additional contact information
Yiming Xiang: The Chinese University of Hong Kong, Financial Engineering
A chapter in Proceedings of the International Workshop on Navigating the Digital Business Frontier for Sustainable Financial Innovation (ICDEBA 2024), 2025, pp 599-614 from Springer
Abstract:
Abstract Predicting stock market trends has become increasingly vital in today’s volatile global economy. This paper addresses the challenge of accurate stock market prediction, particularly within the Chinese stock market, which has been subject to unique regulatory and economic shifts. This paper explores the application of five machine learning models to predict stock trends, comparing their performances and enhancing accuracy through hyperparameter tuning. This paper utilized a dataset from the CSI 300 Index, applying various feature engineering techniques and standardizing inputs. After training and backtesting the models, Bayesian optimization is employed to fine-tune the top-performing models: LightGBM, XGBoost, and GRU. This optimization process focuses on improving the models’ annualized returns, Sharpe ratios, and drawdowns. Each model’s predictions were tested on a backtest dataset from 2024, and key performance metrics were recorded. The initial results indicated that LightGBM, XGBoost, and GRU outperformed the other models, with LightGBM achieving an annualized return of 16.44%, XGBoost 11.44%, and GRU 16.11%. After Bayesian optimization, LightGBM improved to 20.97%, XGBoost to 17.20%, and GRU reached 27.22%, though with a higher drawdown. These results demonstrate that GRU’s risk-adjusted performance can offer significant returns, while LightGBM strikes a balance between risk and return. Future work could extend the dataset beyond the CSI 300 Index for better generalizability. While Bayesian optimization improved gradient boosting models, further tuning of neural networks and exploring advanced architectures like Transformers may enhance performance. Integrating real-world factors, such as transaction costs and macroeconomic indicators may strengthen their practical applications.
Keywords: Machine Learning; Stock Trends Prediction; Bayesian Optimization (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:
There are no downloads for this item, see the EconPapers FAQ for hints about obtaining it.
Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.
Export reference: BibTeX
RIS (EndNote, ProCite, RefMan)
HTML/Text
Persistent link: https://EconPapers.repec.org/RePEc:spr:advbcp:978-94-6463-652-9_63
Ordering information: This item can be ordered from
http://www.springer.com/9789464636529
DOI: 10.2991/978-94-6463-652-9_63
Access Statistics for this chapter
More chapters in Advances in Economics, Business and Management Research from Springer
Bibliographic data for series maintained by Sonal Shukla () and Springer Nature Abstracting and Indexing ().