Deep Learning Small Water Body Mapping by Transfer Learning from Sentinel-2 to PlanetScope

Small water bodies are widely spread and play crucial roles in supporting regional agricultural and aquaculture activities. PlanetScope imagery has a high resolution (3 m) with daily global coverage and has obviously enhanced small water body mapping. Recent studies have demonstrated the effectiveness of deep learning for mapping small water bodies using PlanetScope; however, a persistent challenge remains in the scarcity of high-quality, manually annotated water masks used for model training, which limits the generalization capability of data-driven deep learning models. In this study, we propose a transfer learning framework that leverages Sentinel-2 data to improve PlanetScope-based small water body mapping, capitalizing on the spectral interoperability between PlanetScope and Sentinel-2 bands and the abundance of open-source Sentinel-2 water masks. Eight state-of-the-art segmentation models have been explored. Additionally, this paper presents the first assessment of the VMamba model for small water body mapping, building on its demonstrated success in segmentation tasks. The models were pre-trained using Sentinel-2-derived water masks and subsequently fine-tuned with a limited set (1292 image patches, 256 × 256 pixels in each patch) of manually annotated PlanetScope labels. Experiments were conducted using 5648 image patches and two areas of 9636 km² and 2745 km², respectively. Among the evaluated methods, VMamba achieved higher accuracy compared with both CNN- and Transformer-based models. This study highlights the efficacy of combining global Sentinel-2 datasets for pre-training with localized fine-tuning, which not only enhances mapping accuracy but also reduces reliance on labor-intensive manual annotation in regional small water body mapping.