A Dual Weighting Label Assignment Scheme for Object Detection

by   Chenhang He, et al.

Label assignment (LA), which aims to assign each training sample a positive (pos) and a negative (neg) loss weight, plays an important role in object detection. Existing LA methods mostly focus on the design of pos weighting function, while the neg weight is directly derived from the pos weight. Such a mechanism limits the learning capacity of detectors. In this paper, we explore a new weighting paradigm, termed dual weighting (DW), to specify pos and neg weights separately. We first identify the key influential factors of pos/neg weights by analyzing the evaluation metrics in object detection, and then design the pos and neg weighting functions based on them. Specifically, the pos weight of a sample is determined by the consistency degree between its classification and localization scores, while the neg weight is decomposed into two terms: the probability that it is a neg sample and its importance conditioned on being a neg sample. Such a weighting strategy offers greater flexibility to distinguish between important and less important samples, resulting in a more effective object detector. Equipped with the proposed DW method, a single FCOS-ResNet-50 detector can reach 41.5 schedule, outperforming other existing LA methods. It consistently improves the baselines on COCO by a large margin under various backbones without bells and whistles. Code is available at https://github.com/strongwolf/DW.


page 5

page 7


Learning a Unified Sample Weighting Network for Object Detection

Region sampling or weighting is significantly important to the success o...

OneNet: Towards End-to-End One-Stage Object Detection

End-to-end one-stage object detection trailed thus far. This paper disco...

AutoAssign: Differentiable Label Assignment for Dense Object Detection

In this paper, we propose an anchor-free object detector with a fully di...

OTA: Optimal Transport Assignment for Object Detection

Recent advances in label assignment in object detection mainly seek to i...

Learning to Weight Samples for Dynamic Early-exiting Networks

Early exiting is an effective paradigm for improving the inference effic...

Learning from Noisy Anchors for One-stage Object Detection

State-of-the-art object detectors rely on regressing and classifying an ...

Selecting Learnable Training Samples is All DETRs Need in Crowded Pedestrian Detection

DEtection TRansformer (DETR) and its variants (DETRs) achieved impressiv...

Please sign up or login with your details

Forgot password? Click here to reset