Scale-balanced loss for object detection

Shuang, Kai, Lyu, Zhiheng, Loo, Jonathan ORCID: https://orcid.org/0000-0002-2197-8126 and Zhang, Wentao (2021) Scale-balanced loss for object detection. Pattern Recognition, 117. p. 107997. ISSN 0031-3203

[img] PDF
Scale-balanced loss for object detection.pdf - Accepted Version
Restricted to Repository staff only until 23 April 2022.
Available under License Creative Commons Attribution Non-commercial No Derivatives.

Download (1MB)

Abstract

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
Uncontrolled 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: 13 May 2021 14:40
URI: http://repository.uwl.ac.uk/id/eprint/7864

Actions (login required)

View Item View Item

Menu