基于对抗学习的全局多尺度轻量化冰川分割算法

    Global multi-scale lightweight glacier segmentation algorithm based on adversarial learning

    • 摘要: 冰川遥感分割中存在类间光谱相似、纹理同质化严重等问题,导致高质量标注代价高、样本规模受限,为此提出基于对抗学习的全局多尺度轻量化冰川分割算法(LmgFormer)。采用SegFormer作为生成器主干特征提取网络,以生成伪数据特征图。设计具有对象级上下文信息模块(atrous spatial pyramid-object level context, ASP-OLC),以空洞卷积和对象级上下文信息联合建模多尺度特征,同时降低整体参数量和计算量;引入全局注意力模块(global attention mechanism, GAM)进一步提取、融合全局范围内通道和空间交互关联特征信息;采用全卷积判别器(fully convolutional discriminator, FCD)替代传统分类判别器,通过对真实与生成分割结果的对抗学习提升特征判别性与模型鲁棒性;利用条件随机场CRF消除分割结果图的孤立情况。实验结果表明,LmgFormer模型对于冰川分割的准确率达到了94.2%,综合参数量降低至8.123 M。

       

      Abstract: Glacier remote-sensing segmentation suffers from high inter-class spectral similarity and strong texture homogeneity, which makes high-quality annotation costly and limits the size of labeled datasets. To address this, we proposed LmgFormer, a global multi-scale lightweight glacier segmentation algorithm based on adversarial learning. SegFormer was adopted as the backbone of the generator to extract features and produce pseudo feature maps. The atrous spatial pyramid–object level context (ASP-OLC) module was designed to jointly model multi-scale features with atrous convolutions and object-level contextual information, while reducing the overall number of parameters and computations. The global attention mechanism (GAM) was introduced to further extract and fuse global channel-spatial interaction features. The fully convolutional discriminator (FCD) replaced the conventional classification discriminator, enhancing feature discriminability and model robustness through adversarial learning on real and generated segmentation results. The conditional random field (CRF) was finally employed to remove isolated artifacts in the segmentation maps. Experimental results show that LmgFormer achieves 94.2% accuracy on glacier segmentation with only 8.123 M parameters.

       

    /

    返回文章
    返回