Prediction of Earthquake Death Toll Based on Principal Component Analysis, Improved Whale Optimization Algorithm, and Extreme Gradient Boosting

Abstract

Earthquakes, as one of the most destructive natural disasters, often cause significant casualties and severe economic losses. Accurate prediction of earthquake fatalities is of great importance for pre-disaster prevention and mitigation planning, as well as post-disaster emergency response deployment. To address the challenges of small sample sizes, high dimensionality, and strong nonlinearity in earthquake fatality prediction, this paper proposes an integrated modeling approach (PCA-IWOA-XGBoost) combining Principal Component Analysis (PCA), the Improved Whale Optimization Algorithm (IWOA), and Extreme Gradient Boosting (XGBoost). The method first employs PCA to reduce the dimensionality of the influencing factor data, eliminating redundant information and improving modeling efficiency. Subsequently, the IWOA is used to intelligently optimize key hyperparameters of the XGBoost model, enhancing the prediction accuracy and stability. Using 42 major earthquake events in China from 1970 to 2025 as a case study, covering regions including the west (e.g., Tonghai in Yunnan, Wenchuan, Jiuzhaigou), central (e.g., Lushan in Sichuan, Ya’an), east (e.g., Tangshan, Yingkou), north (e.g., Baotou in Inner Mongolia, Helinger), northwest (e.g., Jiashi in Xinjiang, Wushi, Yongdeng in Gansu), and southwest (e.g., Lancang in Yunnan, Lijiang, Ludian), the empirical results showed that the PCA-IWOA-XGBoost model achieved an average test set accuracy of 97.0%, a coefficient of determination (R2) of 0.996, a root mean square error (RMSE) and mean absolute error (MAE) reduced to 4.410 and 3.430, respectively, and a residual prediction deviation (RPD) of 21.090. These results significantly outperformed the baseline XGBoost, PCA-XGBoost, and IWOA-XGBoost models, providing improved technical support for earthquake disaster risk assessment and emergency response.