Nguyễn Hữu Hòa , Phạm Thế Phi Thái Minh Tuấn *

* Tác giả liên hệ (minhtuan@ctu.edu.vn)

Abstract

Accurate insect classification is a key factor in pest management and crop protection. This study proposes an insect classification method using the YOLOv11 deep learning model, which features architectural and performance improvements. Two datasets are used: IP102 and PEST204, with PEST204 being an extended version of IP102 that includes more species and enhanced image quality. Variants of YOLOv11 are applied in combination with transfer learning and data augmentation to improve accuracy. The models are evaluated using metrics such as accuracy, processing time, and computational resource usage. Results showed that YOLO11x-cls achieves the highest accuracy, while YOLO11n-cls and YOLO11s-cls demonstrate potential for deployment on mobile devices due to their good performance and fast inference speed. Additionally, the Eigen-CAM technique is used to visualize important image regions, enhancing model interpretability. The findings confirm the effectiveness of YOLOv11 in insect classification, contributing to the development of intelligent agricultural systems.

Keywords: Deep learning, insect classification, YOLO

Tóm tắt

Phân loại côn trùng chính xác là yếu tố quan trọng trong quản lý sâu bệnh và bảo vệ cây trồng. Nghiên cứu này đề xuất một phương pháp phân loại côn trùng sử dụng mô hình học sâu YOLOv11 với các cải tiến về kiến trúc và hiệu suất. Hai tập dữ liệu được sử dụng là IP102 và PEST204, trong đó PEST204 là phiên bản mở rộng từ IP102 với số lượng loài và chất lượng hình ảnh được cải thiện. Các biến thể của YOLOv11 được áp dụng kết hợp với học chuyển tiếp và tăng cường dữ liệu nhằm nâng cao độ chính xác. Mô hình được đánh giá qua các chỉ số như độ chính xác, thời gian xử lý và tài nguyên tính toán. Kết quả cho thấy YOLO11x-cls đạt độ chính xác cao nhất, trong khi YOLO11n-cls và YOLO11s-cls thể hiện tiềm năng triển khai trên thiết bị di động nhờ hiệu suất tốt và tốc độ nhanh. Ngoài ra, kỹ thuật Eigen-CAM được sử dụng để trực quan hóa vùng ảnh quan trọng, giúp tăng khả năng diễn giải mô hình. Kết quả cho thấy hiệu quả của YOLOv11 trong phân loại côn trùng, đóng góp vào phát triển hệ thống nông nghiệp thông minh.

Từ khóa: Học sâu, phân loại côn trùng, YOLO

Article Details

Tài liệu tham khảo

Ahmed, S., Hasan, M. B., Ahmed, T., Sony, M. R. K., & Kabir, M. H. (2022). Less is more: Lighter and faster deep neural architecture for tomato leaf disease classification. IEEE Access, 10, 68868-68884.
https://doi.org/10.1109/ACCESS.2022.3187203

Al-Hiary, H., Bani-Ahmad, S., Reyalat, M., Braik, M., & Alrahamneh, Z. (2011). Fast and accurate detection and classification of plant diseases. International journal of computer applications, 17(1), 31-38.
https://doi.org/10.5120/2183-2754

Almryad, A. S., & Kutucu, H. (2020). Automatic identification for field butterflies by convolutional neural networks. Engineering Science and Technology, an International Journal, 23(1), 189-195.
https://doi.org/10.1016/j.jestch.2020.01.006

Chithambarathanu, M., & Jeyakumar, M. K. (2023). Survey on crop pest detection using deep learning and machine learning approaches. Multimedia Tools and Applications, 82(27), 42277-42310.
https://doi.org/10.1007/s11042-023-15221-3

Deng, J., Dong, W., Socher, R., Li, L. J., Li, K., & Fei-Fei, L. (2009). Imagenet: A largescale hierarchical image database. In 2009 IEEE Conference on Computer Vision and Pattern Recognition (pp. 248-255).
https://doi.org/10.1109/CVPR.2009.5206848

Doan, T. N. (2022). An efficient system for real-time mobile smart device-based insect detection. International Journal of Advanced Computer Science and Applications, 13(6).
https://doi.org/10.14569/IJACSA.2022.0130605

Doan, T. N. (2023). Large-scale insect pest image classification. Journal of Advances in Information Technology, 14(2), 328-341.
https://doi.org/10.12720/jait.14.2.328-341

