Ground motion prediction model for shallow crustal earthquakes in Japan based on XGBoost with Bayesian optimization

Ground motion prediction is an important and complex research subject in earthquake engineering and traditional approaches based on statistical regression have much room for improvement in prediction accuracy. Utilizing the 67,164 ground motion records from KiK-net and K-Net for 777 shallow crustal earthquakes between 1997 and 2019 in Japan, this paper proposes a ground motion prediction model XGBoost-SC based on the machine learning algorithm of eXtreme Gradient Boosting (XGBoost) for Japan. Magnitude, focal depth, hypo-central distance, Vs30, site altitude, and focal mechanism were used as the feature parameters, and XGBoost, Random Forest, and Deep Neural Networks (DNN) algorithms were selected for model training while Bayesian optimization was used to search for optimized hyperparameters to improve the prediction accuracy. XGBoost algorithm was selected for further study based on the comparison of results from the three algorithms. Residual change with magnitude and hypo-central distance, the probability distribution of residuals, residual standard deviation (σ), residual mean squared error (MSE), and Pearson correlation coefficient (R) were used as the evaluation parameters, and a comparison study was performed against the ground motion prediction equation based on traditional approaches. Actual earthquake events were selected to compare the prediction results against the observation records. To further validate and explain the proposed model, SHapley Additive exPlanations (SHAP) analysis was performed to explain the impact of selected feature parameters on the proposed model. The results demonstrate that the proposed XGBoost-SC model has good prediction stability for all periods, and its residual errors are smaller than those of other models. Therefore, the proposed model can better reflect the ground motion attenuation for shallow crustal earthquakes in Japan and can serve as a better model for ground motion prediction in future aseismic design and earthquake disaster mitigation efforts.