arXiv reaDer
セマンティックセグメンテーションのためのクラス間距離の蒸留
Distilling Inter-Class Distance for Semantic Segmentation
知識の蒸留は、計算コストを削減するためにセマンティックセグメンテーションで広く採用されています。セマンティックセグメンテーションの以前の知識の蒸留方法は、ピクセル単位の特徴の位置合わせとクラス内の特徴の変化の蒸留に焦点を当てており、クラス間の距離の知識をセマンティックセグメンテーションにとって重要な特徴空間。この問題に対処するために、特徴空間内のクラス間距離を教師ネットワークから学生ネットワークに転送するためのクラス間距離蒸留(IDD)メソッドを提案します。さらに、セマンティックセグメンテーションは位置に依存するタスクであるため、位置情報蒸留モジュールを利用して、学生ネットワークがより多くの位置情報をエンコードできるようにします。 Cityscapes、Pascal VOC、ADE20Kの3つの人気のあるデータセットでの広範な実験は、私たちの方法がセマンティックセグメンテーションモデルの精度を向上させ、最先端のパフォーマンスを達成するのに役立つことを示しています。たとえば、Cityscapesデータセットのベンチマークモデル( "PSPNet + ResNet18")の精度が7.50%向上します。
Knowledge distillation is widely adopted in semantic segmentation to reduce the computation cost.The previous knowledge distillation methods for semantic segmentation focus on pixel-wise feature alignment and intra-class feature variation distillation, neglecting to transfer the knowledge of the inter-class distance in the feature space, which is important for semantic segmentation. To address this issue, we propose an Inter-class Distance Distillation (IDD) method to transfer the inter-class distance in the feature space from the teacher network to the student network. Furthermore, semantic segmentation is a position-dependent task,thus we exploit a position information distillation module to help the student network encode more position information. Extensive experiments on three popular datasets: Cityscapes, Pascal VOC and ADE20K show that our method is helpful to improve the accuracy of semantic segmentation models and achieves the state-of-the-art performance. E.g. it boosts the benchmark model("PSPNet+ResNet18") by 7.50% in accuracy on the Cityscapes dataset.
updated: Sat May 07 2022 13:13:55 GMT+0000 (UTC)
published: Sat May 07 2022 13:13:55 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト