Streamflow forecasting over gauged and ungauged basins play a vital role in water resources planning, especially under the changing climate. Increased availability of large sample hydrology data sets, together with recent advances in deep learning techniques, has presented new opportunities to explore temporal and spatial patterns in hydrological signatures for improving streamflow forecasting. The purpose of this study is to adapt and benchmark several state‐of‐the‐art graph neural network (GNN) architectures, including ChebNet, Graph Convolutional Network (GCN), and GraphWaveNet, for end‐to‐end graph learning. We explicitly represent river basins as nodes in a graph, learn the spatiotemporal nodal dependencies, and then use the learned relations to predict streamflow simultaneously across all nodes in the graph. The efficacy of the developed GNN models is investigated using the Catchment Attributes and MEteorology for Large‐sample Studies (CAMELS) data set under two settings, fixed graph topology (transductive learning), and variable graph topology (inductive learning), with the latter applicable to prediction in ungauged basins (PUB). Results indicate that GNNs are generally robust and computationally efficient, achieving similar or better performance than a baseline model trained using the long short‐term memory (LSTM) network. Further analyses are conducted to interpret the graph learning process at the edge and node levels and to investigate the effect of different model configurations. We conclude that graph learning constitutes a viable machine learning‐based method for aggregating spatiotemporal information from a multitude of sources for streamflow forecasting
Sun, Alexander Y., et al. "Explore Spatio‐Temporal Learning of Large Sample Hydrology Using Graph Neural Networks." Water Resources Research, vol. 57, no. 12, Dec. 2021. https://doi.org/10.1029/2021WR030394
Sun, Alexander Y., Jiang, Peishi, Mudunuru, Maruti K., & Chen, Xingyuan (2021). Explore Spatio‐Temporal Learning of Large Sample Hydrology Using Graph Neural Networks. Water Resources Research, 57(12). https://doi.org/10.1029/2021WR030394
Sun, Alexander Y., Jiang, Peishi, Mudunuru, Maruti K., et al., "Explore Spatio‐Temporal Learning of Large Sample Hydrology Using Graph Neural Networks," Water Resources Research 57, no. 12 (2021), https://doi.org/10.1029/2021WR030394
@article{osti_1833425,
author = {Sun, Alexander Y. and Jiang, Peishi and Mudunuru, Maruti K. and Chen, Xingyuan},
title = {Explore Spatio‐Temporal Learning of Large Sample Hydrology Using Graph Neural Networks},
annote = {Abstract Streamflow forecasting over gauged and ungauged basins play a vital role in water resources planning, especially under the changing climate. Increased availability of large sample hydrology data sets, together with recent advances in deep learning techniques, has presented new opportunities to explore temporal and spatial patterns in hydrological signatures for improving streamflow forecasting. The purpose of this study is to adapt and benchmark several state‐of‐the‐art graph neural network (GNN) architectures, including ChebNet, Graph Convolutional Network (GCN), and GraphWaveNet, for end‐to‐end graph learning. We explicitly represent river basins as nodes in a graph, learn the spatiotemporal nodal dependencies, and then use the learned relations to predict streamflow simultaneously across all nodes in the graph. The efficacy of the developed GNN models is investigated using the Catchment Attributes and MEteorology for Large‐sample Studies (CAMELS) data set under two settings, fixed graph topology (transductive learning), and variable graph topology (inductive learning), with the latter applicable to prediction in ungauged basins (PUB). Results indicate that GNNs are generally robust and computationally efficient, achieving similar or better performance than a baseline model trained using the long short‐term memory (LSTM) network. Further analyses are conducted to interpret the graph learning process at the edge and node levels and to investigate the effect of different model configurations. We conclude that graph learning constitutes a viable machine learning‐based method for aggregating spatiotemporal information from a multitude of sources for streamflow forecasting},
doi = {10.1029/2021WR030394},
url = {https://www.osti.gov/biblio/1833425},
journal = {Water Resources Research},
issn = {ISSN 0043-1397},
number = {12},
volume = {57},
place = {United States},
publisher = {American Geophysical Union (AGU)},
year = {2021},
month = {12}}
KDD '18: The 24th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mininghttps://doi.org/10.1145/3219819.3219890
KDD '20: The 26th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mininghttps://doi.org/10.1145/3394486.3403118