ほとんどの既存のオブジェクト検出器は、バランスの取れたパフォーマンスを妨げるクラスの不均衡の問題に悩まされています。特に、アンカーフリーオブジェクト検出器は、ピクセルごとの予測方法での検出による背景の不均衡の問題と、前景の不均衡の問題を同時に解決する必要があります。本研究では、背景と前景のバランスを総合的に考察することで、バランスの取れた学習を誘発できるバランス指向の焦点損失を提案します。この作業は、アンカーのない物体検出器のショット数と焦点損失を含まない、極値分布ではない一般的な不均衡データを使用する状況での不均衡問題に対処することを目的としています。この不均衡の問題に精巧に対処するために、焦点損失のバッチごとのアルファ平衡バリアントを使用します。これは、一般的な不均衡なデータの再重み付けのみを使用する、シンプルで実用的なソリューションです。追加の学習コストも推論中の構造変更も必要なく、クラスのグループ化も不要です。広範な実験を通じて、各コンポーネントのパフォーマンスの向上を示し、損失の再重み付けを使用した場合の線形スケジューリングの効果を分析します。前景クラスのバランスをとるという観点から焦点損失を改善することにより、私たちの方法は、アンカーフリーのリアルタイム検出器のMS-COCOで+1.2のAPゲインを達成します。
Most existing object detectors suffer from class imbalance problems that hinder balanced performance. In particular, anchor free object detectors have to solve the background imbalance problem due to detection in a per-pixel prediction fashion as well as foreground imbalance problem simultaneously. In this work, we propose Balance-oriented focal loss that can induce balanced learning by considering both background and foreground balance comprehensively. This work aims to address imbalance problem in the situation of using a general unbalanced data of non-extreme distribution not including few shot and the focal loss for anchor free object detector. We use a batch-wise alpha-balanced variant of the focal loss to deal with this imbalance problem elaborately. It is a simple and practical solution using only re-weighting for general unbalanced data. It does require neither additional learning cost nor structural change during inference and grouping classes is also unnecessary. Through extensive experiments, we show the performance improvement for each component and analyze the effect of linear scheduling when using re-weighting for the loss. By improving the focal loss in terms of balancing foreground classes, our method achieves AP gains of +1.2 in MS-COCO for the anchor free real-time detector.