SA-YOLO:self-adaptive loss object detection method under imbalance samples
The phenomenon of sample imbalance refers to the excessive number of background easy samples in the dataset but too few foreground hard samples,which means the sample suffers from inter-class imbalance and hard-easy imbalance.Most of the existing object detection methods are two-stage detectors based on proposed regions or one-stage detectors based on regression.When applied to imbalanced samples,it is impossible to avoid the over-dependence of the prediction bounding box generated in training on a large number of negative samples,which leads to overfitting of the model and low detection accuracy,poor accuracy and generalization.In order to achieve efficient and accurate object detection under imbalanced samples,a new SA-YOLO self-adaptive loss object detection method is proposed in the paper.1)To address the sample imbalance problem,we propose the SA-Focal Loss function,which adjusts the loss adaptively for different datasets and training stages to balance inter-class samples and hard-easy samples.2)In this paper,we construct the CSPDarknet53-SP network architecture based on the multi-scale feature prediction mechanism,which enhances the extraction ability of global features of difficult small target samples and improves the detection accuracy of difficult samples.To verify the performance of the SA-YOLO method,extensive simulation experiments are conducted on the sample imbalance dataset and the COCO dataset respectively.The results show that compared with the optimal metrics of YOLO series method,SA-YOLO reaches 91.46%of mAP in the imbalance dataset,which improves 10.87%,and the enhancement of AP50 for all kinds of objects is more than 2%,with excellent specialization;mAP50 in the COCO dataset is upgraded by 1.58%,and all indexes are not lower than the optimal value,with good effectiveness.
imbalanced sampleself-adaptive lossSA-YOLO algorithmSA-Focal Loss functionCSPDarknet53-SP network architecture
国家科技期刊平台
NSTL
万方数据
