A Parallel Workflow for Polar Sea-Ice Classification using Auto-labeling of Sentinel-2 Imagery

Jurdana Masuma Iqrah,Wei Wang,Hongjie Xie,Sushil Prasad
2024-03-20
Abstract:The observation of the advancing and retreating pattern of polar sea ice cover stands as a vital indicator of global warming. This research aims to develop a robust, effective, and scalable system for classifying polar sea ice as thick/snow-covered, young/thin, or open water using Sentinel-2 (S2) images. Since the S2 satellite is actively capturing high-resolution imagery over the earth's surface, there are lots of images that need to be classified. One major obstacle is the absence of labeled S2 training data (images) to act as the ground truth. We demonstrate a scalable and accurate method for segmenting and automatically labeling S2 images using carefully determined color thresholds. We employ a parallel workflow using PySpark to scale and achieve 9-fold data loading and 16-fold map-reduce speedup on auto-labeling S2 images based on thin cloud and shadow-filtered color-based segmentation to generate label data. The auto-labeled data generated from this process are then employed to train a U-Net machine learning model, resulting in good classification accuracy. As training the U-Net classification model is computationally heavy and time-consuming, we distribute the U-Net model training to scale it over 8 GPUs using the Horovod framework over a DGX cluster with a 7.21x speedup without affecting the accuracy of the model. Using the Antarctic's Ross Sea region as an example, the U-Net model trained on auto-labeled data achieves a classification accuracy of 98.97% for auto-labeled training datasets when the thin clouds and shadows from the S2 images are filtered out.
Computer Vision and Pattern Recognition,Distributed, Parallel, and Cluster Computing,Machine Learning
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is the classification of polar sea ice using Sentinel - 2 satellite images. Specifically, the research aims to develop a powerful, effective, and scalable system for classifying polar sea ice as thick - snow - covered, young - thin, or open water. Since the Sentinel - 2 satellite is actively capturing high - resolution images of the Earth's surface, there are a large number of images that need to be classified. One of the main obstacles is the lack of labeled Sentinel - 2 training data (images) as ground truth labels. To solve this problem, the research proposes an automatic labeling method based on color thresholds and achieves a 9 - fold data - loading acceleration and a 16 - fold map - reduction acceleration using PySpark through a parallel workflow. In addition, to improve the efficiency of model training, the research also adopts the Horovod framework to distribute the training of the U - Net model on 8 GPUs, achieving a 7.21 - fold speedup without affecting the accuracy of the model. Finally, using the Ross Sea region in Antarctica as an example, the U - Net model trained on automatically - labeled data achieved a 98.97% classification accuracy on Sentinel - 2 images after thin clouds and shadows were removed.