An improved class-balanced training sample assignment method for object detection

Chen Huang; Yan Ding; Hong Xu; Yingjie Jiao; Shichao Chen

doi:10.1117/12.2666120

4 April 2023 An improved class-balanced training sample assignment method for object detection

Chen Huang, Yan Ding, Hong Xu, Yingjie Jiao, Shichao Chen

Proceedings Volume 12617, Ninth Symposium on Novel Photoelectronic Detection Technology and Applications; 126173Q (2023) https://doi.org/10.1117/12.2666120
Event: 9th Symposium on Novel Photoelectronic Detection Technology and Applications (NDTA 2022), 2022, Hefei, China

Abstract

Class imbalance usually exists in the task of object detection based on deep learning, which has attracted extensive attention. When the number of instances belonging to different classes in the dataset is obviously unequal, class imbalance will occur, leading to the object detection model being biased towards over-represented classes during training. To handle the issue of foreground-foreground class imbalance, we design a constraint function for balancing the number of inter-class positive samples, and the improved Class-Balanced Training Sample Assignment (CBTSA) method is therefore proposed in this work. In our method, the quantitative characteristics of various classes in training set are utilized in the constraint function in order to keep the classifier in balance by equalizing the numbers of training positive samples for all kinds of ground-truth boxes. Hungarian algorithm combined with constrained positive sample numbers, CIoU loss and extended cost matrix is then used to calculate the globally optimal positive samples allocation scheme. Experiments on the challenging MS COCO 2017 benchmark are carried out to verify the effectiveness of the method given in this paper. The results demonstrate that CBTSA method boosts the performance of classifier for underrepresented classes and improves the baseline detector on detection accuracy.

Citation Download Citation

Chen Huang, Yan Ding, Hong Xu, Yingjie Jiao, and Shichao Chen "An improved class-balanced training sample assignment method for object detection", Proc. SPIE 12617, Ninth Symposium on Novel Photoelectronic Detection Technology and Applications, 126173Q (4 April 2023); https://doi.org/10.1117/12.2666120

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