Guardado en:
Detalles Bibliográficos
Autores principales: Xu, Shaoming, Renganathan, Arvind, Khandelwal, Ankush, Ghosh, Rahul, Li, Xiang, Liu, Licheng, Tayal, Kshitij, Harrington, Peter, Jia, Xiaowei, Jin, Zhenong, Nieber, Jonh, Kumar, Vipin
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:https://arxiv.org/abs/2410.14137
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913553074618368
author Xu, Shaoming
Renganathan, Arvind
Khandelwal, Ankush
Ghosh, Rahul
Li, Xiang
Liu, Licheng
Tayal, Kshitij
Harrington, Peter
Jia, Xiaowei
Jin, Zhenong
Nieber, Jonh
Kumar, Vipin
author_facet Xu, Shaoming
Renganathan, Arvind
Khandelwal, Ankush
Ghosh, Rahul
Li, Xiang
Liu, Licheng
Tayal, Kshitij
Harrington, Peter
Jia, Xiaowei
Jin, Zhenong
Nieber, Jonh
Kumar, Vipin
contents Streamflow, vital for water resource management, is governed by complex hydrological systems involving intermediate processes driven by meteorological forces. While deep learning models have achieved state-of-the-art results of streamflow prediction, their end-to-end single-task learning approach often fails to capture the causal relationships within these systems. To address this, we propose Hierarchical Conditional Multi-Task Learning (HCMTL), a hierarchical approach that jointly models soil water and snowpack processes based on their causal connections to streamflow. HCMTL utilizes task embeddings to connect network modules, enhancing flexibility and expressiveness while capturing unobserved processes beyond soil water and snowpack. It also incorporates the Conditional Mini-Batch strategy to improve long time series modeling. We compare HCMTL with five baselines on a global dataset. HCMTL's superior performance across hundreds of drainage basins over extended periods shows that integrating domain-specific causal knowledge into deep learning enhances both prediction accuracy and interpretability. This is essential for advancing our understanding of complex hydrological systems and supporting efficient water resource management to mitigate natural disasters like droughts and floods.
format Preprint
id arxiv_https___arxiv_org_abs_2410_14137
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Hierarchical Conditional Multi-Task Learning for Streamflow Modeling
Xu, Shaoming
Renganathan, Arvind
Khandelwal, Ankush
Ghosh, Rahul
Li, Xiang
Liu, Licheng
Tayal, Kshitij
Harrington, Peter
Jia, Xiaowei
Jin, Zhenong
Nieber, Jonh
Kumar, Vipin
Machine Learning
Streamflow, vital for water resource management, is governed by complex hydrological systems involving intermediate processes driven by meteorological forces. While deep learning models have achieved state-of-the-art results of streamflow prediction, their end-to-end single-task learning approach often fails to capture the causal relationships within these systems. To address this, we propose Hierarchical Conditional Multi-Task Learning (HCMTL), a hierarchical approach that jointly models soil water and snowpack processes based on their causal connections to streamflow. HCMTL utilizes task embeddings to connect network modules, enhancing flexibility and expressiveness while capturing unobserved processes beyond soil water and snowpack. It also incorporates the Conditional Mini-Batch strategy to improve long time series modeling. We compare HCMTL with five baselines on a global dataset. HCMTL's superior performance across hundreds of drainage basins over extended periods shows that integrating domain-specific causal knowledge into deep learning enhances both prediction accuracy and interpretability. This is essential for advancing our understanding of complex hydrological systems and supporting efficient water resource management to mitigate natural disasters like droughts and floods.
title Hierarchical Conditional Multi-Task Learning for Streamflow Modeling
topic Machine Learning
url https://arxiv.org/abs/2410.14137