YOLO-Parallel: Positive Gradient Modeling for Long-Tail Remote Sensing Object Detection
Xiangyi Gao, Danpei Zhao, Zhichao Yuan · IEEE Geoscience and Remote Sensing Letters · 2024
The long-tail distribution problem is widely prevalent in remote sensing images (RSIs), posing significant challenges to object detection tasks. Most existing methods for long-tail detection are designed for two-stage models. Such approaches of suppressing negative gradients tend to increase false alarms in one-stage detectors, resulting in a decline in overall performance and an increase in post-processing time. This paper presents a novel long-tail loss with broad applicability in diverse You Only Look Once (YOLO) networks. We present a novel Positive Gradient Loss (PGLoss) that effectively enhances the accuracy of tail categories while preserving the accuracy of head categories. Furthermore, to address the performance degradation caused by the pseudo-residual structure, we create Parallel Block with efficient computation and superior feature extraction abilities. We designed and trained the network named YOLO-Parallel to verify the effectiveness of PGLoss and Parallel Block. Extensive experiments were conducted on two large-scale optical remote sensing datasets, DIOR and DOTA, which are severely affected by the long-tail problem. The results powerfully demonstrate the superiority of our algorithm. YOLO-Parallel, with only 33.3% of the parameters of YOLOX, achieved a comparable detection performance of 96.9% on DIOR. On DOTA dataset, PGLoss achieved mean Average Precision (mAP) improvements of around 1.5% for YOLO-Parallel, YOLOv5n, and YOLOv7-tiny without increasing NMS processing time.