用随机窗口注意力与原型正则化,让小模型学懂大模型的语义分割能力
SWARD: Stochastic Window-Attention-Based Relational Distillation for Cross-Architectural Semantic Segmentation

- 设计随机滑窗注意力机制,捕捉长短距空间关系
- 引入原型判别正则化,提升小模型特征区分度
- 适用于医疗影像、城市场景等资源受限部署场景
大规模视觉基础模型在语义分割等密集预测任务中表现优异,但其体量过大难以在资源受限环境中部署,促使知识蒸馏成为将大模型能力迁移至轻量学生网络的有效手段。然而,现代基础教师模型多为基于Transformer的全局上下文编码结构,而高效的学生模型通常为感受野局部化的卷积网络。现有蒸馏方法大多假设架构同质,依赖直接特征模仿,难以弥合这一表征鸿沟,且忽视语义分割所需的结构化空间依赖与判别性组织。本文提出SWARD框架,通过两种互补机制解决该问题:首先,提出多尺度窗口注意力蒸馏(MWAD)模块,在每轮训练中随机重采样窗口偏移,实现随机滑窗分区,消除窗口边界偏差,并结合多尺度设计同时捕捉短程与长程空间依赖;其次,提出原型判别正则化(PDR)损失,通过强制类间分离与类内紧凑,塑造学生特征分布,进一步强化判别结构,超越仅靠特征模仿的性能上限。在不同视觉应用(如城市场景解析与医学图像分割)上的实验表明,SWARD达到当前最优性能。
原文摘要 · Abstract (English)
Large-scale vision foundation models have driven substantial gains on dense prediction tasks such as semantic segmentation, but their size makes deployment impractical in resource-constrained settings, motivating knowledge distillation as a means of transferring their capabilities to lightweight student networks. However, modern foundation teachers are predominantly transformer-based that encode global context, whereas efficient students are typically convolutional networks with locally biased receptive fields. Existing distillation methods largely assume architectural homogeneity and rely on direct feature mimicry, which fails to bridge this representational gap and neglects the structured spatial dependencies and discriminative organization required for accurate semantic segmentation. In this paper, we propose SWARD, a knowledge distillation framework that addresses this gap through two complementary mechanisms. First, we introduce a Multi-Scale Windowed Attention Distillation (MWAD) module that aligns teacher-student attention-based relations within stochastically shifted window partitions whose offsets are randomly resampled at every training iteration. This removes window boundary bias, and, combined with the multi-scale design, captures both short- and long-range spatial dependencies. Second, we introduce Prototype Discriminative Regularization (PDR), a loss that helps shape the student's feature distribution by enforcing inter-class separation and intra-class compactness, further sharpening the discriminative structure beyond what feature mimicry alone can produce under the student's reduced capacity. Experiments across different vision applications (i.e., urban scene parsing and medical image segmentation) show that SWARD achieves state-of-the-art performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。