Autonomous intersection over union (IoU) loss: adaptive dynamic non-monotonic focal IoU loss
In object detection algorithms, the bounding box regression (BBR) loss directly influences the model's accuracy in predicting object positions. Initially, we conducted simulation experiments on the proposed intersection over union-based loss. Through in-depth analysis, we discovered that the width and height components of the distance term could hinder BBR. To address this, we proposed the AIoU-v1 loss, which decouples the width and height components of the distance term, thereby preventing the suppression of BBR loss by the distance term. Additionally, issues such as the imbalance between sample quantity and sample quality in the dataset, as well as labeling errors, can adversely affect BBR. To tackle these dataset problems, we designed an adaptive dynamic non-monotonic focusing mechanism with strong robustness and wide applicability. Finally, we proposed a post-processing algorithm that combines fusion and non-maximum suppression, resulting in more accurate bounding boxes during the post-processing stage. Our source code and data are available at https://github.com/Wenhua-Zheng/AIOU.
Authors
- Qiang Huang (ORCID: https://orcid.org/0000-0003-1120-4685)
- Jianqiang Du (ORCID: https://orcid.org/0000-0001-5584-9181)
- Yanchen Zhu (ORCID: https://orcid.org/0000-0002-6123-6880)
- Wenhua Zheng (ORCID: https://orcid.org/0009-0007-1002-809X)
Institutions
- Jiangxi University of Traditional Chinese Medicine (CN)
- Nanchang University (CN)
Publication Details
- Journal
- PeerJ Computer Science
- Published
- 2024-09-26
- DOI
- https://doi.org/10.7717/peerj-cs.2347
- Citations
- 2
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.50
Funders
- National Natural Science Foundation of China
- Education Department of Jiangxi Province
- Jiangxi University of Traditional Chinese Medicine
- National Key Research and Development Program of China