# Basin2Vec: Learning Basin-Centric Embeddings for Scalable Hydrological Modeling

**Authors:** Muhammad Talha, Xiaofeng Liu, Yuan Yang, Pang-Ning Tan, A. Pouyan Nejadhashemi  
**Lab:** Michigan State University · **Published:** 3 Nov 2026  
**Links:** [arXiv](https://arxiv.org/abs/2611.05164) · [PDF](https://arxiv.org/pdf/2611.05164) · [Code](https://github.com/Muhammad-Talha-MT/basin2vec) · [This page](https://omen-bit.github.io/ai-paper-daily/paper/2611.05164.html)

## Key numbers

- **8.3%**: Basin2Vec provides average relative gains of up to 8.3% temporally and achieves 3.5x the median predictive…
- **3.5×**: …gains of up to 8.3% temporally and achieves 3.5x the median predictive performance of the strongest…

## Abstract

Reliable streamflow prediction across time and in ungauged basins requires representations that capture hydrologically relevant basin characteristics. Existing methods rely on either fixed handcrafted attributes or general-purpose Earth observation embeddings not tailored to basin structure. We introduce \textbf{Basin2Vec}, a self-supervised framework that learns compact basin representations from hydroclimatic and physiographic data without observed streamflow. Basin2Vec organizes representation learning around basin scale and relationships across space and time. Across multiple forecasting architectures, it consistently outperforms predefined attributes and general-purpose geospatial embeddings in temporal generalization and zero-shot spatial transfer. Basin2Vec provides average relative gains of up to 8.3% temporally and achieves 3.5x the median predictive performance of the strongest baselines on unseen basins. These results demonstrate the effectiveness of basin-centric pretraining for transferable hydrologic prediction. Code is available at https://github.com/Muhammad-Talha-MT/basin2vec

---

Notes written by AI from the paper, numbers copied from the abstract. Check the paper before you rely on them. From AI Papers Daily.