TY - GEN
T1 - A Lightweight Attention-Driven Multi-Scale Feature Fusion Network for Ship Detection in UAV Aerial Imagery
AU - Li, Yunchao
AU - Wang, Dongliang
AU - Wang, Kunlun
AU - Wang, Li
AU - Wang, Kai
AU - Wu, Weichao
N1 - Publisher Copyright:
© 2026 IEEE.
PY - 2026
Y1 - 2026
N2 - Ship detection in UAV aerial imagery is critical for maritime surveillance and coastal security. However, dynamic sea clutter, large intra-class scale variance, and limited onboard computational resources pose significant challenges to existing detectors. This paper proposes EBW-YOLO, a lightweight detection network built upon YOLOv8n, with improvements across feature extraction, fusion, and regression. Concretely, the Efficient Multi-Scale Attention (EMA) module enhances discriminative feature representation against cluttered backgrounds; the Bi-directional Feature Pyramid Network (BiFPN) replaces PANet for adaptive multi-scale feature aggregation; and the Wise-IoU (WIoU) loss refines bounding box regression by suppressing low-quality samples. Experiments on a UAV maritime dataset demonstrate that EBW-YOLO achieves 82.6% mAP@0.5, outperforming YOLOv8n by 4.3 pp, while maintaining only 3.2M parameters and 8.2 GFLOPs, confirming its practicality for real-time UAV deployment.
AB - Ship detection in UAV aerial imagery is critical for maritime surveillance and coastal security. However, dynamic sea clutter, large intra-class scale variance, and limited onboard computational resources pose significant challenges to existing detectors. This paper proposes EBW-YOLO, a lightweight detection network built upon YOLOv8n, with improvements across feature extraction, fusion, and regression. Concretely, the Efficient Multi-Scale Attention (EMA) module enhances discriminative feature representation against cluttered backgrounds; the Bi-directional Feature Pyramid Network (BiFPN) replaces PANet for adaptive multi-scale feature aggregation; and the Wise-IoU (WIoU) loss refines bounding box regression by suppressing low-quality samples. Experiments on a UAV maritime dataset demonstrate that EBW-YOLO achieves 82.6% mAP@0.5, outperforming YOLOv8n by 4.3 pp, while maintaining only 3.2M parameters and 8.2 GFLOPs, confirming its practicality for real-time UAV deployment.
KW - Attention mechanism
KW - Feature fusion
KW - Lightweight detection
KW - Ship detection
KW - UAV aerial imagery
UR - https://www.scopus.com/pages/publications/105046893100
U2 - 10.1109/MRAI70020.2026.11621520
DO - 10.1109/MRAI70020.2026.11621520
M3 - Conference contribution
AN - SCOPUS:105046893100
T3 - 2026 2nd International Conference on Mechatronics, Robotics, and Artificial Intelligence, MRAI 2026
SP - 464
EP - 467
BT - 2026 2nd International Conference on Mechatronics, Robotics, and Artificial Intelligence, MRAI 2026
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 2nd International Conference on Mechatronics, Robotics, and Artificial Intelligence, MRAI 2026
Y2 - 22 May 2026 through 24 May 2026
ER -