オブジェクト検出は、分類と回帰を同時に最適化する典型的なマルチタスク学習アプリケーションです。ただし、アンカーベースの方法では、分類の損失が常にマルチタスクの損失を支配し、タスクの一貫性のあるバランスの取れた最適化を妨げます。この論文では、バウンディングボックスをシフトすると、分類における正と負のサンプルの分割が変わる可能性があることを発見しました。つまり、分類は回帰に依存します。さらに、さまざまなデータセット、オプティマイザー、および回帰損失関数を考慮して、損失の重みを微調整することに関する3つの重要な結論を要約します。上記の結論に基づいて、損失の統計的特性に従ってアンカーベースの方法を最適化する際の不均衡を解決するために、適応損失重量調整(ALWA)を提案します。 ALWAを以前の最先端の検出器に組み込むことにより、L1、SmoothL1、およびCIoUが失われた場合でも、PASCALVOCおよびMSCOCOのパフォーマンスが大幅に向上します。コードはhttps://github.com/ywx-hub/ALWAで入手できます。
Object detection is a typical multi-task learning application, which optimizes classification and regression simultaneously. However, classification loss always dominates the multi-task loss in anchor-based methods, hampering the consistent and balanced optimization of the tasks. In this paper, we find that shifting the bounding boxes can change the division of positive and negative samples in classification, meaning classification depends on regression. Moreover, we summarize three important conclusions about fine-tuning loss weights, considering different datasets, optimizers and regression loss functions. Based on the above conclusions, we propose Adaptive Loss Weight Adjustment(ALWA) to solve the imbalance in optimizing anchor-based methods according to statistical characteristics of losses. By incorporating ALWA into previous state-of-the-art detectors, we achieve a significant performance gain on PASCAL VOC and MS COCO, even with L1, SmoothL1 and CIoU loss. The code is available at https://github.com/ywx-hub/ALWA.