Optimizing the Hyperparameters of Fully Convolutional Encoder-Decoder Networks for SAR Image Segmentation

• Published in: IEEE geoscience and remote sensing letters 2024, Vol.21, p.1-5
• Main Authors: Liu, Yuanyue, Zhao, Jin, Fan, Jianchao, Wang, Jun
• Format: Article
• Language: English
• Subjects: Convolution; Feature extraction; Fully convolutional encoder-decoder network; Image segmentation; Kernel; MDOAU-net; Particle swarm optimization; particle swarm optimization (PSO); Radar polarimetry; synthetic aperture radar (SAR) image segmentation; Training
• ISSN: 1545-598X; 1558-0571
• DOI: 10.1109/LGRS.2024.3431216

Fully convolutional encoder-decoder networks have been developed for the segmentation of sensing synthetic aperture radar (SAR) images. A recent one called the multiscaled attention U-net with dilated convolution and offset convolution (MDOAU-net) has been proposed for SAR image segmentation in aquaculture raft monitoring. Despite its excellent performance, its hyperparameters have to be handcrafted based on human experience, consuming a significant amount of time to tune. In this letter, a swarm intelligence algorithm is leveraged to optimize the hyperparameters of fully convolutional encoder-decoder networks (particularly MDOAU-net), including their kernel size, dilation rate, learning rate, batch size, and activation function indicator. Based on segmentation performance, early-stop termination criteria are introduced into a particle swarm optimization (PSO) algorithm to avoid overusing computing resources to train the networks. Specifically, the hyperparameters are optimized using the PSO algorithm with early-stop termination criteria. Experimental results show that the segmentation accuracy of the proposed method reaches 91.49%, which statistically outperforms other methods.