セマンティック セグメンテーションにおける知識蒸留の現在のアプローチは、すべての空間的位置を同等に扱う全体的なアプローチを採用する傾向があります。ただし、高密度の予測タスクの場合は、異なる空間位置の知識表現を異なる方法で考慮することが重要です。さらに、隣接するカテゴリ間のエッジ領域は、コンテキスト情報の漏洩により非常に不確実であり、これは特にコンパクトなネットワークで顕著です。この課題に対処するために、この論文では境界特権知識蒸留 (BPKD) と呼ばれる新しいアプローチを提案します。 BPKD は、教師モデルのボディとエッジの知識を、コンパクトな学生モデルとは別に抽出します。具体的には、次の 2 つの異なる損失関数を使用します。 1) エッジ損失。これは、エッジ領域のピクセル レベルであいまいなクラスを区別することを目的としています。 2) ボディ損失。形状制約を利用し、内部意味領域に選択的に対処します。私たちの実験は、提案された BPKD 手法がエッジ領域とボディ領域に対して広範な改良と集約を提供することを示しています。さらに、このメソッドは 3 つの一般的なベンチマーク データセットでセマンティック セグメンテーションの最先端の蒸留パフォーマンスを達成し、その有効性と一般化能力を強調しています。 BPKD は、さまざまな軽量セマンティック セグメンテーション構造に対して一貫した改善を示しています。コードは https://github.com/AkideLiu/BPKD で入手できます。
Current approaches for knowledge distillation in semantic segmentation tend to adopt a holistic approach that treats all spatial locations equally. However, for dense prediction tasks, it is crucial to consider the knowledge representation for different spatial locations in a different manner. Furthermore, edge regions between adjacent categories are highly uncertain due to context information leakage, which is particularly pronounced for compact networks. To address this challenge, this paper proposes a novel approach called boundary-privileged knowledge distillation (BPKD). BPKD distills the knowledge of the teacher model's body and edges separately from the compact student model. Specifically, we employ two distinct loss functions: 1) Edge Loss, which aims to distinguish between ambiguous classes at the pixel level in edge regions. 2) Body Loss, which utilizes shape constraints and selectively attends to the inner-semantic regions. Our experiments demonstrate that the proposed BPKD method provides extensive refinements and aggregation for edge and body regions. Additionally, the method achieves state-of-the-art distillation performance for semantic segmentation on three popular benchmark datasets, highlighting its effectiveness and generalization ability. BPKD shows consistent improvements over various lightweight semantic segmentation structures. The code is available at https://github.com/AkideLiu/BPKD.