Saved in:
Bibliographic Details
Main Authors: Xia, Xiaobo, Liu, Xiaofeng, Liu, Jiale, Fang, Kuai, Lu, Lu, Oymak, Samet, Currie, William S., Liu, Tongliang
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2503.09947
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912669350494208
author Xia, Xiaobo
Liu, Xiaofeng
Liu, Jiale
Fang, Kuai
Lu, Lu
Oymak, Samet
Currie, William S.
Liu, Tongliang
author_facet Xia, Xiaobo
Liu, Xiaofeng
Liu, Jiale
Fang, Kuai
Lu, Lu
Oymak, Samet
Currie, William S.
Liu, Tongliang
contents Water quality is foundational to environmental sustainability, ecosystem resilience, and public health. Deep learning offers transformative potential for large-scale water quality prediction and scientific insights generation. However, their widespread adoption in high-stakes operational decision-making, such as pollution mitigation and equitable resource allocation, is prevented by unresolved trustworthiness challenges, including performance disparity, robustness, uncertainty, interpretability, generalizability, and reproducibility. In this work, we present a multi-dimensional, quantitative evaluation of trustworthiness benchmarking three state-of-the-art deep learning architectures: recurrent (LSTM), operator-learning (DeepONet), and transformer-based (Informer), trained on 37 years of data from 482 U.S. basins to predict 20 water quality variables. Our investigation reveals systematic performance disparities tied to process complexity, data availability, and basin heterogeneity. Management-critical variables remain the least predictable and most uncertain. Robustness tests reveal pronounced sensitivity to outliers and corrupted targets; notably, the architecture with the strongest baseline performance (LSTM) proves most vulnerable under data corruption. Attribution analyses align for simple variables but diverge for nutrients, underscoring the need for multi-method interpretability. Spatial generalization to ungauged basins remains poor across all models. This work serves as a timely call to action for advancing trustworthy data-driven methods for water resources management and provides a pathway to offering critical insights for researchers, decision-makers, and practitioners seeking to leverage artificial intelligence (AI) responsibly in environmental management.
format Preprint
id arxiv_https___arxiv_org_abs_2503_09947
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Identifying Trustworthiness Challenges in Deep Learning Models for Continental-Scale Water Quality Prediction
Xia, Xiaobo
Liu, Xiaofeng
Liu, Jiale
Fang, Kuai
Lu, Lu
Oymak, Samet
Currie, William S.
Liu, Tongliang
Machine Learning
Artificial Intelligence
Water quality is foundational to environmental sustainability, ecosystem resilience, and public health. Deep learning offers transformative potential for large-scale water quality prediction and scientific insights generation. However, their widespread adoption in high-stakes operational decision-making, such as pollution mitigation and equitable resource allocation, is prevented by unresolved trustworthiness challenges, including performance disparity, robustness, uncertainty, interpretability, generalizability, and reproducibility. In this work, we present a multi-dimensional, quantitative evaluation of trustworthiness benchmarking three state-of-the-art deep learning architectures: recurrent (LSTM), operator-learning (DeepONet), and transformer-based (Informer), trained on 37 years of data from 482 U.S. basins to predict 20 water quality variables. Our investigation reveals systematic performance disparities tied to process complexity, data availability, and basin heterogeneity. Management-critical variables remain the least predictable and most uncertain. Robustness tests reveal pronounced sensitivity to outliers and corrupted targets; notably, the architecture with the strongest baseline performance (LSTM) proves most vulnerable under data corruption. Attribution analyses align for simple variables but diverge for nutrients, underscoring the need for multi-method interpretability. Spatial generalization to ungauged basins remains poor across all models. This work serves as a timely call to action for advancing trustworthy data-driven methods for water resources management and provides a pathway to offering critical insights for researchers, decision-makers, and practitioners seeking to leverage artificial intelligence (AI) responsibly in environmental management.
title Identifying Trustworthiness Challenges in Deep Learning Models for Continental-Scale Water Quality Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2503.09947