AMCD-Net: An Effective Attention-Aided Multilevel Cloud Detection Network for Optical Satellite Imagery

Han Zhai,Lulu Xue
DOI: https://doi.org/10.1109/tgrs.2024.3372589
IF: 8.2
2024-03-26
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Cloud detection is a prerequisite for optical remote sensing applications due to the ubiquitous cloud coverage and negative effect of cloud occlusions. However, it is very challenging because of the heterogeneity of clouds and diversity of underlying surfaces, especially for thin clouds with illegible shapes and dispersed distribution in addition to high transparency, which is a common bottleneck for most methods. To tackle these problems, this article proposes an innovative attention-aided multilevel cloud detection network (AMCD-Net) for optical satellite imagery. Specifically, AMCD-Net takes full account of the variability and complexity of clouds and integrates multilevel features and different attentions within a deep convolutional U-Net framework, thus enabling more accurate cloud identification in complex scenarios. On the one hand, a multilevel asymmetric convolutional (MAC) module embedded encoder is established to learn discriminative representations for clouds with various shapes and integrate the complementarity of multilevel features to improve model robustness, with a regional attention-based decoder constructed to more accurately recover complicated cloud distribution, which effectively balances the integrity and details of clouds. On the other hand, a deformable convolution-based geometry enhancement attention (GEA) is designed to refine information transmission between the encoder and the decoder, with a joint loss of binary cross-entropy (BCE) and structural similarity index (SSIM) constructed to simultaneously focus on category and morphology discriminant excavation, which are favorable for fine-grained cloud prediction. The effectiveness of AMCD-Net was verified on two well-known datasets, i.e., the 38-Cloud dataset and the SPARCS dataset, and the results demonstrate that it outperformed the other state-of-the-art deep networks.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?