Hybrid Architectures Combining Cellular and Convolutional Neural Networks for Fish Classification and Disease Detection

Quang Hong Doan, Hoan Quang Nguyen, Truyen The Nguyen, Anh Duc Duong

Abstract


Fish classification and disease detection are crucial for (play crucial role in) sustainable aquaculture management, requiring high-accuracy, real-time computer vision models. This study introduces FISH-YOLOv8, an enhanced deep learning model built on YOLOv8, replacing all convolutional layers with Cellular Neural Networks (CeCNNs) to leverage their superior dynamics and noise tolerance for improved feature extraction in turbid, occluded underwater conditions. BiFormer Attention and Non-Maximum Suppression (NMS) further optimize detection accuracy and speed (enhance detection accuracy and processing speed). Evaluated on a Roboflow dataset of 1,800 images across 14 classes (10 fish species, 4 diseases), FISHYOLOv8 achieves a Mean Average Precision (mAP) mAP@50 of 0.9936 ± 0.0012 (p < 0.05) and 98.89% accuracy after 50 epochs, outperforming YOLOv8 and peers. With 52 ± 2 Frames Per Second (FPS), it offers a robust, real-time solution for aquaculture monitoring.

References


J. Redmon et al., "You Only Look Once: Unified, Real-Time Object Detection," in Proc. IEEE Conf. Comput. Vis. Pattern Recognit. (CVPR), Las Vegas, NV, USA, pp. 779-788, 2016.

YOLOv8.org, [Online]. Available: https://yolov8.org/what-is-yolov8/. [Accessed 20 Jun 2024].

L. O. Chua and L. Yang, "Cellular Neural Networks: Theory," IEEE Trans. Circuits Syst., vol. 35, no. 10, pp. 1257-1272, 1998.

L. Zhu, X. Wang, Z. Ke, W. Zhang, and R. Lau, "BiFormer: Vision Transformer with Bi-Level Routing Attention," in Proc. IEEE/CVF Conf. Comput. Vis. Pattern Recognit. (CVPR), Vancouver, Canada, pp. 10323-10333, 2023.

N. O. Salscheider, "Feature NMS: Non-Maximum Suppression by Learning Feature Embeddings," in Proc. Int. Conf. Pattern Recognit. (ICPR), Milan, Italy, pp. 7848-7854, 2020.

N. S. Abinaya, D. Susan, and R. Kumar, "Naive Bayesian Fusion Based Deep Learning Networks for Multisegmented Classification of Fishes in Aquaculture Industries," Ecological Informatics, Vols. 31, 101248, p. 101248, 2021.

H. T. Rauf, M. I. U. Lali, S. Zahoor, S. Z. H. Shah, A. U. Rehman, and S. A. C. Bukhari, "Visual Features Based Automated Identification of Fish Species Using Deep Convolutional Neural Networks," Comput. Electron. Agric., vol. 167, pp. 1-17, Dec. 2019.

S. Z. H. Shah, H. T. Rauf, M. IkramUllah, M. S. Khalid, M. Farooq, and M. Fatima, "Fish-Pak: Fish Species Dataset from Pakistan for Visual Features Based Classification," Data Brief, vol. 27, pp. 104-565, Dec. 2019.

Q. Banerjee, A. Das, S. Behra, D. Bhattaharjee, N. T. Srinivason, and M. Nasipuri, "Carp-DCAE: Deep Convolutional Autoencoder for Carp Fish Classification," Computers and Electronics in Agriculture, Vols. 196, 106810, May 2022.

S. A. Shammi, S. Das, M. Hasan, and S. R. H. Noori, "Fishnet: Fish Classification Using Convolutional Neural Network," in Proc. 12th Int. Conf. Comput. Commun. Netw. Technol. (ICCCNT), Kharagpur, India, pp. 1-6, 2021.

X. L. Xu, W. S. Li, and Q. L. Duan, "Transfer Learning and SE-ResNet152 Networks-Based for Small-Scale Unbalanced Fish Species Identification," Comput. Electron. Agric., vol. 180, Jan. 2021.

2. A. Kuswantori, T. Suesut, W. Tangsrirat, and N. Nunak, "Development of Object Detection and Classification with YOLOv4 for Similar and Structural Deformed Fish," EUREKA: Physics and Engineering, vol. 2, pp. 154-165, 2022.

M. A. Ahmed, M. S. Hossain, W. Rahman, A. H. Uddin, and M. T. Islam, "An Advanced Bangladeshi Local Fish Classification System Based on the Combination of Deep Learning and the Internet of Things (IoT)," Agric. Food Res., vol. 14, no. 1, pp. 1-34, Jan. 2023.

B. Gong, K. Dai, J. Shao, L. Jing, and Y. Chen, "Fish-TViT: A Novel Fish Species Classification Method in Multi Water Areas Based on Transfer Learning and Vision Transformer," Heliyon, vol. 9, no. 6, pp. 1-12, Jun. 2023.

M. Sohan, T. Sai Ram, and C. V. Rami Reddy, "A Review on YOLOv8 and Its Advancements," in Proc. Data Intell. Cogn. Inform., Tirunelveli, India, pp. 1-12, 2024.

Z. Zheng, P. Wang, W. Liu, J. Li, R. Ye, and D. Ren, "Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression," in Proc. AAAI Conf. Artif. Intell., Hawaii, USA, pp. 12993-13000, 2019.

H. Zhang, Y. Wang, F. Dayoub, and N. Sunderhauf, "VarifocalNet: An IoU-Aware Dense Object Detector," in Proc. IEEE/CVF Conf. Comput. Vis. Pattern Recognit. (CVPR), TN, USA, pp. 8514-8523, 2020.




DOI: http://dx.doi.org/10.21553/rev-jec.402

Copyright (c) 2025 REV Journal on Electronics and Communications


ISSN: 1859-378X

Copyright © 2011-2024
Radio and Electronics Association of Vietnam
All rights reserved