In collaboration with Iranian Watershed Management Association

Document Type : Research Paper

Authors

1 M.Sc. Graduate in Watershed Science and Engineering, Department of Watershed Engineering, Faculty of Natural Resources, Tarbiat Modares University, Noor, Iran

2 Associate Professor, Department of Watershed Science and Engineering, Faculty of Natural Resources, Tarbiat Modares University, Noor, Iran

Abstract

Introduction
Groundwater level modeling has always been a major challenge in groundwater resource management due to its complex, nonlinear, and spatiotemporally dependent nature. The main objective of this study was to develop and evaluate a distributed deep learning-based modeling approach for the simultaneous prediction of groundwater levels in a network of piezometric wells within the Kouchesfahan aquifer located in the Sefidroud watershed. In this framework, each well was considered as a node in a graph network, and the model predicted groundwater levels for each well at the next monthly time step (t+1). By integrating graph neural networks with convolutional structures and long short-term memory architectures, the study aimed to identify the spatial and temporal dependencies governing aquifer behavior and improve prediction accuracy compared to conventional methods.
 
Materials and methods
In this study, four graph-based deep learning architectures, including Graph Neural Network (GNN), Graph Attention Network (GAT), Graph Convolutional Network (GCN), and the hybrid GCN–LSTM model, were employed. Here, GNN refers to the baseline GCN-based graph neural architecture using the standard graph convolution operator. Each piezometric well was represented as a graph node, and the adjacency matrix was constructed using inverse-distance weighting. Input data included hydroclimatic variables, geological information, pumping characteristics, and spatial distances related to hydrogeological factors. Model performance was evaluated using R², RMSE, MAE, NSE, and KGE indices. In the proposed model, GCN layers were first applied to the spatial graph of wells to extract spatial features and hydraulic dependencies. The resulting hidden representations were then transferred to the LSTM network to model temporal dependencies and groundwater level dynamics.
 
Results and discussion
The results indicated that while the GNN model could partially reconstruct the general trends of groundwater level changes, it lacked accuracy in representing extreme behaviors and temporal dependencies. The GAT model showed a slight improvement over GNN but remained limited in extracting deep temporal patterns. In contrast, the GCN model demonstrated better performance in identifying spatial dependencies, leading to a significant improvement in evaluation metrics. The best performance was achieved by the GCN-LSTM model, which effectively represented both spatial and temporal features simultaneously, showing the highest overlap with observed data. This model reached an R2 of 0.896, RMSE of 0.336, and MAE of 0.269, indicating its high accuracy in predicting groundwater levels.
 
Conclusion
Hybrid graph-sequential architectures, especially GCN-LSTM, are highly effective for modeling complex aquifer hydrodynamics. This model predicted groundwater levels with high accuracy and outperformed purely graph-based models. Therefore, it is recommended for operational groundwater prediction and sustainable water management, serving as a novel framework to support better decision-making in water resources.

Keywords

