Scaling loss: updating gradient of loss for accurate object detection

Jiahao Hu; Zihang He; Xiang Ye; Gaoxin Zhang; Yong Li

doi:10.1117/12.2557236

3 January 2020 Scaling loss: updating gradient of loss for accurate object detection

Jiahao Hu, Zihang He, Xiang Ye, Gaoxin Zhang, Yong Li

Proceedings Volume 11373, Eleventh International Conference on Graphics and Image Processing (ICGIP 2019); 113730O (2020) https://doi.org/10.1117/12.2557236
Event: Eleventh International Conference on Graphics and Image Processing, 2019, Hangzhou, China

Abstract

L₁ loss function and Intersection over Union (IoU) are commonly used in object detection. However, minimizing the loss function through training process does not necessarily amount to maximizing IoUs. L₁ loss simply assigns equal weights to difference of the width, height, and center point between a prediction box and a ground truth box but pays less attention to the contribution of each shape property. Observing this, we propose scaling loss which can be easily embedded in convolutional neural networks for mitigating the gap between IoU and loss function. The key insight is to add in the loss function the adaptive weights for width, height, and center point that encode the shape properties of the bounding box. The contribution of each shape property will be adaptively adjusted according to the difference between a prediction box and a ground truth box, i.e. increasing the weight assigned to the bad-regressed shape property. By this means, the scaling loss is able to obtain more accurate prediction box. The proposed scaling loss was embedded in Faster R-CNN and SSD, and was validated on PASCAL VOC 2007. Experimental results verify that the proposed scaling loss can improve the detection accuracy over the smooth L₁ loss and Softer-NMS.

Citation Download Citation

Jiahao Hu, Zihang He, Xiang Ye, Gaoxin Zhang, and Yong Li "Scaling loss: updating gradient of loss for accurate object detection", Proc. SPIE 11373, Eleventh International Conference on Graphics and Image Processing (ICGIP 2019), 113730O (3 January 2020); https://doi.org/10.1117/12.2557236

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available