Imbalanced High-Resolution SAR Ship Recognition Method Based on a Lightweight CNN
Ying Zhang, Zhiyong Lei, Hui Yu, Long Zhuang
Abstract
Convolutional neural network (CNN)-based methods have become the mainstream in radar ship recognition. However, these methods suffer from two common problems. First, the training samples consist largely of common ship types, giving them an overwhelming numerical advantage over rare ship types. As a result, CNN-based recognition algorithms fail to classify rare ship types correctly. Second, huge high-resolution slices result in heavy computational burdens. To solve the first problem, namely, the class imbalance problem, this letter proposes a CNN training method that combines deep metric learning (DML) with gradually balanced sampling. DML obtains the center of each class in the feature space and performs clustering equally. Gradually balanced sampling adopts a smooth transition from instance-aware resampling to class-aware resampling to improve the recognition rate drop caused by traditional resampling methods. As for the second problem, to reduce the computational complexity of high-resolution synthetic aperture radar (SAR) images, a lightweight CNN is also proposed.