Saved in:
Bibliographic Details
Main Authors: Huang, Yanquan, Zhen, Liu Wei, Hao, Yun, Zhang, Mengyuan, Wu, Qingyao, Deng, Zikun, Liu, Xueming, Deng, Hong
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2408.17182
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929479430963200
author Huang, Yanquan
Zhen, Liu Wei
Hao, Yun
Zhang, Mengyuan
Wu, Qingyao
Deng, Zikun
Liu, Xueming
Deng, Hong
author_facet Huang, Yanquan
Zhen, Liu Wei
Hao, Yun
Zhang, Mengyuan
Wu, Qingyao
Deng, Zikun
Liu, Xueming
Deng, Hong
contents For object detection detectors, enhancing model performance hinges on the ability to simultaneously consider inconsistencies across tasks and focus on difficult-to-train samples. Achieving this necessitates incorporating information from both the classification and regression tasks. However, prior work tends to either emphasize difficult-to-train samples within their respective tasks or simply compute classification scores with IoU, often leading to suboptimal model performance. In this paper, we propose a Hybrid Classification-Regression Adaptive Loss, termed as HCRAL. Specifically, we introduce the Residual of Classification and IoU (RCI) module for cross-task supervision, addressing task inconsistencies, and the Conditioning Factor (CF) to focus on difficult-to-train samples within each task. Furthermore, we introduce a new strategy named Expanded Adaptive Training Sample Selection (EATSS) to provide additional samples that exhibit classification and regression inconsistencies. To validate the effectiveness of the proposed method, we conduct extensive experiments on COCO test-dev. Experimental evaluations demonstrate the superiority of our approachs. Additionally, we designed experiments by separately combining the classification and regression loss with regular loss functions in popular one-stage models, demonstrating improved performance.
format Preprint
id arxiv_https___arxiv_org_abs_2408_17182
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Hybrid Classification-Regression Adaptive Loss for Dense Object Detection
Huang, Yanquan
Zhen, Liu Wei
Hao, Yun
Zhang, Mengyuan
Wu, Qingyao
Deng, Zikun
Liu, Xueming
Deng, Hong
Computer Vision and Pattern Recognition
For object detection detectors, enhancing model performance hinges on the ability to simultaneously consider inconsistencies across tasks and focus on difficult-to-train samples. Achieving this necessitates incorporating information from both the classification and regression tasks. However, prior work tends to either emphasize difficult-to-train samples within their respective tasks or simply compute classification scores with IoU, often leading to suboptimal model performance. In this paper, we propose a Hybrid Classification-Regression Adaptive Loss, termed as HCRAL. Specifically, we introduce the Residual of Classification and IoU (RCI) module for cross-task supervision, addressing task inconsistencies, and the Conditioning Factor (CF) to focus on difficult-to-train samples within each task. Furthermore, we introduce a new strategy named Expanded Adaptive Training Sample Selection (EATSS) to provide additional samples that exhibit classification and regression inconsistencies. To validate the effectiveness of the proposed method, we conduct extensive experiments on COCO test-dev. Experimental evaluations demonstrate the superiority of our approachs. Additionally, we designed experiments by separately combining the classification and regression loss with regular loss functions in popular one-stage models, demonstrating improved performance.
title Hybrid Classification-Regression Adaptive Loss for Dense Object Detection
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.17182