基于自蒸馏边缘细化的遥感图像语义分割OA
Semantic Segmentation of Remote Sensing Images Based on Self-distillation Edge Refinement
针对遥感图像语义分割过程中产生的边缘特征丢失与大量参数冗余现象,本文提出一种基于自蒸馏边缘细化的分割方法.首先基于EfficientNetB4构建主干网络;然后在自教师网络分支中引入轻量级边缘精细化模块(edge refinement module,ERM)以捕捉中间特征图的局部信息,保留被浅层神经网络过滤的中间边缘信息,从而提高遥感图像边缘像素分割精度;最后,使用每幅图像的二值类别标签为预测矩阵创建自适应多视角(self-adaptive multi-view,SAMV)向量,作为一种新知识指导编码器网络的训练,能更好地描述类内与类间分布,拟合层间与层内关系.在公开数据集DeepGlobe与Vaihingen上平均交并比分别达到72.4%和83.3%,对比实验表明,本文提出的方法能增强边缘特征的同时兼顾分割精度、模型参数与推理速度,在轻量化模型的同时具有良好的特征提取能力.
A segmentation method based on self-distillation edge refinement is proposed in this paper to tackle the challenges of edge feature loss and excessive parameter redundancy encountered during semantic segmentation of remote sensing images.Firstly,a backbone network is constructed using EfficientNetB4 as the foundation.Subsequently,a lightweight edge refinement module is integrated into the self-teacher network branch.This module is designed to capture local information from intermediate feature maps while retaining the intermediate edge details filtered by shallow neural networks,with the purpose to improve the accuracy of edge pixel segmentation in remote sensing images.Finally,an adaptive multi-view vector is created to serve as a novel knowledge guide for encoder network training.This is achieves by utilizing the binary category labels of each image as the prediction matrix.The adaptive multi-view vector provides a better description of intra-class and inter-class distributions,as well as fitting inter-layer and intra-layer relationships.On the public datasets DeepGlobe and Vaihingen,the proposed method achieves an average intersection ratio of 72.4%and 83.3%,respectively.Comparative experiments demonstrate that the method introduced in this study enhances edge features while maintaining a balance among segmentation accuracy,model parameters,and inference speed.It has good feature extraction ability while lightweighting the model.
宋冠武;李建军
中南林业科技大学计算机与数学学院,湖南长沙 410004||中南林业科技大学涉外学院信息与工程学院,湖南长沙 410211中南林业科技大学计算机与数学学院,湖南长沙 410004
信息技术与安全科学
自蒸馏边缘细化遥感图像语义分割自适应多视角
self-distillationedge refinementremote sensing imagessemantic segmentationadaptive multi-view
《广西师范大学学报(自然科学版)》 2026 (2)
65-76,12
国家自然科学基金(31570627)国家重点研发计划(2022YFD2200505)湖南省自然科学基金面上项目(202049382)湖南省自然科学基金(2020JJ4938)
评论