loading page

Knowledge-Guided Machine Learning for Interpretable Operational Flood Forecasting
  • +6
  • Zachary McEachran,
  • Rahul Ghosh,
  • Arvind Renganathan,
  • Somya Sharma,
  • Kelly Lindsay,
  • Michael Steinbach,
  • J L Nieber,
  • Chris Duffy,
  • Vipin Kumar
Zachary McEachran
NOAA National Weather Service Central Region

Corresponding Author:[email protected]

Author Profile
Rahul Ghosh
University of Minnesota
Author Profile
Arvind Renganathan
University of Minnesota
Author Profile
Somya Sharma
University of Minnesota Twin Cities
Author Profile
Kelly Lindsay
University of Minnesota Twin Cities
Author Profile
Michael Steinbach
University of Minnesota
Author Profile
J L Nieber
University of Minnesota
Author Profile
Chris Duffy
University of Minnesota Twin Cities
Author Profile
Vipin Kumar
Department of Computer Science and Engineering, University of Minnesota
Author Profile

Abstract

We present a knowledge-guided machine learning framework for operational hydrologic forecasting at the catchment scale. Our approach, a Factorized Hierarchical Neural Network (FHNN), has two main components: inverse and forward models. The inverse model uses observed precipitation, temperature, and streamflow data to generate a representation of the current underlying catchment state. The forward model predicts streamflow using the learned catchment state. The FHNN architecture is designed to model multi-scale processes and capture their interactions while providing explainability and interpretability. FHNN also improves forecasts based on real-time data through an inference-based data integration approach. FHNN’s data integration approach improves forecasts in response to observed data more efficiently than data assimilation methods (e.g., ensemble Kalman filtering) that require computationally intensive optimization. Once an inverse model is trained, it can quickly infer catchment states directly based on data in real-time. To show the operational performance of FHNN, we compare the FHNN forecasts with that of an expert human hydrologic forecaster using a physics-based model where both use the same imperfectly known future precipitation forecast in their modeling. The expert human forecaster creates a more accurate forecast within the first 18 hours of a forecast’s issuance, but FHNN has significantly better predictions at longer lead times. Additionally, FHNN internal states correlate strongly with internal physics-based model states, such as soil moisture, in a synthetic case. This research lays the groundwork for leveraging the predictive performance of AI-based models with the expertise in forecasting agencies to produce better river forecasts at all lead times.
10 Oct 2024Submitted to ESS Open Archive
15 Oct 2024Published in ESS Open Archive