GREY WOLF-OPTIMIZED XGBOOST REGRESSOR FOR STOCK INDEX PREDICTION WITH FINANCIAL FEATURES
Abstract
Accurate stock index prediction is essential for effective investment strategies and economic policymaking. While traditional statistical models often fail to capture the nonlinear dynamics of financial markets, machine learning approaches—particularly Extreme Gradient Boosting (XGBoost)—offer greater flexibility, robustness to overfitting, and computational efficiency. However, the performance of XGBoost strongly depends on hyperparameter tuning, which is difficult to optimize using conventional search methods. To address this, we propose a hybrid framework that integrates XGBoost with Grey Wolf Optimization (GWO) for enhanced hyperparameter selection in stock index forecasting. Using historical data from the Indonesian BBNI stock index (2021–2024) and financial features (price, volume, and temporal), the GWO-optimized XGBoost achieved superior performance, recording the lowest testing MAPE (1.79%), RMSE (108.67), and MAE (84.32). These results surpass classical regressors (Decision Tree, Random Forest, Multilayer Perceptron, Gradient Boosting) by margins of 6–26% and outperform conventional tuning methods (Grid Search, Random Search, Bayesian Optimization) as well as other swarm intelligence approaches (PSO, BA). Moreover, the GWO-based approach reduced error variability and required significantly less optimization time, with the 10-wolf configuration providing the best accuracy–efficiency tradeoff. The scope of this study is limited to a single stock index (BBNI.JK) and financial features, without incorporating macroeconomic indicators, sentiment variables, or cross-market validation. These limitations indicate potential directions for future work to enhance generalizability. Overall, the proposed GWO-XGBoost framework provides a powerful, stable, and time-efficient solution for stock index prediction in volatile market conditions.
Downloads
References
R. K. Yelamanchili, “SHORT-TERM ECONOMIC INDICATORS, STOCK MARKET INDEXES AND INDIAN OIL AND GAS STOCKS RETURNS,” Indian Journal of Finance and Banking, 2020. doi: https://doi.org/10.46281/ijfb.v4i1.454
J. Wang, Q. Cheng, and Y. Dong, “AN XGBOOST-BASED MULTIVARIATE DEEP LEARNING FRAMEWORK FOR STOCK INDEX FUTURES PRICE FORECASTING,” Kybernetes, 2022. doi: https://doi.org/10.1108/K-12-2021-1289
B. Xiu, “BASED ON BAIDU INDEX AND GBDT SHANGHAI INDEX RISE AND FALL FORECAST,” BCP Business & Management, 2022. doi: https://doi.org/10.54691/bcpbm.v34i.3212
C. Liu, J. Wang, D. Xiao, and Q. Liang, “FORECASTING S&P 500 STOCK INDEX USING STATISTICAL LEARNING MODELS,” Open J Stat, 2016. doi: https://doi.org/10.4236/ojs.2016.66086
J. Li, “COMPARISON OF DIFFERENT MACHINE LEARNING APPROACHES FOR FORECASTING STOCK PRICES,” Highlights in Science Engineering and Technology, 2024. doi: https://doi.org/10.54097/2re5n809
C. Fieberg, D. Metko, T. Poddig, and T. Loy, “MACHINE LEARNING TECHNIQUES FOR CROSS-SECTIONAL EQUITY RETURNS’ PREDICTION,” 2022. doi: https://doi.org/10.1007/s00291-022-00693-w.
X. Zhao, “EXPLORING THE PERFORMANCE OF THE CNN-LSTM MODEL IN STOCK PREDICTION,” Highlights in Business Economics and Management, 2024. doi: https://doi.org/10.54097/x9m1cm10
E. Arif, S. Suherman, and A. P. Widodo, “INTEGRATION OF TECHNICAL ANALYSIS AND MACHINE LEARNING TO IMPROVE STOCK PRICE PREDICTION ACCURACY,” 2024. doi: https://doi.org/10.18280/mmep.111106
N. Wu, “ANALYSIS OF ARIMA, LIGHTGBM, XGBOOST, AND LSTM MODELS FOR STOCK PREDICTION,” Applied and Computational Engineering, 2024. doi: https://doi.org/10.54254/2755-2721/54/20241390
X. Wang, “MACHINE LEARNING AND DEEP LEARNING MODELS FOR STOCK PRICE PREDICTION, CASE STUDY: GOOGLE COMPANY,” Applied and Computational Engineering, 2024. doi: https://doi.org/10.54254/2755-2721/49/20241299
Y. E. Gür, “STOCK PRICE FORECASTING USING MACHINE LEARNING AND DEEP LEARNING ALGORITHMS: A CASE STUDY FOR THE AVIATION INDUSTRY,” Fırat Üniversitesi Mühendislik Bilimleri Dergisi, 2024. doi: https://doi.org/10.35234/fumbd.1357613
I. Barkiah and Y. Sari, “OVERCOMING OVERFITTING CHALLENGES WITH HOG FEATURE EXTRACTION AND XGBOOST-BASED CLASSIFICATION FOR CONCRETE CRACK MONITORING,” 2023. doi: https://doi.org/10.24425/ijet.2023.146509
D. N. Gono, H. Napitupulu, and F. Firdaniza, “SILVER PRICE FORECASTING USING EXTREME GRADIENT BOOSTING (XGBOOST) METHOD,” Mathematics, 2023. doi: https://doi.org/10.3390/math11183813
C. Li, H. Li, P. Li, Y. Dang, D. Sun, and D. Guo, “ENHANCING PREDICTIONS OF ACETATE AND ETHANOL PRODUCTION FROM MICROBIAL ELECTROSYNTHESIS USING OPTIMIZED MACHINE LEARNING MODELS,” ACS Sustain Chem Eng, 2024. doi: https://doi.org/10.1021/acssuschemeng.3c08356
S. M. Nzuva, L. Nder, and T. Mwalili, “A NOVEL BAGGING- XGBOOST ENSEMBLE MODEL FOR ATTAINING HIGH ACCURACY AND COMPUTATIONAL EFFICIENCY IN NETWORK INTRUSION DETECTION,” E3s Web of Conferences, 2024. doi: https://doi.org/10.1051/e3sconf/202450101007
A. Mehdary, A. Chehri, A. Jakimi, and R. Saadane, “HYPERPARAMETER OPTIMIZATION WITH GENETIC ALGORITHMS AND XGBOOST: A STEP FORWARD IN SMART GRID FRAUD DETECTION,” Sensors, 2024. doi: https://doi.org/10.3390/s24041230
J. Wang and S. Zhou, “CS-GA-XGBOOST-BASED MODEL FOR A RADIO-FREQUENCY POWER AMPLIFIER UNDER DIFFERENT TEMPERATURES,” Micromachines (Basel), 2023. doi: https://doi.org/10.3390/mi14091673
B. Gülsün and M. R. Aydin, “OPTIMIZING THE EXTREME GRADIENT BOOSTING ALGORITHM THROUGH THE USE OF METAHEURISTIC ALGORITHMS IN SALES FORECASTING,” 2024. doi: https://doi.org/10.21203/rs.3.rs-4515150/v1
W. Lin, L. Liu, G. Zhao, and J. Zheng, “DEVELOPING HYBRID DMO-XGBOOST AND DMO-RF MODELS FOR ESTIMATING THE ELASTIC MODULUS OF ROCK,” Mathematics, 2023. doi: https://doi.org/10.3390/math11183886
F. Poernamawatie, I. N. Susipta, and D. Winarno, “SHARIA BANK OF INDONESIA STOCK PRICE PREDICTION USING LONG SHORT-TERM MEMORY,” Journal of Economics Finance and Management Studies, 2024. doi: https://doi.org/10.47191/jefms/v7-i7-94
E. Ismanto and N. Effendi, “AN LSTM-BASED PREDICTION MODEL FOR GRADIENT-DESCENDING OPTIMIZATION IN VIRTUAL LEARNING ENVIRONMENTS,” Computer Science and Information Technologies, 2024. doi: https://doi.org/10.11591/csit.v4i3.pp199-207
E. Domingos, B. Ojeme, and O. Daramola, “EXPERIMENTAL ANALYSIS OF HYPERPARAMETERS FOR DEEP LEARNING-BASED CHURN PREDICTION IN THE BANKING SECTOR,” Computation, 2021. doi: https://doi.org/10.3390/computation9030034
M. Sher, N. Minallah, T. Ahmad, and W. Khan, “HYPERPARAMETERS ANALYSIS OF LONG SHORT-TERM MEMORY ARCHITECTURE FOR CROP CLASSIFICATION,” International Journal of Electrical and Computer Engineering (Ijece), 2023. doi: https://doi.org/10.11591/ijece.v13i4.pp4661-4670
N. Bakhashwain and A. Sagheer, “ONLINE TUNING OF HYPERPARAMETERS IN DEEP LSTM FOR TIME SERIES APPLICATIONS,” International Journal of Intelligent Engineering and Systems, 2021. doi: https://doi.org/10.22266/ijies2021.0228.21
J. Hong and W. Tian, “PREDICTION IN CATALYTIC CRACKING PROCESS BASED ON SWARM INTELLIGENCE ALGORITHM OPTIMIZATION OF LSTM,” Processes, 2023. doi: https://doi.org/10.3390/pr11051454
Y. Sun, X. Wang, and J. Yang, “MODIFIED PARTICLE SWARM OPTIMIZATION WITH ATTENTION-BASED LSTM FOR WIND POWER PREDICTION,” Energies (Basel), 2022. doi: https://doi.org/10.3390/en15124334
X. Liu, Q. Shi, Z. Liu, and Y. Jia, “USING LSTM NEURAL NETWORK BASED ON IMPROVED PSO AND ATTENTION MECHANISM FOR PREDICTING THE EFFLUENT COD IN A WASTEWATER TREATMENT PLANT,” Ieee Access, 2021. doi: https://doi.org/10.1109/ACCESS.2021.3123225
G. Kumar, U. P. Singh, and S. Jain, “AN ADAPTIVE PARTICLE SWARM OPTIMIZATION-BASED HYBRID LONG SHORT-TERM MEMORY MODEL FOR STOCK PRICE TIME SERIES FORECASTING,” Soft comput, 2022. doi: https://doi.org/10.1007/s00500-022-07451-8
Y. Y. Ji, A. W. Liew, and L. Yang, “A NOVEL IMPROVED PARTICLE SWARM OPTIMIZATION WITH LONG-SHORT TERM MEMORY HYBRID MODEL FOR STOCK INDICES FORECAST,” Ieee Access, 2021. doi: https://doi.org/10.1109/ACCESS.2021.3056713
S. Mirjalili, S. M. Mirjalili, and A. Lewis, “GREY WOLF OPTIMIZER,” Advances in Engineering Software, 2014. doi: https://doi.org/10.1016/j.advengsoft.2013.12.007
F. Xie, G. Liang, and Y. Chien, “HIGHLY ROBUST ADAPTIVE SLIDING MODE TRAJECTORY TRACKING CONTROL OF AUTONOMOUS VEHICLES,” Sensors, 2023. doi: https://doi.org/10.3390/s23073454
M. Y. Silaa, Ó. Barambones, A. Bencherif, and A. Rahmani, “A NEW MPPT-BASED EXTENDED GREY WOLF OPTIMIZER FOR STAND-ALONE PV SYSTEM: A PERFORMANCE EVALUATION VERSUS FOUR SMART MPPT TECHNIQUES IN DIVERSE SCENARIOS,” Inventions, 2023. doi: https://doi.org/10.3390/inventions8060142
H. Zhang, J. Yan, and L. Wang, “HYBRID TABU-GREY WOLF OPTIMIZER ALGORITHM FOR ENHANCING FRESH COLD-CHAIN LOGISTICS DISTRIBUTION,” PLoS One, 2024. doi: https://doi.org/10.1371/journal.pone.0306166
F. Senel, “A HYPERPARAMETER OPTIMIZATION FOR GALAXY CLASSIFICATION,” Computers Materials & Continua, 2023. doi: https://doi.org/10.32604/cmc.2023.033155
Q. Zhu, Y. Li, and Z. Zhang, “SWARM ROBOTS SEARCH FOR MULTIPLE TARGETS BASED ON HISTORICAL OPTIMAL WEIGHTING GREY WOLF OPTIMIZATION,” Mathematics, 2023. doi: https://doi.org/10.3390/math11122630
Y. Dong, Z. Ma, J. Ma, S. Li, B. Ding, and J. Zhang, “OPTIMIZATION OF PARAMETERS OF FUZZY PID CONTROLLER USING GREY WOLF ALGORITHM,” 2024. doi: https://doi.org/10.1117/12.3049213
L. Xu, F. Geng, R.-B. Hu, and R.-B. Wang, “BINARY GANNET OPTIMIZATION ALGORITHM FOR FEATURE SELECTION USING TIME-VARYING TRANSFER FUNCTION,” 2023. doi: https://doi.org/10.21203/rs.3.rs-3111122/v1
X. J. Gu and X. Shi, “INTEGRATED DIAGNOSIS OPTIMIZATION DESIGN OF THE ELECTRONIC EQUIPMENT BASED ON SPATIAL MAPPING,” Sci Prog, 2024. doi: https://doi.org/10.1177/00368504241285770
H. H. Htun, M. Biehl, and N. Petkov, “SURVEY OF FEATURE SELECTION AND EXTRACTION TECHNIQUES FOR STOCK MARKET PREDICTION,” 2023. doi: https://doi.org/10.1186/s40854-022-00441-7
M. Yang, “PREDICTING THE DIRECTION OF STOCK PRICE MOVEMENT WITH MACHINE LEARNING ALGORITHMS,” 2023. doi: https://doi.org/10.54254/2754-1169/52/20230758
P.-G. L. P.-G. Lin, Q.-T. L. P.-G. Lin, J.-Q. Z. Q.-T. Li, J.-H. W. J.-Q. Zhou, M.-W. J. J.-H. Wang, and C. Z. M.-W. Jian, “FINANCIAL FORECASTING METHOD FOR GENERATIVE ADVERSARIAL NETWORKS BASED ON MULTI-MODEL FUSION,” 2023. doi: https://doi.org/10.53106/199115992023023401010
J. K. Chiang and R. Chi, “PREDICTING STOCK PRICES BASED ON PRICE/VOLUME WITH DEEP LEARNING AND SYSTEM ENGINEERING AGGREGATE WITH DYNAMIC BEHAVIORS AND TRADING SIGNALS,” 2023. doi: https://doi.org/10.20944/preprints202308.1979.v1
M. Li, “THE IMPACT OF TRADING VOLUME ON STOCK PRICE VOLATILITY,” 2024. doi: https://doi.org/10.54097/wasnyj47.
Z.-L. Wang, “RESEARCH ON THE AMPLIFICATION EFFECT OF TRADING VOLUME ON MISPRICING IN THE CHINESE A-SHARE MARKET,” 2024. doi: https://doi.org/10.54097/571jf356
M. Sun, J. Wang, Q. Li, J. Zhou, C. Cui, and M. Jian, “STOCK INDEX TIME SERIES PREDICTION BASED ON ENSEMBLE LEARNING MODEL,” 2023. doi: https://doi.org/10.3233/JCM-226523.
Copyright (c) 2026 Syaiful Anam, Mohd Razif Shamsuddin, Elyas Kustiawan, Dwi Mifta Mahanani, Feby Indriana Yusuf
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Authors who publish with this Journal agree to the following terms:
- Author retain copyright and grant the journal right of first publication with the work simultaneously licensed under a creative commons attribution license that allow others to share the work within an acknowledgement of the work’s authorship and initial publication of this journal.
- Authors are able to enter into separate, additional contractual arrangement for the non-exclusive distribution of the journal’s published version of the work (e.g. acknowledgement of its initial publication in this journal).
- Authors are permitted and encouraged to post their work online (e.g. in institutional repositories or on their websites) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published works.
1.gif)
