Attention-Enhanced YOLOv8 for Real-Time Milkfish Disease Detection with Synthetic Data Augmentation

Milkfish (Chanos chanos) aquaculture faces persistent challenges due to disease outbreaks that are difficult to detect at early stages using traditional manual monitoring methods. Varied underwater lighting conditions, turbidity, and fish movement further complicate visual assessment and may reduce detection reliability. This study proposes an automated disease detection framework that combines YOLOv8m object detection and the Convolutional Block Attention Module (CBAM) with Cycle GAN-based synthetic data augmentation. Cycle GAN is employed to generate additional diseased samples from unpaired image domains, addressing dataset imbalance and improving training diversity. CBAM is integrated into the YOLOv8m feature extraction pipeline to enhance spatial and channel attention while preserving real-time inference capability. Multiple experimental training configurations were evaluated, including raw-image training, traditional augmentation techniques, Cycle GAN-augmented datasets, and attention-enhanced detection models. Performance was assessed using precision, recall, mAP@0.50, and mAP@0.50–0.95 metrics. The proposed YOLOv8m with CBAM and Cycle GAN framework achieved the best overall performance, with a precision of 0.940, a recall of 0.910, mAP@0.50 of 0.945, and mAP@0.50–0.95 of 0.725. These results indicate that the combined use of GAN-based augmentation and attention mechanisms significantly improves detection performance under a realistic aquatic environment.

Milkfish disease detection, YOLOv8, Cycle GAN, CBAM, Synthetic data augmentation

1. Alfred, R., Leo, J., & Kaijage, S. (2025). Optimizing dataset diversity for a robust deep-learning model in rice blast disease identification to enhance crop health assessment across diverse conditions. Smart Agricultural Technology, 10. doi:https://doi.org/10.1016/j.atech.2024.100726. [Google Scholar] [Crossref]

2. Bharti, A. (2024). Object Detection Using YOLO Algorithm. International Research Journal of Modernization in Engineering Technology and Science, 06(08). doi:DOI: https://www.doi.org/10.56726/IRJMETS60812 [Google Scholar] [Crossref]

3. Biswal, S., Kumar, A., & Punuri, S. (2024). 2024 International Conference on Integrated Intelligence and Communication Systems (ICIICS). 2024 International Conference on Integrated Intelligence and Communication Systems (ICIICS). Kalaburagi, India: IEEE. doi:10.1109/ICIICS63763.2024.10860112 [Google Scholar] [Crossref]

4. Chang, C.-C., Ubina, N. A., Cheng, S.-C., Lan, H.-Y., Chen, K.-C., & Huang, C.-C. (2022, October 7). A Two-Mode Underwater Smart Sensor Object for Precision Aquaculture Based on AIoT Technology. Development and Application of Methods and Algorithms to Smart Objects and Smart Sensors, 22(19)(7603). doi:https://doi.org/10.3390/s22197603 [Google Scholar] [Crossref]

5. Chanos chanos, Milkfish : fisheries, aquaculture, gamefish, bait. (n.d.). Retrieved from https://www.fishbase.se/search.php: https://www.fishbase.se/summary/chanos-chanos.html [Google Scholar] [Crossref]

6. Chen, Y., Pan, J., & Wu, Q. (2023). Apple leaf disease identification via improved CycleGAN and Convolutional Neural Network. Soft Comput, 27, 9773–9786. doi:https://doi.org/10.1007/s00500-023-07811-y [Google Scholar] [Crossref]

7. Deepika, H., Raghuveer shetty, V., & G N, S. (2020, June). An Overview of You Only Look Once: Unified,Real-Time Object Detection . International Journal for Research in Applied Science & Engineering Technology (IJRASET), 8(VI). doi:https://doi.org/10.22214/ijraset.2020.6098 [Google Scholar] [Crossref]

8. Deng, Q., Ryu, J., & Lee, D. (2025). A Novel Rare Image Data Augmentation Method for Fish Diseases Classification. Journal of the Korean Society of Multimedia, 28(2), 137 - 148 (12 pages). doi:10.9717/kmms.2025.28.2.137 [Google Scholar] [Crossref]

9. Feng, J., & Jin, T. (2024). CEH-YOLO: A composite enhanced YOLO-based model for underwater object detection. Ecological Informatics, 82. doi:https://doi.org/10.1016/j.ecoinf.2024.102758 [Google Scholar] [Crossref]

10. Fung, B. S., Chan, W., Lo, I. M., & Tsang, D. H. (2023). Freshwater Microscopic Algae Detection Based on Deep Neural Network with GAN-Based Augmentation for Imbalanced Algal Data. ACS EST Water 2024 (pp. 982–990). American Chemical Society. doi:https://doi.org/10.1021/acsestwater.3c00150 [Google Scholar] [Crossref]

11. G. Mariani, F. Scheidegger, R. Istrate,, C. Bekas, & A. Malossi. (2018). BAGAN: Data Augmentation with Balancing GAN. ArXiv. doi:abs/1803.09655 [Google Scholar] [Crossref]

12. Garcia, R., Alvarez, J., & Marcia, R. (2023). Machine Learning for Classifying Images with Motion Blur. 2022 21st IEEE International Conference on Machine Learning and Applications (ICMLA). IEEE. doi:10.1109/ICMLA55696.2022.00079 [Google Scholar] [Crossref]

13. Gaude, G. S., & Borkar, S. (2020). Fish Detection And Tracking For Turbid Underwater Video. Madurai, India: IEEE. doi:10.1109/ICCS45141.2019.9065425 [Google Scholar] [Crossref]

