Enhanced CNN-based Small Target Detection in Sea Clutter with Controllable False Alarm

As targets floating on the sea surface get more miniature and less visible, it is becoming vital and challenging to effectively detect small targets from strong sea clutter. Besides, fitting the distribution of sea clutter is a hard task because of the complex characteristics of clutter. Classic model-based detectors are prone to suffer from the mismatch problem and a high probability of false alarm (PFA). In this paper, a PFA controllable and data-driven detection method based on an attention-enhanced convolutional neural network (CNN) is proposed. Different from mainstream CNN-based detectors, the proposed method takes time-frequency maps obtained by the Wigner-Ville distribution (WVD) as inputs. Then time-frequency maps are converted into feature images as inputs of the designed CNN strengthened with representation powers and feature refinement ability. The designed CNN is capable of automatically learning and classifying different features between targets and clutter. Meanwhile, a PFA control unit is employed to ensure an expected actual PFA. The advantages of the WVD are demonstrated theoretically and experimentally. Results with the IPIX database show that the probability of detection of the proposed method is about 0.9066 with the PFA being 10-3 and the observation time being 1.024 s. Compared with five typical feature-based detectors, the proposed method achieves better detection performance. Besides, results with the Sea-detecting Radar Data-sharing Program database also verify the feasibility of the proposed detector.

For more details, one could read our paper [1]. In this project, ADN.py is a PyTorch implementation code, and you could easily transfer this network to your research by "from ADN import adn18". The required environments include "torch" and "functools". These codes refer to some parts of ACNet (https://github.com/DingXiaoH/ACNet) and SimAM (https://github.com/ZjjConan/SimAM).

[1] Q. Qu, W. Liu, J. Wang, B. Li, N. Liu and Y. -L. Wang, "Enhanced CNN-based Small Target Detection in Sea Clutter with Controllable False Alarm," in IEEE Sensors Journal, doi: 10.1109/JSEN.2023.3259953.