arXiv reaDer
高密度物体検出における蒸留に関するクロスタスクプロトコルの不一致のブリッジング
Bridging Cross-task Protocol Inconsistency for Distillation in Dense Object Detection
知識蒸留 (KD) は、密集オブジェクト検出におけるコンパクトなモデルを学習できる可能性を示しています。ただし、一般的に使用されるソフトマックス ベースの蒸留では、個々のカテゴリの絶対分類スコアが無視されます。したがって、蒸留損失の最適化は、必ずしも密集物体検出器の最適なスチューデント分類スコアにつながるわけではありません。このクロスタスク プロトコルの不一致は、前景カテゴリが非常に不均衡であるため、特に高密度オブジェクト検出器の場合に重大です。蒸留と分類の間のプロトコルの違いの問題に対処するために、高密度オブジェクトの検出に合わせて調整された、クロスタスクの一貫したプロトコルを備えた新しい蒸留方法を提案します。分類蒸留では、教師モデルと生徒モデルの両方の分類ロジット マップを複数の二値分類マップとして定式化し、二値分類蒸留損失を各マップに適用することで、タスク間プロトコルの不一致の問題に対処します。局在化蒸留については、特定のネットワーク構造に依存せず、既存の局在化蒸留損失と比較できる IoU ベースの局在化蒸留損失を設計します。私たちが提案した手法はシンプルですが効果的であり、実験結果は既存の手法よりも優れていることを示しています。コードは https://github.com/TinyTigerPan/BCKD で入手できます。
Knowledge distillation (KD) has shown potential for learning compact models in dense object detection. However, the commonly used softmax-based distillation ignores the absolute classification scores for individual categories. Thus, the optimum of the distillation loss does not necessarily lead to the optimal student classification scores for dense object detectors. This cross-task protocol inconsistency is critical, especially for dense object detectors, since the foreground categories are extremely imbalanced. To address the issue of protocol differences between distillation and classification, we propose a novel distillation method with cross-task consistent protocols, tailored for the dense object detection. For classification distillation, we address the cross-task protocol inconsistency problem by formulating the classification logit maps in both teacher and student models as multiple binary-classification maps and applying a binary-classification distillation loss to each map. For localization distillation, we design an IoU-based Localization Distillation Loss that is free from specific network structures and can be compared with existing localization distillation losses. Our proposed method is simple but effective, and experimental results demonstrate its superiority over existing methods. Code is available at https://github.com/TinyTigerPan/BCKD.
updated: Tue Mar 12 2024 09:29:51 GMT+0000 (UTC)
published: Mon Aug 28 2023 03:57:37 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アソシエイト