Unbiased Regression Loss for DETRs

Abstract

In this paper, we introduce a novel unbiased regression loss for DETR-based detectors. The conventional L_1 regression loss tends to bias towards larger boxes, as they disproportionately contribute more towards the overall loss compared to smaller boxes. Consequently, the detection performance for small objects suffers. To alleviate this bias, the proposed new unbiased loss, termed Sized L_1 loss, normalizes the size of all boxes based on their individual width and height. Our experiments demonstrate consistent improvements in both fully-supervised and semi-supervised settings using the MS-COCO benchmark dataset.

Tasks

Reproductions