Spatio-temporal prediction of groundwater vulnerability based on CNN-LSTM model with self-attention mechanism: A case study in Hetao Plain, northern China

Located in northern China, the Hetao Plain is an important agro-economic zone and population centre. The deterioration of local groundwater quality has had a serious impact on human health and economic development. Nowadays, the groundwater vulnerability assessment (GVA) has become an essential task to identify the current status and development trend of groundwater quality. In this study, the convolutional neural network (CNN) and long short-term memory (LSTM) are integrated to realize the spatio-temporal prediction of regional groundwater vulnerability by introducing the Self-attention mechanism. The study first evaluates the prediction accuracy of the CNN-LSTM model for groundwater vulnerability compared to other common machine learning models such as Support Vector Machine (SVM), Random Forest (RF), and Extreme Gradient Boosting (XGBoost). The results indicate that the CNN-LSTM model outperforms these models, demonstrating its significance in groundwater vulnerability assessment. The predictions suggest a heightened risk of groundwater vulnerability in the study area in the future. This increase can be attributed to the synergistic impact of global climate anomalies and intensified local human activities. Moreover, the overall groundwater vulnerability risk in the entire region has increased, evident from both the notably high value and standard deviation. This suggests that the spatial variability of groundwater vulnerability in the area is expected to expand in the future due to the sustained progression of climate change and human activities. The model can be optimized for diverse applications across regional environmental assessment, pollution prediction, and risk statistics. This study holds particular significance for ecological protection and groundwater resource management.