Transfer learning with convolutional neural networks for hydrological streamline delineation

Abstract

Hydrological streamline delineation is critical for effective environmental management, influencing agriculture sustainability, river dynamics, watershed planning, and more. This study develops a novel approach to combining transfer learning with convolutional neural networks that capitalize on ImageNet pre-trained models to improve the accuracy and transferability of streamline delineation. We evaluate the performance of eleven ImageNet pre-trained models and a baseline model using datasets from Rowan County, NC, and Covington River, VA in the USA. Our results demonstrate that when models are adapted to a new area, the fine-tuned ImageNet pre-trained model exhibits superior predictive accuracy, markedly higher than the models trained from scratch or those only fine-tuned on the same area. Moreover, the ImageNet model achieves better smoothness and connectivity between classified streamline channels. These findings underline the effectiveness of transfer learning in enhancing the delineation of hydrological streamlines across varied geographies, offering a scalable solution for accurate and efficient environmental modelling.