Scale-balanced loss for object detection

Shuang, Kai, Lyu, Zhiheng, Loo, Jonathan ORCID: and Zhang, Wentao (2021) Scale-balanced loss for object detection. Pattern Recognition, 117. p. 107997. ISSN 0031-3203

[thumbnail of Scale-balanced loss for object detection.pdf]
Scale-balanced loss for object detection.pdf - Accepted Version
Available under License Creative Commons Attribution Non-commercial No Derivatives.

Download (1MB) | Preview


Object detection is an important field in computer vision. Nevertheless, a research area that has so far not received much attention is the study into the effectiveness of anchor matching strategy and imbalance in anchor-based object detection, in particular small object detection. It is clear that the objects with larger sizes tend to match more anchors than smaller ones. This matching imbalance may result in poor performance in detecting small objects. It can be alleviated by paying more attention to the objects that match with fewer anchors. We propose an innovative flexible loss function for object detection, which is compatible with popular anchor-based detection methods. The proposed method, called the scale-balanced loss, does not add any extra computational cost to the original pipelines. By re-weighting strategy, the proposed method significantly improves the accuracy of multi-scale object detection, especially for small objects. Comprehensive experiments indicate that the scale-balanced loss achieved excellent generalization performance when separately applied to some popular detection methods. The scale-balanced loss attained up to 15% improvements on recall rates of small and medium objects in both the PASCAL VOC and MS COCO dataset. It is also beneficial to the AP result on MS COCO with an improvement of more than 1.5%.

Item Type: Article
Identifier: 10.1016/j.patcog.2021.107997
Keywords: Object detection, Neural network, Matching imbalance
Subjects: Computing > Intelligent systems
Related URLs:
Depositing User: Jonathan Loo
Date Deposited: 10 May 2021 14:21
Last Modified: 06 Feb 2024 16:05


Downloads per month over past year

Actions (login required)

View Item View Item