Semi-Supervised Method for Underwater Object Detection Algorithm Based on Improved YOLOv8
Source abstract
Abstract
Deep learning-based object detection technology is rapidly developing, and underwater object detection, an important subcategory, plays a crucial role in various fields such as underwater structure repair and maintenance, as well as marine scientific research. Some of the major challenges in underwater object detection are the relatively limited availability of underwater image and video datasets and the high cost of acquiring high-quality, diverse training data. To address this, we propose a novel underwater object detection method, SUD-YOLO, based on the Mean Teacher semi-supervised learning strategy. More specifically, it combines a small number of labeled samples with a large number of unlabeled samples, using the teacher model to guide the generation of pseudo-labels. In addition, a multi-scale pseudo-label enhancement module is developed specifically to address the issue of low-quality pseudo-labels. To overcome the model’s difficulty in learning underwater image feature extraction, we integrate a receptive-field attention mechanism with local spatial features and then design a lightweight detection head based on the task alignment concept to further improve the model’s feature extraction capability. Experimental results on the DUO dataset show that, by using only 10% of the labeled data, the proposed method achieves an average precision of 50.8, which is an improvement of 11.0% over the fully supervised YOLOv8 algorithm, 11.3% over the fully supervised YOLOv11 algorithm, 9.3% over the semi-supervised Efficient Teacher algorithm, and 3.4% on the semi-supervised Unbiased Teacher algorithm, while only 20% of the computational cost is required.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Image Enhancement Techniques · Water Quality Monitoring Technologies · Advanced Neural Network Applications
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Artificial Intelligence and Computer Science, Jiangnan University, Wuxi 214122, China
Engineering Research Center of Intelligent Technology for Healthcare, Ministry of Education, Jiangnan University, Wuxi 214122, China
China Ship Scientific Research Center, Wuxi 214082, China
References (44)
Show source-provided references
1. Underwater object detection: Architectures and algorithms—A comprehensive review10.1007/s11042-022-12502-1 ↗
2. Faster R-CNN: Towards real-time object detection with region proposal networks10.1109/TPAMI.2016.2577031 ↗
3. Dai, J., Li, Y., He, K., and Sun, J. (2016, January 5–10). R-FCN: Object detection via region-based fully convolutional networks. Proceedings of the Advances in Neural Information Processing Systems (NIPS 2016), Barcelona, Spain.
4. Cai, Z., and Vasconcelos, N. (2018, January 18–22). Cascade R-CNN: Delving into high quality object detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR 2018), Salt Lake City, UT, USA.10.1109/CVPR.2018.00644 ↗
5. Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C., and Berg, A. (2016, January 11–14). Ssd: Single shot multibox detector. Proceedings of the Computer Vision–ECCV 2016: 14th European Conference, Amsterdam, The Netherlands. Proceedings, Part I.10.1007/978-3-319-46448-0_2 ↗
6. Redmon, J. (2016, January 27–30). You only look once: Unified, real-time object detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR 2016), Las Vegas, NV, USA.10.1109/CVPR.2016.91 ↗
7. FCOS: A simple and strong anchor-free object detector
8. Underwater target detection based on Faster R-CNN and adversarial occlusion network10.1016/j.engappai.2021.104190 ↗
9. Underwater human detection using Faster R-CNN with data augmentation10.1016/j.matpr.2021.05.653 ↗
10. Real-time detection of uneaten feed pellets in underwater images for aquaculture using an improved YOLO-V4 network10.1016/j.compag.2021.106135 ↗
11. Underwater object detection based on enhanced YOLOv4 architecture10.1007/s11042-023-17649-z ↗
12. Zhang, J., Zhang, J., Zhou, K., Zhang, Y., Chen, H., and Yan, X. (2023). An improved YOLOv5-based underwater object-detection framework. Sensors, 23.10.3390/s23073693 ↗
13. Song, G., Chen, W., Zhou, Q., and Guo, C. (2024). Underwater Robot Target Detection Algorithm Based on YOLOv8. Electronics, 13.10.3390/electronics13173374 ↗
14. Zhang, F., Cao, W., Gao, J., Liu, S., Li, C., Song, K., and Wang, H. (2024). Underwater Object Detection Algorithm Based on an Improved YOLOv8. J. Mar. Sci. Eng., 12.10.3390/jmse12111991 ↗
15. Blind Image Quality Assessment by Relative Gradient Statistics and AdaBoosting Neural Network10.1016/j.image.2015.10.005 ↗
16. Tarvainen, A., and Valpola, H. (2017, January 4–9). Mean teachers are better role models: Weight-averaged consistency targets improve semi-supervised deep learning results. Proceedings of the 31st Conference on Neural Information Processing Systems (NIPS 2017), Long Beach, CA, USA.
17. Jeong, J., Lee, S., Lee, J., and Park, H. (2019, January 8–14). Consistency-Based Semi-Supervised Learning for Object Detection. Proceedings of the 33rd Conference on Neural Information Processing Systems (NeurIPS 2019), Vancouver, BC, Canada.
18. Liu, Y.C., Ma, C.Y., and Kira, Z. (2022, January 18–24). Unbiased teacher v2: Semi-supervised object detection for anchor-free and anchor-based detectors. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition 2022, New Orleans, LA, USA.10.1109/CVPR52688.2022.00959 ↗
19. Sohn, K., Kim, Y., Lee, D., Kim, H., and Yoo, J. (2020). A Simple Semi-Supervised Learning Framework for Object Detection. arXiv.
20. Liu, Y.-C., Chen, C.-H., and Chou, C.-P. (2021). Unbiased Teacher for Semi-Supervised Object Detection. arXiv.
21. Xu, M., Zhang, Z., Hu, H., Wang, J., Wang, L., Wei, F., Bai, X., and Liu, Z. (2021, January 11–17). End-to-End Semi-Supervised Object Detection with Soft Teacher. Proceedings of the IEEE/CVF International Conference on Computer Vision, Montreal, BC, Canada.10.1109/ICCV48922.2021.00305 ↗
22. Xu, B., Chen, M., Guan, W., and Hu, L. (2023). Efficient Teacher: Semi-Supervised Object Detection for YOLOv5. arXiv.
23. Tang, Y., Chen, W., Luo, Y., and Zhang, Y. (2021, January 19–25). Humble Teachers Teach Better Students for Semi-Supervised Object Detection. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, TN, USA.10.1109/CVPR46437.2021.00315 ↗
24. Liu, L., Zhang, B., Zhang, J., Zhang, W., Gan, Z., Tian, G., Zhu, W., Wang, Y., and Wang, C. (2023, January 17–24). MixTeacher: Mining Promising Labels with Mixed Scale Teacher for Semi-Supervised Object Detection. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Vancouver, BC, Canada.10.1109/CVPR52729.2023.00712 ↗
25. Zhou, H., Ge, Z., Liu, S., Mao, W., Li, Z., Yu, H., and Sun, J. (2022). Dense Teacher: Dense Pseudo-Labels for Semi-supervised Object Detection. arXiv.10.1007/978-3-031-20077-9_3 ↗
26. Zhang, J., Lin, X., Zhang, W., Wang, K., Tan, X., Han, J., Ding, E., Wang, J., and Li, G. (2023, January 17–24). Semi-DETR: Semi-Supervised Object Detection with Detection Transformers. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Vancouver, BC, Canada.10.1109/CVPR52729.2023.02280 ↗
27. Mahmood, A., Bennamoun, M., An, S., Sohel, F., Boussaid, F., Hovey, R., Kendrick, G., and Fisher, R.B. (2016, January 25–28). Coral Classification with Hybrid Feature Representations. Proceedings of the 2016 IEEE International Conference on Image Processing (ICIP), Phoenix, AZ, USA.10.1109/ICIP.2016.7532411 ↗
28. Deep Image Representations for Coral Image Classification10.1109/JOE.2017.2786878 ↗
29. A Marine Object Detection Algorithm Based on SSD and Feature Enhancement10.1155/2020/5476142 ↗
30. Chen, L., Liu, Z., Tong, L., Jiang, Z., Wang, S., Dong, J., and Zhou, H. (2020, January 19–24). Underwater Object Detection Using Invert Multi-Class Adaboost with Deep Learning. Proceedings of the 2020 International Joint Conference on Neural Networks (IJCNN), Glasgow, UK.10.1109/IJCNN48605.2020.9207506 ↗
31. Liu, C., Li, H., Wang, S., Zhu, M., Wang, D., Fan, X., and Wang, Z. (2021, January 5–9). A dataset and benchmark of underwater object detection for robot picking. Proceedings of the IEEE International Conference on Multimedia & Expo Workshops (ICMEW), Shenzhen, China.10.1109/ICMEW53276.2021.9455997 ↗
32. Lin, J., Lin, H., and Wang, F. (2023). A Semi-Supervised Method for Real-Time Forest Fire Detection Algorithm Based on Adaptively Spatial Feature Fusion. Forests, 14.10.3390/f14020361 ↗
33. A survey on semi-supervised learning10.1007/s10994-019-05855-6 ↗
34. Xu, W., and Wan, Y. (2024). ELA: Efficient Local Attention for Deep Convolutional Neural Networks. arXiv.
35. Liu, S., and Huang, D. (2018, January 8–14). Receptive field block net for accurate and fast object detection. Proceedings of the European Conference on Computer Vision (ECCV), Munich, Germany.10.1007/978-3-030-01252-6_24 ↗
36. Zhang, X., Liu, C., Yang, D., Song, T., Ye, Y., Li, K., and Song, Y. (2023). RFAConv: Innovating spatial attention and standard convolutional operation. arXiv.
37. Feng, C., Zhong, Y., Gao, Y., Scott, M.R., and Huang, W. (2021, January 11–17). Tood: Task-aligned One-Stage Object Detection. Proceedings of the 2021 IEEE/CVF International Conference on Computer Vision (ICCV), Montreal, BC, Canada.10.1109/ICCV48922.2021.00349 ↗
38. Wang, R., Shivanna, R., Cheng, D.Z., Jain, S., Lin, D., Hong, L., and Chi, E.H. (2021, January 19–23). DCN v2: Improved Deep & Cross Network and Practical Lessons for Web-Scale Learning to Rank Systems. Proceedings of the Web Conference 2021, Ljubljana, Slovenia.10.1145/3442381.3450078 ↗
39. Lin, T.-Y., Maire, M., Belongie, S., Hays, J., Perona, P., Ramanan, D., Dollár, P., and Zitnick, C.L. (2014, January 6–12). Microsoft COCO: Common Objects in Context. Proceedings of the European Conference on Computer Vision (ECCV), Zurich, Switzerland.10.1007/978-3-319-10602-1_48 ↗
40. Zhang, S., Chi, C., Yao, Y., Lei, Z., and Li, S.Z. (2020, January 13–19). Bridging the Gap Between Anchor-Based and Anchor-Free Detection via Adaptive Training Sample Selection. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA.10.1109/CVPR42600.2020.00978 ↗
41. Wang, C.-Y., Bochkovskiy, A., and Liao, H.-Y.M. (2023, January 17–24). YOLOv7: Trainable Bag-of-Freebies Sets New State-of-the-Art for Real-Time Object Detectors. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Vancouver, BC, Canada.10.1109/CVPR52729.2023.00721 ↗
42. Wang, C.-Y., Yeh, I.-H., and Liao, H.-Y.M. (2025, January 15–16). YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information. Proceedings of the European Conference on Computer Vision, London, UK.10.1007/978-3-031-72751-1_1 ↗
43. Wang, A., Chen, H., Liu, L., Chen, K., Lin, Z., Han, J., and Ding, G. (2024). YOLOv10: Real-Time End-to-End Object Detection. arXiv.
44. Khanam, R., and Hussain, M. (2024). YOLOv11: An Overview of the Key Architectural Enhancements. arXiv.