基于残差-状态空间模型混合架构的北极海冰分割网络

    A residual-state space model hybrid architecture for arctic sea ice segmentation

    • 摘要: 针对北极海冰多任务分割中局部与全局特征提取难以兼顾的挑战,提出一种融合了残差网络与状态空间模型的混合架构。该架构以残差网络为编码器提取多尺度局部特征,解码器则采用增强型视觉状态空间模块,通过结合二维选择性扫描与双重注意力机制实现全局的上下文建模;在跳跃连接部分,设计了跨尺度上下文融合模块以整合编码器多层级特征,利用注意力机制强化关键区域表达;在训练阶段,引入了解码器分级监督策略优化损失函数计算;最后通过并行分割头同步输出海冰密集度、发展阶段及浮冰尺寸预测。在AI4Arctic数据集上的试验表明,所提模型综合评分达84.014,在所有的3种预测任务上的表现均优于Swin Transformer、Seg Former等主流方法,实现了海冰分布的精细分割,其成果可为后续识别并划定可通航水域范围提供技术支撑。所获分割结果在北极航道规划与船舶航行安全评估中具有重要的应用价值。

       

      Abstract: To address the challenge of simultaneously extracting local and global features in multi-task segmentation of Arctic sea ice,a hybrid architecture integrating residual networks and state space models was proposed in this paper. In this architecture,residual networks were employed as the encoder to extract multi-scale local features. Enhanced Visual State Space(EVSS) modules were adopted in the decoder to achieve global context modeling through the combination of 2D Selective Scan and dual attention mechanisms. In the skip connections,Cross-Scale Contextual Fusion(CSCF) modules were designed to integrate multi-level features from the encoder. Attention mechanisms were utilized to enhance the representation of key regions. In the training phase,a Decoder Stage-Wise Supervision(DSWS) was introduced to optimize the calculation of loss functions. Finally,sea ice concentration,development stages,and floe size predictions were generated simultaneously through parallel segmentation heads. Experiments on the AI4 Arctic dataset demonstrate that the proposed model achieves a comprehensive score of 84.014. Its performance exceeds that of mainstream methods such as Swin Transformer and Seg Former in all three prediction tasks. Fine-grained segmentation of sea ice distribution is achieved in this paper. The results provide technical support for subsequent identification and delimitation of navigable water areas.The obtained segmentation results hold significant application value in Arctic route planning and ship navigation safety assessment.

       

    /

    返回文章
    返回