YOLO-Shrimp: A Lightweight Detection Model for Shrimp Feed Residues Fusing Multi-Attention Features
Abstract
1. Introduction
- Model Architecture Innovation: Introduced a RepGhost-based lightweight backbone, reducing parameters by 19.7% and GFLOPs by 14.6% while improving mAP@0.5 by 2.80 percentage points, achieving dual enhancement of efficiency and accuracy.
- Attention Mechanism Design: Proposed EnSimAM, a parameter-free multi-scale attention mechanism fusing global, local and edge responses, improving mAP@0.5 by 1.27 percentage points without extra computation and enhancing dense small-target feature extraction.
- Loss Function and Optimization: Designed EnWIoU with small-target morphological/directional constraints, improving mAP@0.5 by 1.05 percentage points and enhancing irregular uneaten feed localization.
- Dataset and Application Validation: YOLO-Shrimp validated on 3461-image real farm dataset, achieving 70.01% mAP@0.5 and 28.01% mAP@0.5:0.95 (2.08 M params, 5.5 GFLOPs), outperforming advanced lightweight detectors and verifying practical performance.
2. Related Works
3. Proposed Methods
3.1. EnSimAM
3.2. EnWIoU
3.3. RepGhost
3.4. Overall Algorithm of YOLO-Shrimp
4. Experiment and Result Analysis
4.1. Dataset and Environment
4.2. Comparative and Ablation Experiments
4.3. Visualization Analysis Experiments
4.4. Discussion on Generalization and Limitations
4.5. Practical Deployment Considerations
4.6. Failure Case Analysis
5. Conclusions and Future Work
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Conflicts of Interest
References
- Hossain, M.; Uddin, M.; Fakhruddin, A. Impacts of shrimp farming on the coastal environment of Bangladesh and approach for management. Rev. Environ. Sci. Bio/Technol. 2013, 12, 313–332. [Google Scholar] [CrossRef] [Scilit]
- Macusi, E.D.; Estor, D.E.P.; Borazon, E.Q.; Clapano, M.B.; Santos, M.D. Environmental and socioeconomic impacts of shrimp farming in the Philippines: A critical analysis using PRISMA. Sustainability 2022, 14, 2977. [Google Scholar] [CrossRef] [Scilit]
- Asmild, M.; Hukom, V.; Nielsen, R.; Nielsen, M. Is economies of scale driving the development in shrimp farming from Penaeus monodon to Litopenaeus vannamei? The case of Indonesia. Aquaculture 2024, 579, 740178. [Google Scholar] [CrossRef] [Scilit]
- Le, N.T.T.; Armstrong, C.W.; Brækkan, E.H.; Eide, A. Climatic events and disease occurrence in intensive Litopenaeus vannamei shrimp farming in the Mekong area of Vietnam. Aquaculture 2024, 587, 740867. [Google Scholar] [CrossRef] [Scilit]
- Goh, J.X.H.; Tan, L.T.H.; Law, J.W.F.; Ser, H.L.; Khaw, K.Y.; Letchumanan, V.; Lee, L.H.; Goh, B.H. Harnessing the potentialities of probiotics, prebiotics, synbiotics, paraprobiotics, and postbiotics for shrimp farming. Rev. Aquac. 2022, 14, 1478–1557. [Google Scholar] [CrossRef] [Scilit]
- Emerenciano, M.G.; Miranda-Baeza, A.; Martínez-Porchas, M.; Poli, M.A.; Vieira, F.D.N. Biofloc technology (BFT) in shrimp farming: Past and present shaping the future. Front. Mar. Sci. 2021, 8, 813091. [Google Scholar] [CrossRef] [Scilit]
- Li, M.; Tao, Z.; Lin, S.; Feng, K. LAF: Enhancing person re-identification via Latent-Assisted Feature Fusion. Alex. Eng. J. 2025, 127, 116–128. [Google Scholar]
- Nguyen, H.; Chu, L.; Harper, R.J.; Dell, B.; Hoang, H. Mangrove-shrimp farming: A triple-win approach for communities in the Mekong River Delta. Ocean Coast. Manag. 2022, 221, 106082. [Google Scholar] [CrossRef] [Scilit]
- Rahman, M.S.; Majumder, M.K.; Sujan, M.H.K.; Manjira, S. Livelihood status of coastal shrimp farmers in Bangladesh: Comparison before and during COVID-19. Aquac. Rep. 2021, 21, 100895. [Google Scholar] [CrossRef] [Scilit]
- Phong, T.N.; Thang, V.T.; Hoai, N.T. What motivates farmers to accept good aquaculture practices in development policy? Results from choice experiment surveys with small-scale shrimp farmers in Vietnam. Econ. Anal. Policy 2021, 72, 454–469. [Google Scholar] [CrossRef] [Scilit]
- Wang, Y.; Chen, Z.; Chang, Z.; Meng, G.; Li, J. Ecological and economic analysis for different shrimp farming models in China. Aquac. Int. 2023, 31, 1941–1958. [Google Scholar] [CrossRef] [Scilit]
- Li, M.; Tao, Z.; Lin, S.; Feng, K. Mix-net: Hybrid attention/diversity network for person re-identification. Electronics 2024, 13, 1001. [Google Scholar]
- Duy, D.T.; Trung, T.Q.; Lan, T.H.P.; Berg, H.; Thi Da, C. Assessment of the impacts of social capital on the profit of shrimp farming production in the Mekong Delta, Vietnam. Aquac. Econ. Manag. 2022, 26, 152–170. [Google Scholar] [CrossRef] [Scilit]
- Li, M.; Tao, Z.; Yan, W.; Lin, S.; Feng, K.; Zhang, Z.; Jing, Y. Apnet: Lightweight network for apricot tree disease and pest detection in real-world complex backgrounds. Plant Methods 2025, 21, 4. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Zou, Z.; Chen, K.; Shi, Z.; Guo, Y.; Ye, J. Object detection in 20 years: A survey. Proc. IEEE 2023, 111, 257–276. [Google Scholar] [CrossRef] [Scilit]
- Mao, W.L.; Wang, C.C.; Chou, P.H.; Liu, Y.T. Automated defect detection for mass-produced electronic components based on YOLO object detection models. IEEE Sensors J. 2024, 24, 26877–26888. [Google Scholar] [CrossRef] [Scilit]
- Xie, X.; Cheng, G.; Wang, J.; Yao, X.; Han, J. Oriented R-CNN for object detection. In Proceedings of the IEEE/CVF International Conference on Computer Vision, Montreal, QC, Canada, 11–17 October 2021; pp. 3520–3529. [Google Scholar]
- Jiang, P.; Ergu, D.; Liu, F.; Cai, Y.; Ma, B. A Review of Yolo algorithm developments. Procedia Comput. Sci. 2022, 199, 1066–1073. [Google Scholar] [CrossRef] [Scilit]
- Liu, W.; Anguelov, D.; Erhan, D.; Szegedy, C.; Reed, S.; Fu, C.Y.; Berg, A.C. Ssd: Single shot multibox detector. In Proceedings of the European Conference on Computer Vision, Amsterdam, The Netherlands, 11–14 October 2016; Springer: Berlin/Heidelberg, Germany, 2016; pp. 21–37. [Google Scholar]
- Cheng, G.; Yuan, X.; Yao, X.; Yan, K.; Zeng, Q.; Xie, X.; Han, J. Towards large-scale small object detection: Survey and benchmarks. IEEE Trans. Pattern Anal. Mach. Intell. 2023, 45, 13467–13488. [Google Scholar] [CrossRef] [Scilit]
- Giri, K.J. SO-YOLOv8: A novel deep learning-based approach for small object detection with YOLO beyond COCO. Expert Syst. Appl. 2025, 280, 127447. [Google Scholar]
- Hussain, M. YOLO-v1 to YOLO-v8, the rise of YOLO and its complementary nature toward digital manufacturing and industrial defect detection. Machines 2023, 11, 677. [Google Scholar] [CrossRef] [Scilit]
- Diwan, T.; Anirudh, G.; Tembhurne, J.V. Object detection using YOLO: Challenges, architectural successors, datasets and applications. Multimed. Tools Appl. 2023, 82, 9243–9275. [Google Scholar] [PubMed]
- Hussain, M. Yolov1 to v8: Unveiling each variant–a comprehensive review of yolo. IEEE Access 2024, 12, 42816–42833. [Google Scholar] [CrossRef] [Scilit]
- Wang, C.Y.; Yeh, I.H.; Liao, H.Y.M. YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information. In Proceedings of the European Conference on Computer Vision (ECCV), Milan, Italy, 29 September–4 October 2024; Springer: Berlin/Heidelberg, Germany, 2024; pp. 1–17. [Google Scholar] [CrossRef] [Scilit]
- Khanam, R.; Hussain, M. Yolov11: An overview of the key architectural enhancements. arXiv 2024, arXiv:2410.17725. [Google Scholar] [CrossRef] [Scilit]
- Kotthapalli, M.; Ravipati, D.; Bhatia, R. YOLOv1 to YOLOv11: A Comprehensive Survey of Real-Time Object Detection Innovations and Challenges. arXiv 2025, arXiv:2508.02067. [Google Scholar] [CrossRef] [Scilit]
- Sinha, D.; El-Sharkawy, M. Thin mobilenet: An enhanced mobilenet architecture. In Proceedings of the 2019 IEEE 10th Annual Ubiquitous Computing, Electronics & Mobile Communication Conference (UEMCON), New York, NY, USA, 10–12 October 2019; IEEE: New York, NY, USA, 2019; pp. 0280–0285. [Google Scholar]
- Zhang, X.; Zhou, X.; Lin, M.; Sun, J. Shufflenet: An extremely efficient convolutional neural network for mobile devices. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, 18–22 June 2018; pp. 6848–6856. [Google Scholar]
- Li, Y.; Liu, J.; Wang, L. Lightweight network research based on deep learning: A review. In Proceedings of the 2018 37th Chinese Control Conference (CCC), Wuhan, China, 25–27 July 2018; IEEE: New York, NY, USA, 2018; pp. 9021–9026. [Google Scholar]
- Wan, L. Research on the Construction of Fine-Grained Recognition Model of Genuine Medicinal Materials Images Based on GhostNet. In Proceedings of the 2025 6th International Conference on Computer Vision, Image and Deep Learning (CVIDL), Changsha, China, 22–24 May 2026; IEEE: New York, NY, USA, 2025; pp. 910–914. [Google Scholar]
- Corvino, F. The Compound Injustice of the EU Carbon Border Adjustment Mechanism (CBAM). Ethics Policy Environ. 2025, 28, 26–45. [Google Scholar]
- Gao, J.; Guo, Y.; Gao, G. Sim-ConvFormer: A lightweight fault diagnosis framework incorporating SimAM and external attention. J. Supercomput. 2025, 81, 1–28. [Google Scholar]
- Trinh, C.D.; Do Le, T.M.; Do, T.H.; Bui, N.M.; Nguyen, T.H.; Ngo, Q.U.; Ngo, P.T.; Bui, D.T. Improving YOLOv8 Deep leaning model in rice disease detection by using Wise-IoU loss function. J. Meas. Control Autom. 2025, 29, 1–6. [Google Scholar]
- Lun, H.; Lu, K.; Hou, L.; Wang, S.; Xue, J. Beyond 3D: Generic IoU for 3D Object Detection. IEEE Trans. Multimed. 2025, 27, 6516–6528. [Google Scholar]
- Ma, M.; Zhang, X.; Bin, W.; Wang, Q.; Che, K. Charging Out Rectangle: Quadrangle IoU for Scene Text Detection under Balanced Supervision. In Proceedings of the 4th International Conference on Computer, Artificial Intelligence and Control Engineering, Heifei, China, 10–12 January 2025; pp. 581–589. [Google Scholar]
- Liu, G.; Jin, C.; Ni, Y.; Yang, T.; Liu, Z. UCIW-YOLO: Multi-category and high-precision obstacle detection model for agricultural machinery in unstructured farmland environments. Expert Syst. Appl. 2025, 294, 128686. [Google Scholar]
- Che, J. Research on an Improved Algorithm for Small Object Detection in Remote Sensing Ground Images Based on SW-YOLOv9. In Proceedings of the 2025 44th Chinese Control Conference (CCC), Chongqing, China, 28–30 July 2025; IEEE: New York, NY, USA, 2025; pp. 7416–7423. [Google Scholar]
- Hachicha, M.; Ben Halima, R.; Frikha, T. PreSA: An intelligent blockchain-based platform for monitoring and predicting water quality for smart aquaculture. Neural Comput. Appl. 2025, 37, 22129–22140. [Google Scholar]
- Chandran, P.J.I.; Khalil, S.H.A.; Hashir, P. Smart Technologies in Aquaculture: An Integrated IoT, AI, and Blockchain Framework for Sustainable Growth. Aquac. Eng. 2025, 111, 102584. [Google Scholar] [CrossRef] [Scilit]
- Mdoe, C.N.; Mahonge, C.P.; Ngowi, E.E. Drivers and barriers to climate-smart aquaculture adoption: Insights from Mwanza and Mara Regions, Tanzania. Aquac. Int. 2025, 33, 341. [Google Scholar] [CrossRef] [Scilit]
- Tina, F.W.; Afsarimanesh, N.; Nag, A.; Alahi, M.E.E. Integrating AIoT technologies in aquaculture: A systematic review. Future Internet 2025, 17, 199. [Google Scholar] [CrossRef] [Scilit]
- Akram, W.; Din, M.U.; Saoud, L.S.; Hussain, I. A review of generative AI in aquaculture: Applications, case studies and challenges for smart and sustainable farming. Aquac. Eng. 2025, 112, 102637. [Google Scholar] [CrossRef] [Scilit]
- Ren, S.; He, K.; Girshick, R.; Sun, J. Faster R-CNN: Towards real-time object detection with region proposal networks. In Proceedings of the Advances in Neural Information Processing Systems 28 (NIPS 2015), Montreal, QC, Canada, 7–12 December 2015. [Google Scholar]
- Zhao, Y.; Lv, W.; Xu, S.; Wei, J.; Wang, G.; Dang, Q.; Liu, Y.; Chen, J. Detrs beat yolos on real-time object detection. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA, 17–18 June 2024; pp. 16965–16974. [Google Scholar]
- Mathew, M.P.; Mahesh, T.Y. Leaf-based disease detection in bell pepper plant using YOLO v5. Signal Image Video Process. 2022, 16, 841–847. [Google Scholar] [CrossRef] [Scilit]
- Qadri, S.A.A.; Huang, N.F.; Wani, T.M.; Bhat, S.A. Plant disease detection and segmentation using end-to-end yolov8: A comprehensive approach. In Proceedings of the 2023 IEEE 13th International Conference on Control System, Computing and Engineering (ICCSCE), Penang, Malaysia, 25–26 August 2023; IEEE: New York, NY, USA, 2023; pp. 155–160. [Google Scholar]
- Wang, A.; Chen, H.; Liu, L.; Chen, K.; Lin, Z.; Han, J. Yolov10: Real-time end-to-end object detection. Adv. Neural Inf. Process. Syst. 2024, 37, 107984–108011. [Google Scholar]
- Sekar, K.; Dheepa, T.; Sheethal, R.; Suvarna Smita, R.; Teja, V.D. Efficient Object Detection on Low-Resource Devices Using Lightweight MobileNet-SSD. In Proceedings of the 2025 International Conference on Intelligent Systems and Computational Networks (ICISCN), Bidar, India, 24–25 January 2025; IEEE: New York, NY, USA, 2025; pp. 1–6. [Google Scholar]
- Li, C.; Zhao, R.; Wang, Z.; Xu, H.; Zhu, X. Remdet: Rethinking efficient model design for uav object detection. In Proceedings of the AAAI Conference on Artificial Intelligence, Philadelphia, PA, USA, 25 February–4 March 2025; Volume 39, pp. 4643–4651. [Google Scholar]










