arXiv reaDer
ロングテールオブジェクト検出における分類平衡の調査
Exploring Classification Equilibrium in Long-Tailed Object Detection
従来の検出器は、トレーニングデータの分布が大きく歪んでいると、分類が不均衡になり、パフォーマンスが低下する傾向があります。本論文では、平均分類スコアを使用して、トレーニング中の各カテゴリの分類精度を示すことを提案します。この指標に基づいて、平衡損失(EBL)とメモリ拡張機能サンプリング(MFS)メソッドを介して分類のバランスを取ります。具体的には、EBLは、任意の2つのクラス間で設計されたスコア誘導損失マージンによって、弱いクラスの決定境界の調整の強度を高めます。一方、MFSは、弱いクラスのインスタンス機能をオーバーサンプリングすることにより、弱いクラスの決定境界の調整の頻度と精度を向上させます。したがって、EBLとMFSは、ロングテール検出の分類平衡を見つけるために協力して機能し、ヘッドクラスのパフォーマンスを維持または改善しながら、テールクラスのパフォーマンスを劇的に改善します。提案手法の優位性を示すために、ResNet-50-FPNやResNet-101-FPNなどのさまざまなバックボーンを備えたマスクR-CNNを使用してLVISで実験を行います。これにより、テールクラスの検出パフォーマンスが15.6 AP向上し、最新のロングテールオブジェクト検出器よりも1AP以上優れています。コードはhttps://github.com/fcjian/LOCEで入手できます。
The conventional detectors tend to make imbalanced classification and suffer performance drop, when the distribution of the training data is severely skewed. In this paper, we propose to use the mean classification score to indicate the classification accuracy for each category during training. Based on this indicator, we balance the classification via an Equilibrium Loss (EBL) and a Memory-augmented Feature Sampling (MFS) method. Specifically, EBL increases the intensity of the adjustment of the decision boundary for the weak classes by a designed score-guided loss margin between any two classes. On the other hand, MFS improves the frequency and accuracy of the adjustment of the decision boundary for the weak classes through over-sampling the instance features of those classes. Therefore, EBL and MFS work collaboratively for finding the classification equilibrium in long-tailed detection, and dramatically improve the performance of tail classes while maintaining or even improving the performance of head classes. We conduct experiments on LVIS using Mask R-CNN with various backbones including ResNet-50-FPN and ResNet-101-FPN to show the superiority of the proposed method. It improves the detection performance of tail classes by 15.6 AP, and outperforms the most recent long-tailed object detectors by more than 1 AP. Code is available at https://github.com/fcjian/LOCE.
updated: Wed Aug 18 2021 01:48:14 GMT+0000 (UTC)
published: Tue Aug 17 2021 08:39:04 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アソシエイト