定義上、オブジェクト検出では、分類タスクと回帰タスクを同時に解決するために、マルチタスクの損失が必要です。ただし、実際には減量は手動で設定する傾向があります。したがって、これまで研究されていない非常に実用的な問題が発生します。それは、現在の損失関数に適合する損失重みをすばやく見つける方法です。さらに、さまざまな回帰損失関数を選択する場合、損失の重みを調整する必要があるかどうか、調整する必要がある場合は、どのように調整する必要があるかは、迅速な解決を要求する問題です。この論文では、予測ボックスシフトの実験と理論的分析を通じて、最初に次のような最適な損失重み割り当て戦略に関する3つの重要な結論を見つけます。(1)分類損失曲線は回帰損失曲線よりも速く減衰します。 (2)損失重量が1未満である。 (3)分類と回帰損失の重みの間のギャップは大きすぎてはなりません。次に、上記の結論に基づいて、損失値の統計的特性に従ってトレーニングプロセスで損失重量を動的に調整することにより、上記の2つの問題を解決するための適応損失重量調整(ALWA)を提案します。 ALWAを1ステージと2ステージの両方のオブジェクト検出器に組み込むことにより、L1、SmoothL1、CIoU損失、PASCALVOCやMSCOCOなどの一般的なオブジェクト検出ベンチマークのパフォーマンス測定を使用してパフォーマンスが一貫して向上することを示します。コードはhttps://github.com/ywx-hub/ALWAで入手できます。
By definition, object detection requires a multi-task loss in order to solve classification and regression tasks simultaneously. However, loss weight tends to be set manually in actuality. Therefore, a very practical problem that has not been studied so far arises: how to quickly find the loss weight that fits the current loss functions. In addition, when we choose different regression loss functions, whether the loss weight need to be adjusted and if so, how should it be adjusted still is a problem demanding prompt solution. In this paper, through experiments and theoretical analysis of prediction box shifting, we firstly find out three important conclusions about optimal loss weight allocation strategy, including (1) the classification loss curve decays faster than regression loss curve; (2) loss weight is less than 1; (3) the gap between classification and regression loss weight should not be too large. Then, based on the above conclusions, we propose an Adaptive Loss Weight Adjustment(ALWA) to solve the above two problems by dynamically adjusting the loss weight in the training process, according to statistical characteristics of loss values. By incorporating ALWA into both one-stage and two-stage object detectors, we show a consistent improvement on their performance using L1, SmoothL1 and CIoU loss, performance measures on popular object detection benchmarks including PASCAL VOC and MS COCO. The code is available at https://github.com/ywx-hub/ALWA.