ALGA-DenseNet ground-based cloud classification network based on multi-scale features

Abstract

Automatic recognition of ground-based clouds is crucial for meteorology and especially for the operational safety of Unmanned Aerial Vehicles (UAVs), but it is challenged by variable cloud shapes, complex lighting, and background interference. This paper introduces ALGA-DenseNet, an improved DenseNet model with a multi-scale attention mechanism. The model employs Color Jitter to enhance image robustness and improve learning of intra-class variations and inter-class differences. It incorporates Adaptive Local and Global Attention (ALGA) to merge features, enhancing feature selection. Additionally, it integrates mixed and depthwise separable convolutions to optimize multi-scale feature extraction, reducing parameters and computational complexity. Furthermore, integrating a Vision Transformer (ViT) and Dynamic Multi-head Attention (DMA) enhances representation of complex cloud features. Experimental results show recognition accuracies of 97.94% on the TJNU (Tianjin Normal University) Ground-based Cloud Dataset (GCD) and 97.25% on the Cirrus Cumulus Stratus Nimbus (CCSN) dataset. This indicates the model’s capability for fine-grained, multi-scale extraction of cloud textures, shapes, and color features, along with strong generalization performance.