Welcome to Francis Academic Press

Academic Journal of Computing & Information Science, 2025, 8(5); doi: 10.25236/AJCIS.2025.080508.

Predictive Model Construction Based on SSA-XGBoost Algorithm

Author(s)

Junshuo Liu1, Zhenyu Zhao2, Zixuan Wu3

Corresponding Author:
Junshuo Liu
Affiliation(s)

1School of Emergency Management and Safety Engineering, North China University of Science and Technology, Tangshan, China

2School of Artificial Intelligence, North China University of Science and Technology, Tangshan, China

3School of Mining Engineering, North China University of Science and Technology, Tangshan, China

Abstract

In this paper, we propose an intelligent prediction model based on SSA-XGBoost, focusing on the synergistic optimization mechanism of Sparrow Search Algorithm (SSA) and Extreme Gradient Boosting Regression Tree (XGBoost). First, the input data are preprocessed in a standardized way to eliminate the feature scale differences, and the 70%-30% ratio is used to divide the training set and test set. Second, the initial model of XGBoost is constructed, the loss function is optimized by second-order Taylor expansion, and the regularization term is introduced to control the model complexity. Further, the SSA algorithm is used to dynamically optimize the key hyperparameters, including the maximum tree depth, the minimum number of leaf node samples and the number of iterations, and the fitness function is used to guide the parameter search to improve the model generalization ability. The experimental results show that the optimized SSA-XGBoost model significantly outperforms the benchmark model in terms of MAE, RMSE and R², in which the MAE is reduced by 20.44% and the R² is improved to 0.9216, which verifies its superiority in nonlinear high-dimensional data prediction. The model provides an efficient solution for accurate prediction of complex systems by combining adaptive parameter optimization and integrated learning.

Keywords

SSA, XGBoost, Predictive Evaluation Metrics, Spontaneous Combustion of Coal

Cite This Paper

Junshuo Liu, Zhenyu Zhao, Zixuan Wu. Predictive Model Construction Based on SSA-XGBoost Algorithm. Academic Journal of Computing & Information Science (2025), Vol. 8, Issue 5: 72-78. https://doi.org/10.25236/AJCIS.2025.080508.

References

[1] Tan Bo, Shao Zhuangzhuang, Guo Yan, Zhao Tong, Zhu Hongqing, Li Chuanxing. Early warning of coal spontaneous combustion based on correlation analysis of indicator gases[J]. China Journal of Safety Science, 2021, 31 (02): 33-39.

[2] Jiang Peng. Research on the prediction model of coal spontaneous combustion temperature based on machine learning [D]. Xi'an: Xi'an University of Science and Technology,2020.

[3] Chao Jiangkun, Liu Shuang, Hu Daimin, Han Xuefeng, Yu Minggao, Pan Rongkun. Early prediction of coal spontaneous combustion in deep hollow zone[J]. Journal of Henan University of Science and Technology (Natural Science Edition), 2024, 43 (03): 34-41.

[4] Yang Xiaodong, Li Jianwei. Fire extinguishing technology during stoping of mining in a shallow buried high gas autogenous coal seam[J]. Coal Science and Technology, 2021, 49 (S2): 179-182.

[5] Lei Changkui, Jiang Lijuan, Deng Cubao, Deng Jun, Ma Li, Wang Weifeng, Zhang Yonggan. Gray correlation analysis and prediction of spontaneous combustion limiting parameters of coal in the hollow zone[J]. Coal Mine Safety, 2022, 53 (09): 113-121.

[6] Wang Minhua, Niu Xian. Predictive model of spontaneous combustion of coal remains in the mining zone based on data reconstruction enhancement[J]. Coal Mine Safety, 2022, 53(9):86-93.

[7] Liu Bao, Mu Kun, Ye Fei, et al. Coal spontaneous combustion prediction method based on correlation vector machine[J]. Industrial and mining automation, 2020,46(09):104-108.

[8] Wen Hu, Zhao Xiangtao, Wang Weifeng, et al. Characterization of indicative gas function models for spontaneous combustion of different coal bodies [J]. Coal Conversion, 2020, 43(1):16-25.

[9] Luo Zhaoxian, Yu Wenpeng, Yu Shunqin. Research on improved BP neural network for predicting spontaneous coal combustion[J]. Coal Technology, 2017, 36(12):144-145.

[10] Ma Z, Qin B, Shi Q, et al. The location analysis and efficient control of hidden coal spontaneous combustion disaster in coal mine goaf: A case study[J].Process Safety and Environmental Protection, 2024, 184:66-78.DOI:10.1016/j.psep.2024.01.054.