journal article·2025-10-01Open access

Graph-embedded time-interval-aware transformer for event-driven groundwater level forecasting

Jian Hua, XiaoYu Hu, WeiYa Ge, Yi Shen, CongXuan Kang

AIP Advances· Volume 15· Issue 10

Abstract

This paper proposes a groundwater level prediction method that integrates time-interval awareness with event-driven modeling, aiming to enhance model performance in non-stationary and abrupt hydrological processes. By incorporating event features into the attention mechanism, the framework effectively captures local mutations in groundwater level sequences, while probabilistic forecasting strengthens robustness against uncertain data. Experimental evaluations on eight monitoring wells from the California Department of Water Resources demonstrate that the proposed approach consistently outperforms multiple baseline models under diverse testing scenarios. Specifically, the method achieves an average reduction of 12.4% in MAE and 10.7% in RMSE, while the R2 metric exceeds 0.92. Even under conditions of high missing rates or perturbed timestamps, the model maintains stable predictive performance. These results confirm that the proposed framework delivers higher accuracy and stronger robustness in groundwater level forecasting under complex conditions, providing effective support for groundwater resource management and early warning applications.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Neural Networks and Applications · Reservoir Engineering and Simulation Methods · Computational Physics and Python Applications

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

Nanjing Center, China Geological Survey 1 , Nanjing 210016,

Jiangsu Water Resources Service Center 2 , Nanjing 210029,

Jiangsu Province Hydrology and Water Resources Investigation Bureau Suzhou Branch 3 , Suzhou 215129,

Funding

National Natural Science Foundation of China · China Geological Survey · Jiangsu Province Water Conservancy Science and Technology Project 2024

References (33)

Show source-provided references
1. Groundwater in the commonwealth of independent states: A bibliometric analysis of scopus-based papers from 1972 to 2023, emphasizing the significance of drainage10.1016/j.gsd.2024.101083 ↗
2. Groundwater contamination and health risk assessment in Indian subcontinent: A geospatial approach10.1016/j.coesh.2024.100555 ↗
3. Unraveling microbial community variation along a salinity gradient and indicative significance to groundwater salinization in the coastal aquifer10.1016/j.jhydrol.2024.131893 ↗
4. New approach into human health risk assessment associated with heavy metals in surface water and groundwater using Monte Carlo method10.1038/s41598-023-50000-y ↗
5. Assessing groundwater quality dynamics in Madhya Pradesh: Chemical contaminants and their temporal patterns10.1016/j.envres.2024.118887 ↗
6. Chemical influence factors and contribution rate in shallow groundwater of Wuxi city10.16788/j.hddz.32-1865/P.2023.12.017 ↗
7. The problems faced by groundwater resources in China and countermeasures suggestion
8. Dynamic characteristics of water resources and water environmental carrying capacity in Wuxi urban area10.16788/j.hddz.32-1865/P.2024.02.009 ↗
9. Deep dive into predictive excellence: Transformer’s impact on groundwater level prediction10.1016/j.jhydrol.2024.131250 ↗
10. Advanced machine vision techniques for groundwater level prediction modeling geospatial and statistical research10.1016/j.asr.2024.11.018 ↗
11. Towards a global spatial machine learning model for seasonal groundwater level predictions in Germany10.5194/hess-29-3405-2025 ↗
12. Predicting groundwater level using traditional and deep machine learning algorithms10.3389/fenvs.2024.1291327 ↗
13. Groundwater level prediction using machine learning and geostatistical interpolation models10.3390/w16192771 ↗
14. Groundwater level estimation using improved transformer model: A case study of the yellow river basin10.3390/w17152318 ↗
15. Daily groundwater level prediction and uncertainty using LSTM coupled with PMI and bootstrap incorporating teleconnection patterns information10.3390/su141811598 ↗
16. Experimental study of time series forecasting methods for groundwater level prediction
17. Forecasting groundwater level by characterizing multiple spatial dependencies of environmental factors using graph-based deep learning10.1029/2024jh000520 ↗
18. Research on a non-stationary groundwater level prediction model based on VMD-itransformer and its application in sustainable water resource management of ecological reserves10.3390/su16219185 ↗
19. G. Woo , C.Liu, D.Sahoo, A.Kumar, and S.Hoi, “ETSformer: Exponential smoothing transformers for time-series forecasting,” arXiv:2202.01381 (2022).
20. Non-stationary transformers: Exploring the stationarity in time series forecasting
21. TS-Fastformer: Fast transformer for time-series forecasting10.1145/3630637 ↗
22. Time series forecasting model based on the adapted transformer neural network and FFT-based features extraction10.3390/s25030652 ↗
23. A closer look at transformers for time series forecasting: Understanding why they work and where they struggle,” in Forty-S
24. Deep learning for time series forecasting: Review and applications in geotechnics and geosciences10.1007/s11831-025-10244-5 ↗
25. Improving the accuracy of groundwater level forecasting by coupling ensemble machine learning model and coronavirus herd immunity optimizer10.1007/s11269-025-04210-w ↗
26. Groundwater level prediction with deep learning methods10.3390/w15173118 ↗
27. W. Ma and J.Lu, “An equivalence of fully connected layer and convolutional layer,” arXiv:1712.01252 (2017).
28. Long short-term memory10.1162/neco.1997.9.8.1735 ↗
29. State-of-the-art in 1d convolutional neural networks: A survey10.1109/access.2024.3433513 ↗
30. FEDformer: Frequency enhanced decomposed transformer for long-term series forecasting
31. Crossformer: Transformer utilizing cross-dimension dependency for multivariate time series forecasting
32. Y. Liu , T.Hu, H.Zhang, H.Wu, S.Wang, L.Ma, and M.Long, “itransformer: Inverted transformers are effective for time series forecasting,” arXiv:2310.06625 (2023).
33. Enhancing energy consumption forecasting for electric vehicle charging stations with time series dense encoder (tide)