High-Quality Instance Mining and Weight Re-Assigning for Weakly Supervised Object Detection in Remote Sensing Images
Peixu Xing,Mengxing Huang,Chenhao Wang,Yang Cao
DOI: https://doi.org/10.3390/electronics13234753
IF: 2.9
2024-12-01
Electronics
Abstract:Weakly supervised object detection (WSOD) in remote sensing images (RSIs) achieves high-value object classification and localization by only using image-level labels. However, two problems limit its performance. Firstly, adjacent instances are often misclassified because their pseudo-labels are determined solely based on the spatial distances between them and their corresponding seed instances. Secondly, most WSOD methods assign the highest weight to the instance that covers the discriminative part of an object, thereby urging WSOD models to focus on the discriminative part rather than the whole object. To handle the first problem, a high-quality instance mining (HQIM) module, which incorporates the feature similarities between instances into the label propagation process, enabling some misclassified adjacent instances to be removed, is proposed. To tackle the second problem, a weight re-assigning (WRA) strategy, which redistributes the loss weights of instances, is proposed. Specifically, the loss weights of instances focusing on the discriminative part are exchanged with those of instances that broadly cover the whole object. Ablation studies demonstrate the effectiveness of HQIM and WRA, while comparisons with popular models on two RSI benchmarks further verify the effectiveness of our model.
engineering, electrical & electronic,computer science, information systems,physics, applied