Aslam, R.A., Shrestha, S., Pandey, V.P., 2018. Groundwater vulnerability to climate change: A review of the assessment methodology. Sci. Total Environ. 612, 853–875. https://doi.org/10.1016/j.scitotenv.2017.08.237.
Bai, T., Tahmasebi, P., 2021. Sequential gaussian simulation for geosystems modeling: A machine learning approach. Geosci. Front. 13(1), 101258. https://doi.org/10.1016/j.gsf.2021.101258.
Bai, T., Tahmasebi, P., 2023. Graph neural network for groundwater level forecasting. J. Hydrol. 616, 128792. https://doi.org/10.1016/j.jhydrol.2022.128792.
Chamanpira, G., Zehtabian, G., Ahmadi, H., Melkian, A., 2014. Investigating the impact of drought on groundwater resources for optimal exploitation management: case study of Alishtar plain. Water. Engin. Manage. 6(1), 10–20. https://doi.org/10.22092/ijwmse.2014.101733.
Chen, J., Dai, Z., Yang, Z., Pan, Y., Zhang, X., Wu, J., Reza Soltanian, M., 2021. An improved tandem neural network architecture for inverse modeling of multicomponent reactive transport in porous media. Water Resour. Res. 57(12), e2021WR030595. https://doi.org/10.1029/2021WR030595.
Chen, L., Zhang, D., Xu, J., Zhou, Z., Jin, J., Luan, J., Wulamu, A., 2025. Enhancing the accuracy of groundwater level prediction at different scales using spatio-temporal graph convolutional model. Earth Sci. Inform. 18(2), 250. https://doi.org/10.1007/s12145-024-01502-3.
Dudley, R.W., Hodgkins, G.A., Nielsen, M.G., Qi, S.L., 2018. Estimating historical groundwater levels based on relations with hydrologic and meteorological variables in the U.S. glacial aquifer system. J. Hydrol. 562, 530–543. https://doi.org/10.1016/j.jhydrol.2018.05.019.
Famiglietti, J.S., 2014. The global groundwater crisis. Nat. Clim. Change 4(11), 945–948. https://doi.org/10.1038/nclimate2425.
He, K., Zhang, X., Ren, S., Sun, J., 2016. Deep residual learning for image recognition. Proc. IEEE Conf. Comput. Vis. Pattern Recognit. (CVPR), 770–778. https://doi.org/10.1109/CVPR.2016.90.
Heidari, N., Hedegaard, L., Iosifidis, A., 2022. Graph convolutional networks. In: Deep Learning for Robot Perception and Cognition 71–99. https://doi.org/10.1016/B978-0-32-385787-1.00009-9.
Homayounpour, B., Goodarzi, M., Zehtabian, G., Motamed-Vaziri, B., Ahmadi, H., 2022. Evaluating the impact of climate change on groundwater fluctuations in Barkhar plain, Isfahan. Water. Engin. Manage. 14(4), 465–480. https://doi.org/10.22092/ijwmse.2022.353543.1880.
Kamrava, S., Sahimi, M., Tahmasebi, P., 2021. Simulating fluid flow in complex porous materials by integrating the governing equations with deep-layered machines. NPJ Comput. Mater. 7(1), 1–9. https://doi.org/10.1038/s41524-021-00598-2.
Khorramdel, G.N., Mohammadi, K., Monem, M.J., 2008. Optimization of observation well network for estimation of groundwater balance using double water table fluctuation method. J. Water Soil 22(2), 358–370 (in Persian).
Kilinc, H.C., Apak, S., Ozkan, F., Yurtsever, A., 2025. AH-GCAN-LSTM: Adaptive hybrid graph convolution attention network with LSTM and genetic optimization for hydrological time series forecasting. Water Resour. Manage. 39(15), 8293–8312. https://doi.org/10.1007/s11269-024-03984-2.
Li, D., Zhao, W., Hu, J., Zhao, S., Liu, S., 2023. A long-term water quality prediction model for marine ranch based on time-graph convolutional neural network. Ecol. Indic. 154, 110782. https://doi.org/10.1016/j.ecolind.2023.110782.
Liang, X.X., Gloaguen, E., Claprood, M., Paradis, D., Lauzon, D., 2025. Graph neural network framework for spatiotemporal groundwater level forecasting. Math. Geosci. 57(6), 1071–1093. https://doi.org/10.1007/s11004-024-10145-1.
Lin, X., Lu, Q., Zhao, P., Chen, L., Tang, J., Guan, D., Broyd, T., 2026. Field-theory inspired physics-informed graph neural network for reliable traffic flow prediction under urban flooding. Reliab. Eng. Syst. Saf. 265, 111487. https://doi.org/10.1016/j.ress.2025.111487.
Long, D., Yang, W., Scanlon, B.R., Zhao, J., Liu, D., Burek, P., Pan, Y., You, L., Wada, Y., 2020. South-to-North water diversion stabilizing Beijing’s groundwater levels. Nat. Commun. 11(1), 1–10. https://doi.org/10.1038/s41467-020-17428-6.
Ma, C., Xu, K., Li, Y., Hao, Y., Sun, H., Gao, S., Wang, X., 2026. Heterogeneous spatiotemporal graph attention network for karst spring discharge prediction: Advancing sustainable groundwater management under climate change. Sustain. 18(2), 933. https://doi.org/10.3390/su18020933.
Marufinia, E., Sharafati, A., Abooghari, H., Hassanzadeh, Y., 2025. Application of machine learning and deep wavelet algorithms for predicting the Korkorsar River flow in Nowshahr. Water. Engin. Manage. 17(4), 424–447. https://doi.org/10.22092/ijwmse.2025.369325.2115.
Minnig, M., Moeck, C., Radny, D., Schirmer, M., 2018. Impact of urbanization on groundwater recharge rates in Dübendorf, Switzerland. J. Hydrol. 563, 1135–1146. https://doi.org/10.1016/j.jhydrol.2017.09.058.
Mirmoshtaghi, M., Amirnejad, R., Khaledian, M.R., 2012. Evaluating of water quality of Sefid-Rood and zoning by water quality index NSFWQI and OWQI. J. Wetl. Ecobiol. 3(9), 23–34 (in Persian).
Nazari, A., Jamshidi, M., Roozbahani, A., Golparvar, B., 2025. Groundwater level forecasting using empirical mode decomposition and wavelet-based long short-term memory (LSTM) neural networks. Groundw. Sustain. Dev. 28, 101397. https://doi.org/10.1016/j.gsd.2024.101397.
Raissi, M., Perdikaris, P., Karniadakis, G.E., 2019. Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations. J. Comput. Phys. 378, 686–707. https://doi.org/10.1016/j.jcp.2018.10.045.
Saatsaz, M., 2012. Groundwater resource assessment of Astaneh-Kouchesfahan plain. PhD Thesis, Universiti Putra Malaysia, Serdang, Malaysia.
Sahoo, S., Russo, T.A., Elliott, J., Foster, I., 2017. Machine learning algorithms for modeling groundwater level changes in agricultural regions of the U.S. Water Resour. Res. 53(5), 3878–3895. https://doi.org/10.1002/2016WR019933.
Shiri, J., Kisi, O., Yoon, H., Lee, K.K., Nazemi, A.H., 2013. Predicting groundwater level fluctuations with meteorological effect implications—A comparative study among soft computing techniques. Comput. Geosci. 56, 32–44. https://doi.org/10.1016/j.cageo.2013.01.007.
Taccari, M.L., Wang, H., Nuttall, J., Chen, X., Jimack, P.K., 2024. Spatial-temporal graph neural networks for groundwater data. Sci. Rep. 14(1), 24564. https://doi.org/10.1038/s41598-024-75892-5.
Tang, Y., Zang, C., Wei, Y., Jiang, M., 2019. Data-driven modeling of groundwater level with least-square support vector machine and spatial-temporal analysis. Geotech. Geol. Eng. 37, 1661–1670. https://doi.org/10.1007/s10706-018-0713-6.
Wang, C., Zhu, Y., Zang, T., Liu, H., Yu, J., 2021. Modeling inter-station relationships with attentive temporal graph convolutional network for air quality prediction. Proc. 14th ACM Int. Conf. Web Search Data Min., 616–624. https://doi.org/10.1145/3437963.3441731.
Wang, L., Jiang, Z., Song, L., Yu, X., Yuan, S., Zhang, B., 2024. A groundwater level spatiotemporal prediction model based on graph convolutional networks with a long short-term memory. J. Hydroinform. 26(11), 2962–2979. https://doi.org/10.2166/hydro.2024.168.
Wang, Q., Zhu, S., 2025. Multi-station joint runoff forecasting using graph neural network coupled with spatial connectivity of hydrological stations. Math. Geosci. 57(4), 759–797. https://doi.org/10.1007/s11004-024-10142-4.
Wu, C., Zhang, X., Wang, W., Lu, C., Zhang, Y., Qin, W., Tick, G.R., Liu, B., Shu, L., 2021. Groundwater level modeling framework by combining the wavelet transform with a long short-term memory data-driven model. Sci. Total Environ. 783, 146948. https://doi.org/10.1016/j.scitotenv.2021.146948.
Wu, Y., Mei, G., Shao, K., Xu, N., Peng, J., 2025. Forecasting groundwater level by characterizing multiple spatial dependencies of environmental factors using graph-based deep learning. J. Geophys. Res. Mach. Learn. Comput. 2(2), e2024JH000520. https://doi.org/10.1029/2024JH000520.
Wunsch, A., Liesch, T., Broda, S., 2021. Groundwater level forecasting with artificial neural networks: A comparison of Long Short-Term Memory (LSTM), Convolutional Neural Networks (CNNs), and non-linear autoregressive Networks with Exogenous Input (NARX). Hydrol. Earth Syst. Sci. 25(3), 1671–1687. https://doi.org/10.5194/hess-25-1671-2021.
Yao, Y., Zheng, C., Liu, J., Cao, G., Xiao, H., Li, H., Li, W., 2015. Conceptual and numerical models for groundwater flow in an arid inland river basin. Hydrol. Process. 29(6), 1480–1492. https://doi.org/10.1002/hyp.10276.
Yousefi, H., Zahedi, S., Niksokhan, M.H., Momeni, M., 2019. Ten-year prediction of groundwater level in Karaj plain (Iran) using MODFLOW2005-NWT in MATLAB. Environ. Earth Sci. 78, 1–14. https://doi.org/10.1007/s12665-019-8340-y.
Zha, Y., Yang, Y., 2024. Innovative graph neural network approach for predicting soil heavy metal pollution in the Pearl River Basin, China. Sci. Rep. 14(1), 16505. https://doi.org/10.1038/s41598-024-67210-w.
Zhu, Y., Liu, Q., 2025. Toward transparent groundwater contamination risk forecasting: Integrating causal discovery and Bayesian graph neural networks. Sci. Total Environ. 998, 180233. https://doi.org/10.1016/j.scitotenv.2024.180233.