arXiv:2506.20688eess.IV2025-06被引 5

通过双关系蒸馏,让轻量模型更好模仿大模型的特征分布。

Building Lightweight Semantic Segmentation Models for Aerial Images Using Dual Relation Distillation

  • 分别蒸馏教师模型的空间与通道关系图
  • 在三个数据集上提升学生模型精度,无额外计算开销
  • 适合遥感图像分割的高效轻量化部署

近年来,卷积神经网络在语义分割任务中取得了显著精度提升,但模型通常庞大且推理速度慢,限制了实际应用。为此,知识蒸馏成为平衡精度与效率的有力方法。本文提出一种新型双关系蒸馏(DRD)技术,从复杂教师模型向轻量学生模型迁移特征图中的空间与通道关系。具体地,分别计算师生模型的空间与通道关系图,并通过最小化二者距离实现对齐。由于教师模型通常蕴含更丰富的空间和通道关联信息,将这些关系迁移到学生模型有助于其更好地模拟教师的特征分布,从而提升分割性能。我们在三个数据集上进行了全面实验,包括遥感领域两个常用基准(Vaihingen 和 Potsdam 数据集)及通用场景一个流行基准(Cityscapes 数据集)。结果表明,所提蒸馏框架能显著提升学生网络性能,且不引入额外计算开销。

原文摘要 · Abstract (English)

Recently, there have been significant improvements in the accuracy of CNN models for semantic segmentation. However, these models are often heavy and suffer from low inference speed, which limits their practical application. To address this issue, knowledge distillation has emerged as a promising approach to achieve a good trade-off between segmentation accuracy and efficiency. In this paper, we propose a novel dual relation distillation (DRD) technique that transfers both spatial and channel relations in feature maps from a cumbersome model (teacher) to a compact model (student). Specifically, we compute spatial and channel relation maps separately for the teacher and student models, and then align corresponding relation maps by minimizing their distance. Since the teacher model usually learns more information and collects richer spatial and channel correlations than the student model, transferring these correlations from the teacher to the student can help the student mimic the teacher better in terms of feature distribution, thus improving the segmentation accuracy of the student model. We conduct comprehensive experiments on three segmentation datasets, including two widely adopted benchmarks in the remote sensing field (Vaihingen and Potsdam datasets) and one popular benchmark in general scene (Cityscapes dataset). The experimental results demonstrate that our novel distillation framework can significantly boost the performance of the student network without incurring extra computational overhead.

语义分割知识蒸馏轻量化遥感图像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。