mlsanityData to 14 Oct 2026

Identifying Trustworthiness Challenges in Deep Learning Models for Continental-Scale Water Quality Prediction

arXiv 2503.09947 ↗First posted 13 Mar 2025cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 68 references that are in the map. First posted on arXiv 13 Mar 2025.

Builds on

20 of 68 references.

References · 20

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
  4. 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,415
  5. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  6. 2017Axiomatic Attribution for Deep Networkscited by 3,384
  7. 2016Adversarial examples in the physical worldcited by 1,769
  8. 2021Learning the solution operator of parametric partial differential equations with physics-informed DeepOnetscited by 540
  9. 2018A Benchmark for Interpretability Methods in Deep Neural Networkscited by 476
  10. 2021Recent Advances in Adversarial Training for Adversarial Robustnesscited by 279
  11. 2021Trustworthy AI: From Principles to Practicescited by 207
  12. 2017Physics-guided Neural Networks (PGNN): An Application in Lake Temperature Modelingcited by 131
8 more
  1. 2020The role of explainability in creating trustworthy artificial intelligence for health care: a comprehensive survey of the terminology, design choices, and evaluation strategiescited by 92
  2. 2022Training Uncertainty-Aware Classifiers with Conformalized Deep Learningcited by 64
  3. 2022Reliable extrapolation of deep neural operators informed by physics or sparse observationscited by 62
  4. 2021Learning Probabilistic Ordinal Embeddings for Uncertainty-Aware Regressioncited by 58
  5. 2017A trans-disciplinary review of deep learning research for water resources scientistscited by 47
  6. 2021Uncertainty-Aware Reliable Text Classificationcited by 23
  7. 2021The Need for Ethical, Responsible, and Trustworthy Artificial Intelligence for Environmental Sciencescited by 12
  8. 2021Enhancing predictive skills in physically-consistent way: Physics Informed Machine Learning for Hydrological Processescited by 6

Built on by

Not cited yet.