arXiv reaDer
イコライゼーションロスv2:ロングテールオブジェクト検出のための新しい勾配バランスアプローチ
Equalization Loss v2: A New Gradient Balance Approach for Long-tailed Object Detection
最近提案された分離トレーニング方法は、ロングテールオブジェクト検出の主要なパラダイムとして浮上しています。ただし、追加の微調整段階が必要であり、表現と分類子の最適化がばらばらであると、結果が最適ではなくなる可能性があります。ただし、イコライゼーションロス(EQL)などのエンドツーエンドのトレーニング方法は、分離されたトレーニング方法よりもパフォーマンスが低下します。この論文では、ロングテールオブジェクト検出の主な問題がポジティブとネガティブの間の不均衡な勾配であることを明らかにし、EQLがそれをうまく解決しないことを発見しました。不均衡な勾配の問題に対処するために、等化損失v2(EQL v2)と呼ばれる新しいバージョンの等化損失を導入します。これは、各カテゴリのトレーニングプロセスを独立して均等に再調整する新しい勾配ガイド付き再計量メカニズムです。挑戦的なLVISベンチマークで広範な実験が行われます。 EQL v2は、AP全体で元のEQLを約4ポイント上回り、まれなカテゴリで14〜18ポイント改善されています。さらに重要なことに、それはまた、分離されたトレーニング方法を上回ります。 Open Imagesデータセットをさらに調整しなくても、EQL v2はEQLを6.3ポイントAP改善し、強力な一般化機能を示します。コードはhttps://github.com/tztztztztz/eqlv2でリリースされます
Recently proposed decoupled training methods emerge as a dominant paradigm for long-tailed object detection. But they require an extra fine-tuning stage, and the disjointed optimization of representation and classifier might lead to suboptimal results. However, end-to-end training methods, like equalization loss (EQL), still perform worse than decoupled training methods. In this paper, we reveal the main issue in long-tailed object detection is the imbalanced gradients between positives and negatives, and find that EQL does not solve it well. To address the problem of imbalanced gradients, we introduce a new version of equalization loss, called equalization loss v2 (EQL v2), a novel gradient guided reweighing mechanism that re-balances the training process for each category independently and equally. Extensive experiments are performed on the challenging LVIS benchmark. EQL v2 outperforms origin EQL by about 4 points overall AP with 14-18 points improvements on the rare categories. More importantly, It also surpasses decoupled training methods. Without further tuning for the Open Images dataset, EQL v2 improves EQL by 6.3 points AP, showing strong generalization ability. Codes will be released at https://github.com/tztztztztz/eqlv2
updated: Tue Dec 15 2020 19:01:48 GMT+0000 (UTC)
published: Tue Dec 15 2020 19:01:48 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アソシエイト