Prediction model for high arch dam stress during the operation period using LightGBM with MSSA and SHAP

Bo Li,Jing Ning,Shengmei Yang,Lixin Zhang
DOI: https://doi.org/10.1016/j.advengsoft.2024.103635
IF: 4.255
2024-04-04
Advances in Engineering Software
Abstract:Dam stress is an important physical quantity to assess high arch dam safety during the operation period. To address the issues of low prediction capability and poor interpretability for high arch dam stress, a prediction model for high arch dam stress during the operation period incorporating the multi-strategy sparrow search algorithm (MSSA), light gradient boosting machine (LightGBM) and Shapley additive explanation (SHAP) is proposed utilizing the measurement data from the dam. First, the initial population diversity and quality of sparrows are improved by using Tent chaotic mapping and refractive inverse learning strategy, and the global search ability of the sparrow search algorithm (SSA) is enhanced by making sparrows jump out of local optimal positions based on adaptive t-distribution mutation. Second, the MSSA conducts optimization analysis on the LightGBM model, effectively solving the issue of the machine learning algorithm converging to local optimal solutions and enabling the determination of optimal hyperparameters. Finally, SHAP is introduced to enhance the interpretability of the MSSA-LightGBM model. The performance measurement indicates that the MSSA, which combines several excellent methods, outperforms SSA and particle swarm algorithm (PSO) in terms of global search ability. The MSSA-LightGBM dam stress model exhibits superior prediction precision compared to the SSA-LightGBM, PSO-LightGBM, LightGBM, and extreme gradient boosting models. SHAP can enhance the interpretability of the MSSA-LightGBM model and identify the crucial features affecting high arch dam stress. This research is highly significant in enhancing the prediction and analysis capabilities of high arch dam stress.
computer science, interdisciplinary applications,engineering, multidisciplinary, software engineering
What problem does this paper attempt to address?