Enhanced U-Net models with encoder and augmentation for phytoplankton segmentation

Ovide Decroly Wisnu Ardhi (Universitas Diponegoro)
Tri Retnaningsih Soeprobowati (Universitas Diponegoro)
Kusworo Adi (Universitas Diponegoro)
Esa Prakasa (National Research and Innovation Agency (BRIN))
Arief Rachman (National Research and Innovation Agency (BRIN))

International Journal of Advances in Applied Sciences
Vol 13, No 4: December 2024

This study comprehensively analyzes U-Net models for semantic segmentation in phytoplankton image recognition, leveraging encoders such as EfficientNet-B5, MobileNetV2, ResNet50, and ResNeXt50 and employing the Adam optimizer. The research highlights the U-Net MobileNetV2 model with optical distortion, which achieves notable test scores with 93.69% Dice, 88.14% intersection over union (IoU), 99.89% Precision, and 100% Recall, underscoring the efficacy of the applied augmentation strategies, including geometric and distortion transforms, and color and blur techniques. The U-Net ResNet50 model with mix transform consistently demonstrates high accuracy in critical metrics, outperforming others, while EfficientNet-B5 with blur suggests increased model sensitivity with improved recall. These results underscore the crucial role of encoder-augmentation synergy in model performance. Training and testing times across models have remained under 250 seconds, reflecting methodological efficiency. Overall, these results demonstrate the model's excellent performance for the semantic segmentation task.