14. Gong, C., Wang, D., Li, M., Chandra, V., & Liu, Q. (2021). KeepAugment: A Simple Information-Preserving Data Augmentation Approach. 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE. doi:10.1109/CVPR46437.2021.00111 [Google Scholar] [Crossref]

15. Guanbo, F., Xiong, Z., Pang, H., Gao, Y., Zhang, Z., Yang, J., & Ma, Z. (2024). RTL-YOLOv8n: A Lightweight Model for Efficient and Accurate Underwater Target Detection. Fishes. 2024, 8. doi: https://doi.org/10.3390/fishes908029 [Google Scholar] [Crossref]

16. He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep Residual Learning for Image Recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 770-778. doi:DOI:10.1109/cvpr.2016.90 [Google Scholar] [Crossref]

17. Khabusi, S., Huang, Y.-P., & Lee, M.-F. (2023). Attention-Based Mechanism for Fish Disease Classification in Aquaculture. 2023 International Conference on System Science and Engineering (ICSSE). Ho Chi Minh, Vietnam: IEEE. doi:10.1109/ICSSE58758.2023.10227224 [Google Scholar] [Crossref]

18. Komori , O., & Eguchi , S. (2019, July 3). Introduction to Imbalanced Data. In: Statistical Methods for Imbalanced Data in Ecological and Biological Studies. Statistical Methods for Imbalanced Data in Ecological and Biological Studies . doi:https://doi.org/10.1007/978-4-431-55570-4_1 [Google Scholar] [Crossref]

19. Li, S., & Peng, X. (2025). DyAqua-YOLO: a high-precision real-time underwater object detection model based on dynamic adaptive architecture. (Y. Ju, Ed.) Front. Mar. Sci., 12:1678417. doi:doi: 10.3389/fmars.2025.1678417 [Google Scholar] [Crossref]

20. Liu, X., & Goh, K. (2025). ResNet: Enabling Deep Convolutional Neural Networks through Residual Learning. arXiv.org. doi:DOI:10.48550/arXiv.2510.24036 [Google Scholar] [Crossref]

21. Martínez, F. S., Tseng, M., & Yeh, S.-P. (2006, September 1). Milkfish (Chanos chanos) Culture: Situations and Trends. Environmental Science, Agricultural and Food Sciences, Biology. doi:DOI:10.29822/JFST.200609.0004 [Google Scholar] [Crossref]

22. Minouei, M., Soheili, M., & Stricker, D. (2024). Multimodal approach for imbalanced document classification. Proceedings Volume 13517, Seventeenth International Conference on Machine Vision (ICMV 2024). Edinburg, United Kingdom: Proc. SPIE 13517. doi:https://doi.org/10.1117/12.3055119 [Google Scholar] [Crossref]

23. Prova, N. (2024, December 3). Enhancing Fish Disease Classification in Bangladeshi Aquaculture through Transfer Learning, and LIME Interpretability Techniques. 2024 4th International Conference on Sustainable Expert Systems (ICSES). doi:10.1109/ICSES63445.2024.10763350 [Google Scholar] [Crossref]

24. Woo, S., Park, J., Lee, J.-Y., & Kweon, I. (2018). CBAM: Convolutional Block Attention Module. Computer Vision – ECCV 2018, 11211. doi:https://doi.org/10.1007/978-3-030-01234-2_1 [Google Scholar] [Crossref]

25. Xi, Y., & Yin, J. (2025). UOD-YOLO: a lightweight real-time model for detecting marine organisms. Front. Mar. Sci., 12. doi:https://doi.org/10.3389/fmars.2025.1728563 [Google Scholar] [Crossref]

26. Yang, H., & Zhou, Y. (2021). IDA-GAN: A Novel Imbalanced Data Augmentation GAN. 2020 25th International Conference on Pattern Recognition (ICPR). Milan, Italy: IEEE. doi:10.1109/ICPR48806.2021.9411996 [Google Scholar] [Crossref]

27. Yuwana, R., Fauziah, F., Heryana, A., Krisnandi, D., Kusumo, R., & Pardede, H. (2020, August 31). Data Augmentation using Adversarial Networks for Tea Disesse Detection. Jurnal Elektronika dan Telekomunikasi, 20. doi:http://dx.doi.org/10.14203/jet.v20.29-35 [Google Scholar] [Crossref]

28. Zhang, Z., Li, Y., & Liu, C. (2022). Collaborative Discrimination-Enabled Generative Adversarial Network (CoD-GAN) for the Data Augmentation in Imbalanced Classification. 2022 IEEE 18th International Conference on Automation Science and Engineering (CASE). Mexico City, Mexico: IEEE. doi:10.1109/CASE49997.2022.9926707 [Google Scholar] [Crossref]

29. Zhu, J.-Y., Park, T., Isola, P., & Efros, A. A. (2017). Unpaired Image-to-Image Translation Using Cycle-Consistent Adversarial Networks. 2017 IEEE International Conference on Computer Vision (ICCV). Venice, Italy: IEEE. doi:10.1109/ICCV.2017.244 [Google Scholar] [Crossref]

• The Impact of Ownership Structure on Dividend Payout Policy of Listed Plantation Companies in Sri Lanka
• Urban Sustainability in North-East India: A Study through the lens of NER-SDG index
• Performance Assessment of Predictive Forecasting Techniques for Enhancing Hospital Supply Chain Efficiency in Healthcare Logistics
• The Fractured Self in Julian Barnes' Postmodern Fiction: Identity Crisis and Deflation in Metroland and the Sense of an Ending
• Impact of Flood on the Employment, Labour Productivity and Migration of Agricultural Labour in North Bihar