An attention U-Net model for detection of fine-scale hydrologic streamlines

Zewei Xu, Shaowen Wang, Lawrence V. Stanislawski, Zhe Jiang, Nattapon Jaroenchai, Arpan Man Sainju, Ethan Shavers, E. Lynn Usery, Li Chen, Zhiyu Li, Bin Su

Research output: Contribution to journalArticlepeer-review


Surface water is an irreplaceable resource for human survival and environmental sustainability. Accurate, finely detailed cartographic representations of hydrologic streamlines are critically important in various scientific domains, such as assessing the quantity and quality of present and future water resources, modeling climate changes, evaluating agricultural suitability, mapping flood inundation, and monitoring environmental changes. Conventional approaches to detecting such streamlines cannot adequately incorporate information from the complex three-dimensional (3D) environment of streams and land surface features. Such information is vital to accurately delineate streamlines. In recent years, high accuracy lidar data has become increasingly available for deriving both 3D information and terrestrial surface reflectance. This study develops an attention U-net model to take advantage of high-accuracy lidar data for finely detailed streamline detection and evaluates model results against a baseline of multiple traditional machine learning methods. The evaluation shows that the attention U-net model outperforms the best baseline machine learning method by an average F1 score of 11.25% and achieves significantly better smoothness and connectivity between classified streamline channels. These findings suggest that our deep learning approach can harness high-accuracy lidar data for fine-scale hydrologic streamline detection, and in turn produce desirable benefits for many scientific domains.

Original languageEnglish (US)
Article number104992
JournalEnvironmental Modelling and Software
StatePublished - Jun 2021


  • CyberGIS
  • Deep learning
  • Hydrography
  • Hydrologic streamlines
  • Lidar data analysis

ASJC Scopus subject areas

  • Software
  • Environmental Engineering
  • Ecological Modeling


Dive into the research topics of 'An attention U-Net model for detection of fine-scale hydrologic streamlines'. Together they form a unique fingerprint.

Cite this