SAR Ship Detection Based on Explainable Evidence Learning Under Intraclass Imbalance

被引:25
作者
Liu, Yingbing [1 ,2 ]
Yan, Gang [1 ,2 ]
Ma, Fei [1 ,2 ]
Zhou, Yongsheng [1 ,2 ]
Zhang, Fan [1 ,2 ]
机构
[1] Beijing Univ Chem Technol, Coll Informat Sci & Technol, Beijing 100029, Peoples R China
[2] Beijing Univ Chem Technol, Remote Sensing Technol Inst, Beijing 100029, Peoples R China
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2024年 / 62卷
基金
中国国家自然科学基金;
关键词
Uncertainty; Marine vehicles; Feature extraction; Training; Data models; Object detection; Radar polarimetry; Contrastive learning; evidence learning; intraclass imbalance; synthetic aperture radar (SAR) ship detection; UNCERTAINTY;
D O I
10.1109/TGRS.2024.3373668
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Synthetic aperture radar (SAR) ship detection is an important technology supporting water traffic monitoring and marine safety maintenance. In recent years, many methods based on deep neural networks (DNNs) have been used to improve the performance of SAR ship detection. These methods mainly focus on two issues: one is the false alarm of ship detection in complex inshore environments, and the other is the effective extraction and utilization of SAR ship features. The topic discussed in this article is one of the culprits that has caused the aforementioned two problems but has long been overlooked. Specifically, it pertains to the issue of intraclass imbalance in SAR ship detection. There are imbalances in the size distribution, azimuth distribution, and background distribution under the real data collection environment. However, since SAR ship detection is a single-class detection task, the aforementioned imbalances lack reliable descriptors during training. This article proposes using evidence learning to obtain the epistemic uncertainty as a descriptor of biased learning on samples. Contrastive learning is used to further utilize the uncertainty label of samples to correct biased learning under intraclass imbalance. The proposed method is proven to be effective on multiple network models. AP50 reaches 94.8% on the HRSID dataset, 98.4% on the SSDD dataset, and 80.9% on the LS-SSDD dataset, both achieving SOTA performance.
引用
收藏
页码:1 / 15
页数:15
相关论文
共 44 条
[1]   UncertaintyFuseNet: Robust uncertainty-aware hierarchical feature fusion model with Ensemble Monte Carlo Dropout for COVID-19 detection [J].
Abdar, Moloud ;
Salari, Soorena ;
Qahremani, Sina ;
Lam, Hak-Keung ;
Karray, Fakhri ;
Hussain, Sadiq ;
Khosravi, Abbas ;
Acharya, U. Rajendra ;
Makarenkov, Vladimir ;
Nahavandi, Saeid .
INFORMATION FUSION, 2023, 90 :364-381
[2]  
Amini A., 2020, Advances in Neural Information Processing Systems, V33, P14927
[3]  
[Anonymous], 2023, Int. J. Remote Sens., V44, P6033
[4]   Evaluation of a Neural Network With Uncertainty for Detection of Ice and Water in SAR Imagery [J].
Asadi, Nazanin ;
Scott, K. Andrea ;
Komarov, Alexander S. ;
Buehner, Mark ;
Clausi, David A. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (01) :247-259
[5]  
Blundell C, 2015, PR MACH LEARN RES, V37, P1613
[6]   Cascade R-CNN: Delving into High Quality Object Detection [J].
Cai, Zhaowei ;
Vasconcelos, Nuno .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6154-6162
[7]   Uncertainty-Incorporated Ice and Open Water Detection on Dual-Polarized SAR Sea Ice Imagery [J].
Chen, Xinwei ;
Scott, K. Andrea ;
Xu, Linlin ;
Jiang, Mingzhe ;
Fang, Yuan ;
Clausi, David A. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[8]   Grid Network: Feature Extraction in Anisotropic Perspective for Hyperspectral Image Classification [J].
Chen, Zhonghao ;
Hong, Danfeng ;
Gao, Hongmin .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
[9]   Local aggregation and global attention network for hyperspectral image classification with spectral-induced aligned superpixel segmentation [J].
Chen, Zhonghao ;
Wu, Guoyong ;
Gao, Hongmin ;
Ding, Yao ;
Hong, Danfeng ;
Zhang, Bing .
EXPERT SYSTEMS WITH APPLICATIONS, 2023, 232
[10]   Global to Local: A Hierarchical Detection Algorithm for Hyperspectral Image Target Detection [J].
Chen, Zhonghao ;
Lu, Zhengtao ;
Gao, Hongmin ;
Zhang, Yiyan ;
Zhao, Jia ;
Hong, Danfeng ;
Zhang, Bing .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60