arXiv reaDer
Wise-IoU: 動的集束メカニズムによるバウンディング ボックス回帰損失
Wise-IoU: Bounding Box Regression Loss with Dynamic Focusing Mechanism
バウンディング ボックス回帰 (BBR) の損失関数は、オブジェクト検出に不可欠です。その優れた定義により、モデルのパフォーマンスが大幅に向上します。ほとんどの既存の作業は、トレーニング データの例が高品質であることを前提としており、BBR 損失のフィッティング能力の強化に焦点を当てています。低品質の例でやみくもに BBR を強化すると、ローカリゼーションのパフォーマンスが損なわれます。この問題を解決するために Focal-EIoU v1 が提案されましたが、その静的集束メカニズム (FM) が原因で、非単調 FM の可能性は十分に活用されませんでした。この考えに基づいて、Wise-IoU (WIoU) と呼ばれる動的で非単調な FM を使用した IoU ベースの損失を提案します。動的で非単調な FM は、IoU の代わりに異常値を使用してアンカー ボックスの品質を評価し、賢明な勾配ゲイン割り当て戦略を提供します。この戦略は、高品質のアンカー ボックスの競争力を低下させると同時に、低品質の例によって生成される有害な勾配も削減します。これにより、WIoU は通常の品質のアンカー ボックスに集中し、検出器の全体的なパフォーマンスを向上させることができます。 WIoU を最先端のリアルタイム検出器 YOLOv7 に適用すると、MS-COCO データセットの AP-75 が 53.03% から 54.50% に改善されます。コードは https://github.com/Instinct323/wiou で入手できます。
The loss function for bounding box regression (BBR) is essential to object detection. Its good definition will bring significant performance improvement to the model. Most existing works assume that the examples in the training data are high-quality and focus on strengthening the fitting ability of BBR loss. If we blindly strengthen BBR on low-quality examples, it will jeopardize localization performance. Focal-EIoU v1 was proposed to solve this problem, but due to its static focusing mechanism (FM), the potential of non-monotonic FM was not fully exploited. Based on this idea, we propose an IoU-based loss with a dynamic non-monotonic FM named Wise-IoU (WIoU). The dynamic non-monotonic FM uses the outlier degree instead of IoU to evaluate the quality of anchor boxes and provides a wise gradient gain allocation strategy. This strategy reduces the competitiveness of high-quality anchor boxes while also reducing the harmful gradient generated by low-quality examples. This allows WIoU to focus on ordinary-quality anchor boxes and improve the detector's overall performance. When WIoU is applied to the state-of-the-art real-time detector YOLOv7, the AP-75 on the MS-COCO dataset is improved from 53.03% to 54.50%. Code is available at https://github.com/Instinct323/wiou.
updated: Sat Apr 08 2023 13:58:40 GMT+0000 (UTC)
published: Tue Jan 24 2023 14:50:40 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アソシエイト