91做厙

Skip to main content
SHARE
Publication

Temporal sequence transformer to advance long-term streamflow prediction

by Ruhaan Singh, Dan Lu, Kshitij Tayal
Publication Type
Conference Paper
Book Title
Advances in Neural Information Processing Systems 37
Publication Date
Page Numbers
1 to 8
Publisher Location
California, United States of America
Conference Name
Tackling Climate Change with Machine Learning workshop at NeurIPS 2024
Conference Location
Vancouver, Canada
Conference Sponsor
Multiple sponsors
Conference Date

Accurate streamflow prediction is crucial for understanding climate change impacts on water resources and for effective management of extreme hydrological events. While Long Short-Term Memory (LSTM) networks have been the dominant data-driven approach for streamflow forecasting, recent advancements in transformer architectures for time series tasks have shown promise in outperforming traditional LSTM models. This study introduces a transformer-based model that integrates historical streamflow data with climatic variables to enhance streamflow prediction accuracy. We evaluated our transformer model against a benchmark LSTM across five diverse basins in the United States. Results demonstrate that the transformer architecture consistently outperforms the LSTM model across all evaluation metrics, highlighting its potential as a more effective tool for hydrological forecasting. This research contributes to the ongoing development of advanced AI techniques for improved water resource management and climate change adaptation strategies.