Spatio-Temporal Graph Neural Networks for Streamflow Prediction in the Upper Colorado Basin

Spatio-Temporal Graph Neural Networks for Streamflow Prediction in the Upper Colorado Basin

Streamflow prediction is vital for effective water resource management, enabling a better understanding of hydrological variability and its response to environmental factors. This study presents a spatio-temporal graph neural network (STGNN) model for streamflow prediction in the Upper Colorado Rive...

Full description

Saved in:
Bibliographic Details
Main Authors: Akhila Akkala, Soukaina Filali Boubrahimi, Shah Muhammad Hamdi, Pouya Hosseinzadeh, Ayman Nassar
Format: Article
Language:English
Published: MDPI AG 2025-03-01
Series:Hydrology
Subjects:
streamflow prediction
machine learning
graph network
temporal and spatial characteristics
Upper Colorado River Basin
Online Access:https://www.mdpi.com/2306-5338/12/3/60
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Streamflow prediction is vital for effective water resource management, enabling a better understanding of hydrological variability and its response to environmental factors. This study presents a spatio-temporal graph neural network (STGNN) model for streamflow prediction in the Upper Colorado River Basin (UCRB), integrating graph convolutional networks (GCNs) to model spatial connectivity and long short-term memory (LSTM) networks to capture temporal dynamics. Using 30 years of monthly streamflow data from 20 monitoring stations, the STGNN predicted streamflow over a 36-month horizon and was evaluated against traditional models, including random forest regression (RFR), LSTM, gated recurrent units (GRU), and seasonal auto-regressive integrated moving average (SARIMA). The STGNN outperformed these models across multiple metrics, achieving an R<sup>2</sup> of 0.78, an RMSE of 0.81 mm/month, and a KGE of 0.79 at critical locations like Lees Ferry. A sequential analysis of input–output configurations identified the (36, 36) setup as optimal for balancing historical context and forecasting accuracy. Additionally, the STGNN showed strong generalizability when applied to other locations within the UCRB. These results underscore the importance of integrating spatial dependencies and temporal dynamics in hydrological forecasting, offering a scalable and adaptable framework to improve predictive accuracy and support adaptive water resource management in river basins.
ISSN:2306-5338

Similar Items