EISD-YOLO: Efficient Infrared Ship Detection with Param-Reduced PEMA Block and Dynamic Task Alignment
Ship detection is critical for maritime traffic management, as infrared imaging in complex marine environments encounters significant challenges, such as strong background interference and weak target features. Thus, we propose EISD-YOLO (Efficient Infrared Ship Detection-YOLO), a high-performance lightweight algorithm specifically designed for infrared ship detection. The algorithm aims to improve detection accuracy while simultaneously reducing model parameters and enhancing computational efficiency. It integrates three core architectural innovations: first, we optimized the backbone C3k2 module by replacing the traditional bottleneck with a PEMA block to significantly reduce the parameter count; second, we integrated a lightweight DS_ADNet module, using depth-wise separable convolution to reduce parameters and alleviate computational load while maintaining robust feature representation; and third, we adopted the DyTAHead detection head, which integrates classification and localization features through dynamic task alignment, thereby achieving robust performance in complex infrared ship detection scenarios. The experimental results on the IRShip dataset demonstrate that, compared with YOLOv11n, EISD-YOLO reduced the parameters by 48.83%, while [email protected], precision, and recall all increased by 1.2%. This breaks the traditional rule that lightweight models inevitably lead to reduced accuracy. Additionally, the model size reduced from 10.1 MB to 5.7 MB, which highlights its enhanced computational efficiency and practical applicability in maritime deployment scenarios.
Authors
- Huifeng Tao
- Changqi Zhou
- Jin Wei (ORCID: https://orcid.org/0000-0003-2665-278X)
- Yunsong Feng
- Siyu Wang (ORCID: https://orcid.org/0009-0002-1346-7729)
- Liping Liu (ORCID: https://orcid.org/0009-0008-2190-8428)
- Juan Chen (ORCID: https://orcid.org/0000-0002-2084-3115)
Institutions
- Anhui Agricultural University (CN)
- Anhui University (CN)
- National University of Defense Technology (CN)
- Hefei University (CN)
Publication Details
- Journal
- Photonics
- Published
- 2025-10-22
- DOI
- https://doi.org/10.3390/photonics12111044
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00