arXiv reaDer
適応型クラスリバランス自己トレーニングによる半教師あり物体検出
Semi-Supervised Object Detection with Adaptive Class-Rebalancing Self-Training
この研究では、半教師ありオブジェクト検出(SSOD)を詳しく調べて、ラベルのないデータを追加して検出器のパフォーマンスを向上させます。最先端のSSODパフォーマンスは、トレーニングの監督がグラウンドトゥルースと疑似ラベルで構成されるセルフトレーニングによって最近達成されました。現在の研究では、SSODのクラスの不均衡が自己トレーニングの有効性を著しく妨げることを観察しています。クラスの不均衡に対処するために、CropBankと呼ばれる新しいメモリモジュールを使用した適応型クラスリバランスセルフトレーニング(ACRST)を提案します。 ACRSTは、CropBankから抽出されたフォアグラウンドインスタンスを使用してトレーニングデータを適応的にリバランスし、それによってクラスの不均衡を軽減します。検出タスクは非常に複雑であるため、自己トレーニングとデータリバランスの両方がSSODのノイズの多い疑似ラベルに悩まされていることがわかります。したがって、正確な疑似ラベルを生成するための新しい2段階フィルタリングアルゴリズムを提案します。私たちの方法は、MS-COCOおよびVOCベンチマークで満足のいく改善を達成します。 MS-COCOで1%のラベル付きデータのみを使用する場合、私たちの方法は、監視ありベースラインに対して17.02 mAPの改善を達成し、最先端の方法と比較して5.32mAPの改善を達成します。
This study delves into semi-supervised object detection (SSOD) to improve detector performance with additional unlabeled data. State-of-the-art SSOD performance has been achieved recently by self-training, in which training supervision consists of ground truths and pseudo-labels. In current studies, we observe that class imbalance in SSOD severely impedes the effectiveness of self-training. To address the class imbalance, we propose adaptive class-rebalancing self-training (ACRST) with a novel memory module called CropBank. ACRST adaptively rebalances the training data with foreground instances extracted from the CropBank, thereby alleviating the class imbalance. Owing to the high complexity of detection tasks, we observe that both self-training and data-rebalancing suffer from noisy pseudo-labels in SSOD. Therefore, we propose a novel two-stage filtering algorithm to generate accurate pseudo-labels. Our method achieves satisfactory improvements on MS-COCO and VOC benchmarks. When using only 1% labeled data in MS-COCO, our method achieves 17.02 mAP improvement over supervised baselines, and 5.32 mAP improvement compared with state-of-the-art methods.
updated: Sun Jul 11 2021 12:14:42 GMT+0000 (UTC)
published: Sun Jul 11 2021 12:14:42 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アソシエイト