| Input Image I, Ground-truth boxes |
| Output Predicted boxes |
| Initialization |
| Training Phase |
| for to do: |
| for in dataloader do: |
| Optimizer.step() |
| end for |
| end for |
| Inference Phase |
| Model.eval() |
| return |
| Hardware/Software | Configuration |
|---|---|
| GPU | NVIDIA RTX 4090 (24 GB) (Nvidia, Santa Clara, CA, USA) |
| CPU | Intel(R) Xeon(R) Gold 5418Y (10 Cores) (Intel, Santa Clara, CA, USA) |
| Memory | 120 GB |
| Operating System | Ubuntu 22.04 |
| Deep Learning Framework | PyTorch 1.13.1 |
| Python Version | 3.9 |
| Model | mAP@0.5 | mAP@0.95 | P | R | Params (M) | GFLOPs |
|---|---|---|---|---|---|---|
| SSD [19] | 6.52 | 1.41 | 18.50 | 23.30 | 6.13 | 3.04 |
| Faster R-CNN [44] | 16.68 | 4.46 | 30.24 | 24.53 | 18.93 | 41.80 |
| RT-DETR [45] | 42.40 | 13.10 | 47.40 | 51.05 | 19.00 | 54.09 |
| YOLOv5n [46] | 62.60 | 23.06 | 67.07 | 62.34 | 2.19 | 5.92 |
| YOLOv8n [47] | 64.57 | 24.13 | 68.37 | 64.22 | 2.69 | 6.94 |
| YOLOv9n [25] | 63.57 | 23.91 | 67.69 | 63.07 | 1.77 | 6.7 |
| YOLOv10n [48] | 60.93 | 22.72 | 63.98 | 60.55 | 2.70 | 8.39 |
| MobileNet-SSD [49] | 18.52 | 4.86 | 32.15 | 30.72 | 2.35 | 3.20 |
| RTMDet-tiny [50] | 61.85 | 22.30 | 66.20 | 61.50 | 2.80 | 4.00 |
| YOLOv11n [26] | 65.88 | 25.08 | 69.28 | 64.25 | 2.59 | 6.44 |
| Ours | 70.01 | 28.01 | 72.53 | 65.88 | 2.08 | 5.50 |
| R | E | U | mAP@0.5 | mAP@0.95 | P | R | Params (M) | GFLOPs |
|---|---|---|---|---|---|---|---|---|
| 65.88 | 25.08 | 69.28 | 64.25 | 2.59 | 6.44 | |||
| ✓ | 68.68 | 27.39 | 70.91 | 66.09 | 2.08 | 5.50 | ||
| ✓ | 67.15 | 25.75 | 70.02 | 64.92 | 2.59 | 6.44 | ||
| ✓ | 66.93 | 25.72 | 69.66 | 64.84 | 2.59 | 6.44 | ||
| ✓ | ✓ | ✓ | 70.01 | 28.01 | 72.53 | 65.88 | 2.08 | 5.50 |
| Model/Strategy | mAP@0.5 | mAP@0.95 | P | R | Params (M) | GFLOPs |
|---|---|---|---|---|---|---|
| No Augmentation | 65.32 | 24.17 | 68.21 | 61.05 | 2.08 | 5.50 |
| Full Data Augmentation | 68.15 | 26.42 | 70.38 | 63.72 | 2.08 | 5.50 |
| Local Data Augmentation | 69.24 | 27.18 | 71.65 | 64.90 | 2.08 | 5.50 |
| Full + Local Augmentation | 70.01 | 28.01 | 72.53 | 65.88 | 2.08 | 5.50 |
| Module | Component | mAP@0.5 | mAP@0.5:0.95 | mAP@0.5 |
|---|---|---|---|---|
| Baseline | – | 65.88 | 25.08 | – |
| EnSimAM | + Global | 66.58 | 25.41 | +0.70 |
| + Global + Local | 66.88 | 25.63 | +1.00 | |
| + Global + Local + Edge | 67.15 | 25.75 | +1.27 | |
| EnWIoU | + WIoU (Baseline) | 65.88 | 25.08 | – |
| + WIoU + Aspect | 66.30 | 25.33 | +0.42 | |
| + WIoU + Direction | 66.13 | 25.21 | +0.25 | |
| + WIoU + Aspect + Direction | 66.93 | 25.72 | +1.05 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Hou, T.; Miao, X.; Wang, Z.; Zhang, Y.; He, Z.; Sun, Y.; Wang, W.; Ren, P. YOLO-Shrimp: A Lightweight Detection Model for Shrimp Feed Residues Fusing Multi-Attention Features. Sensors 2026, 26, 791. https://doi.org/10.3390/s26030791
Hou T, Miao X, Wang Z, Zhang Y, He Z, Sun Y, Wang W, Ren P. YOLO-Shrimp: A Lightweight Detection Model for Shrimp Feed Residues Fusing Multi-Attention Features. Sensors. 2026; 26(3):791. https://doi.org/10.3390/s26030791
Chicago/Turabian StyleHou, Tianwen, Xinying Miao, Zhenghan Wang, Yi Zhang, Zhipeng He, Yifei Sun, Wei Wang, and Ping Ren. 2026. "YOLO-Shrimp: A Lightweight Detection Model for Shrimp Feed Residues Fusing Multi-Attention Features" Sensors 26, no. 3: 791. https://doi.org/10.3390/s26030791
APA StyleHou, T., Miao, X., Wang, Z., Zhang, Y., He, Z., Sun, Y., Wang, W., & Ren, P. (2026). YOLO-Shrimp: A Lightweight Detection Model for Shrimp Feed Residues Fusing Multi-Attention Features. Sensors, 26(3), 791. https://doi.org/10.3390/s26030791

