Stream temperature prediction in a shifting environment: The influence of deep learning architecture

Water Resources Research
By: , and 

Links

Abstract

Stream temperature is a fundamental control on ecosystem health. Recent efforts incorporating process guidance into deep learning models for predicting stream temperature have been shown to outperform existing statistical and physical models. This performance is in part because deep learning architectures can actively learn spatiotemporal relationships that govern how water and energy propagate through a river network. However, exploration of how spatiotemporal awareness and process guidance influence a model's generalizability under shifting environmental conditions such as climate change is limited. Here, we use Explainable Artificial Intelligence (XAI) to interrogate how differing deep learning architectures affect a model's learned spatial and temporal dependencies, and how those learned dependencies affect a model's ability to maintain high accuracy when applied to unseen environmental conditions. Using the Delaware River Basin in the northeastern United States as a test case, we compare two spatiotemporally aware process-guided deep learning models for predicting stream temperature (a recurrent graph convolution network—RGCN, and a temporal convolution graph model—Graph WaveNet). Both models achieve equally high predictive performance when testing data are well represented in the training data (test root mean squared errors of 1.64°C and 1.65°C); however, Graph WaveNet significantly outperforms RGCN in 4 out of 5 experiments where test partitions represent different types of unseen environmental conditions. XAI results show that the architecture of Graph WaveNet leads to learned spatial relationships with greater fidelity to physical processes, and that this fidelity improves the generalizability of the model when applied to shifting and/or unseen environmental conditions.

Study Area

Publication type Article
Publication Subtype Journal Article
Title Stream temperature prediction in a shifting environment: The influence of deep learning architecture
Series title Water Resources Research
DOI 10.1029/2022WR033880
Volume 59
Issue 4
Year Published 2023
Language English
Publisher American Geophysical Union
Contributing office(s) New England Water Science Center, WMA - Integrated Information Dissemination Division
Description e2022WR033880, 19 p.
Country United States
Other Geospatial Delaware River Basin
Google Analytic Metrics Metrics page
Additional publication details