Designing effective power law-based loss function for faster and better bounding box regression

Effective bounding box regression is essential for running any real-time object detection algorithm with acceptable accuracy. The currently available loss functions have issues like high computations, and sometimes they suffer from a subtle problem of plateau for non-overlapping bounding boxes, as t...

Full description

Saved in:
Bibliographic Details
Published in:Machine vision and applications 2021-07, Vol.32 (4), Article 87
Main Authors: Aswal, Diksha, Shukla, Priya, Nandi, G. C.
Format: Article
Language:English
Subjects:
Algorithms
Artificial intelligence
Automation
Boxes
Communications Engineering
Computer Science
Computing time
Image Processing and Computer Vision
Localization
Networks
Object recognition
Original Paper
Pattern Recognition
Power law
Vision systems
Citations: Items that this one cites
Items that cite this one
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Effective bounding box regression is essential for running any real-time object detection algorithm with acceptable accuracy. The currently available loss functions have issues like high computations, and sometimes they suffer from a subtle problem of plateau for non-overlapping bounding boxes, as the resultant bounding boxes are found to be far from the ground truth. In the present investigation, we have proposed a loss function with a new power-law term introduced in it for the normalized distance, which converges as fast as the Complete Intersection over Union (CIoU), but turns out to be computationally much faster than the Intersection over Union (IoU) and Generalised IoU (GIoU). The proposed function is simpler than CIoU. The incorporated power term has been optimized based on the corresponding computational time and on the sum of errors simulated for about multi-million cases, the details of which have been elaborated in the paper. The proposed Absolute IoU (AIoU) loss function has been successfully implemented and tested using the state-of-the-art object detection algorithms, such as You Only Look Once (YOLO) and Single Shot Multibox Detector (SSD) and is found to achieve significant performance improvement, using well-known metric Average Precision (AP), indicating the effectiveness of our approach.
ISSN:0932-8092
1432-1769
DOI:10.1007/s00138-021-01206-5