Estruch, J. J., Carozzi, N. B., Desai, N., Duck, N. B., Warren, G. W., & Koziel, M.G. (1997). Transgenic plants: an emerging approach to pest control. Nature biotechnology, 15(2), 137-141.
https://doi.org/10.1038/nbt0297-137

Hansen, O.L., Svenning, J.C., Olsen, K., Dupont, S., Garner, B.H., Iosifidis, A., Price, B.W. and Høye, T.T.. (2020). Species-level image classification with convolutional neural network enables insect identification from habitus images. Ecology and Evolution, 10(2), 737-747.
https://doi.org/10.1002/ece3.5921

.
https://doi.org/10.1016/j.icte.2020.04.010

Kirkeby, C., Rydhmer, K., Cook, S.M., Strand, A., Torrance, M.T., Swain, J.L., Prangsma, J., Johnen, A., Jensen, M., Brydegaard, M. and Græsbøll, K. (2021). Advances in automatic identification of flying insects using optical sensors and machine learning. Scientific reports, 11(1), 1555.
https://doi.org/10.1038/s41598-021-81005-0

Krause, J., Stark, M., Deng, J., & Fei-Fei, L. (2013). 3D object representations for finegrained categorization. In Proceedings of the IEEE International Conference on Computer Vision Workshops (pp. 554-561).
https://doi.org/10.1109/ICCVW.2013.77

Li, C., Zhen, T., & Li, Z. (2022). Image classification of pests with residual neural network based on transfer learning. Applied Sciences, 12(9), 4356.
https://doi.org/10.3390/app12094356

Li, Y., Wang, H., Dang, L. M., Sadeghi-Niaraki, A., & Moon, H. (2020). Crop pest recognition in natural scenes using convolutional neural networks. Computers and Electronics in Agriculture, 169, 105174.
https://doi.org/10.1016/j.compag.2019.105174

Mikołajczyk, A., & Grochowski, M. (2018). Data augmentation for improving deep learning in image classification problem. In 2018 international interdisciplinary PhD workshop (pp. 117-122). IEEE.
https://doi.org/10.1109/IIPHDW.2018.8388338

Nanni, L., Manfè, A., Maguolo, G., Lumini, A., & Brahnam, S. (2022). High performing ensemble of convolutional neural networks for insect pest image detection. Ecological Informatics, 67, 101515.
https://doi.org/10.1016/j.ecoinf.2021.101515

Oberti, R., Marchi, M., Tirelli, P., Calcante, A., Iriti, M., Tona, E., & Ulbrich, H. (2016). Selective spraying of grapevines for disease control using a modular agricultural robot. Biosystems engineering, 146, 203-215.
https://doi.org/10.1016/j.biosystemseng.2015.12.004

Rustia, D. J. A., Chao, J. J., Chiu, L. Y., Wu, Y. F., Chung, J. Y., Hsu, J. C., & Lin, T. T. (2021). Automatic greenhouse insect pest detection and recognition based on a cascaded deep learning classification method. Journal of Applied Entomology, 145(3), 206-222.
https://doi.org/10.1111/jen.12834

Shorten, C., & Khoshgoftaar, T. M. (2019). A survey on image data augmentation for deep learning. Journal of big data, 6(1), 1-48.
https://doi.org/10.1186/s40537-019-0197-0

Wang, J., Lin, C., Ji, L., & Liang, A. (2012). A new automatic identification system of insect images at the order level. Knowledge-Based Systems, 33, 102-110.
https://doi.org/10.1016/j.knosys.2012.03.014

Wang, R., Zhang, J., Dong, W., Yu, J., Xie, C., Li, R., & Chen, H. (2017). A crop pests image classification algorithm based on deep convolutional neural network. TELKOMNIKA (Telecommunication Computing Electronics and Control), 15(3), 1239-1246.
https://doi.org/10.12928/telkomnika.v15i3.5382

Wu, X., Zhan, C., Lai, Y. K., Cheng, M. M., & Yang, J. (2019). IP102: A large-scale benchmark dataset for insect pest recognition. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition (pp. 8787-8796).
https://doi.org/10.1109/CVPR.2019.00899

Xia, D., Chen, P., Wang, B., Zhang, J., & Xie, C. (2018). Insect detection and classification based on an improved convolutional neural network. Sensors, 18(12), 4169.
https://doi.org/10.3390/s18124169