arXiv reaDer
物体検出のための動的マルチスケール損失最適化
Dynamic Multi-Scale Loss Optimization for Object Detection
高度なモデルアーキテクチャによるオブジェクト検出器のパフォーマンスの継続的な改善により、トレーニングプロセスにおける不均衡の問題がより注目されています。マルチスケール検出を実行することは、オブジェクト検出フレームワークの一般的なパラダイムです。ただし、各スケールはトレーニング中に等しく扱われます。この論文では、マルチスケール検出器トレーニングの客観的不均衡を注意深く研究します。各スケールレベルでの損失は、等しく重要でも独立でもないことを主張します。マルチタスクの重みを設定する既存のソリューションとは異なり、トレーニングプロセスで各スケールレベルの損失の重みを動的に最適化します。具体的には、統計的分散に従ってマルチスケール損失のバランスをとるために、適応分散重み付け(AVW)を提案します。次に、トレーニング中に確率的に重み付けスキームを決定するための新しい強化学習最適化(RLO)を開発します。提案された動的な方法は、余分な計算の複雑さやバックプロパゲーションのための学習可能なパラメータなしで、マルチスケールのトレーニング損失をよりよく利用します。実験によると、私たちのアプローチは、PascalVOCおよびMSCOCOベンチマークのさまざまなベースライン検出器よりも一貫してパフォーマンスを向上させることができます。
With the continuous improvement of the performance of object detectors via advanced model architectures, imbalance problems in the training process have received more attention. It is a common paradigm in object detection frameworks to perform multi-scale detection. However, each scale is treated equally during training. In this paper, we carefully study the objective imbalance of multi-scale detector training. We argue that the loss in each scale level is neither equally important nor independent. Different from the existing solutions of setting multi-task weights, we dynamically optimize the loss weight of each scale level in the training process. Specifically, we propose an Adaptive Variance Weighting (AVW) to balance multi-scale loss according to the statistical variance. Then we develop a novel Reinforcement Learning Optimization (RLO) to decide the weighting scheme probabilistically during training. The proposed dynamic methods make better utilization of multi-scale training loss without extra computational complexity and learnable parameters for backpropagation. Experiments show that our approaches can consistently boost the performance over various baseline detectors on Pascal VOC and MS COCO benchmark.
updated: Mon Aug 09 2021 13:12:41 GMT+0000 (UTC)
published: Mon Aug 09 2021 13:12:41 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アソシエイト