Adaptive Particle Swarm Optimization of U-net Parameters for Marine Aquaculture Image Segmentation

Marine aquaculture image segmentation plays a crucial role in managing aquatic resources and environmental protection. Traditional deep learning models rely on manual parameter tuning for image segmentation, which limits their efficiency and accuracy. This paper proposes an adaptive Particle Swarm Optimization (PSO) algorithm to optimize the parameters of the U-net model automatically. The algorithm dynamically adjusts the PSO parameters based on the population’s entropy and clustering metrics and employs a hill-climbing algorithm to address the issue of the PSO easily falling into local optima, enhancing the algorithm’s adaptability. The proposed algorithm updates and iterates particles to automatically find suitable model parameters. The remote sensing data used in this experiment were from the Yellow Sea aquaculture area near Dalian, China, with a segmentation accuracy of 91.8%. This method improves segmentation accuracy, reduces the burden of manual parameter tuning, and provides an effective solution for optimizing deep learning models.