推論コストや検出器の変更なしに、オブジェクト検出器の1.0 APを改善したいですか?そのようなレシピを教えてあげましょう。驚くほど簡単です。周期的な学習率を使用して、検出器をさらに12エポックトレーニングし、これらの12のチェックポイントを最終的な検出モデルとして平均します。この強力なレシピは、ディープニューラルネットワークの一般化を改善するためにarXiv:1803.05407で提案されている確率的重み平均化(SWA)に触発されています。物体検出にも非常に効果的であることがわかりました。このテクニカルレポートでは、オブジェクトの検出とインスタンスのセグメンテーションにSWAを適用した場合の影響を体系的に調査します。広範な実験を通じて、オブジェクト検出でSWAを実行するという優れたポリシーを発見し、挑戦的なCOCOベンチマークでさまざまな一般的な検出器よりも一貫して約1.0APの改善を達成しています。この作業により、より多くの物体検出の研究者がこの手法を理解し、より優れた物体検出器のトレーニングに役立つことを願っています。コードはhttps://github.com/hyz-xmaster/swa_object_detectionで入手できます。
Do you want to improve 1.0 AP for your object detector without any inference cost and any change to your detector? Let us tell you such a recipe. It is surprisingly simple: train your detector for an extra 12 epochs using cyclical learning rates and then average these 12 checkpoints as your final detection model. This potent recipe is inspired by Stochastic Weights Averaging (SWA), which is proposed in arXiv:1803.05407 for improving generalization in deep neural networks. We found it also very effective in object detection. In this technique report, we systematically investigate the effects of applying SWA to object detection as well as instance segmentation. Through extensive experiments, we discover a good policy of performing SWA in object detection, and we consistently achieve ∼1.0 AP improvement over various popular detectors on the challenging COCO benchmark. We hope this work will make more researchers in object detection know this technique and help them train better object detectors. Code is available at: https://github.com/hyz-xmaster/swa_object_detection .