Skip to main navigation Skip to search Skip to main content

A Lightweight Attention-Driven Multi-Scale Feature Fusion Network for Ship Detection in UAV Aerial Imagery

  • Yunchao Li
  • , Dongliang Wang
  • , Kunlun Wang
  • , Li Wang
  • , Kai Wang
  • , Weichao Wu*
  • *Corresponding author for this work
  • Norinco Group Test And Measuring Academy
  • Beijing Institute of Technology

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

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.

Original languageEnglish
Title of host publication2026 2nd International Conference on Mechatronics, Robotics, and Artificial Intelligence, MRAI 2026
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages464-467
Number of pages4
ISBN (Electronic)9798331551643
DOIs
Publication statusPublished - 2026
Externally publishedYes
Event2nd International Conference on Mechatronics, Robotics, and Artificial Intelligence, MRAI 2026 - Changsha, China
Duration: 22 May 202624 May 2026

Publication series

Name2026 2nd International Conference on Mechatronics, Robotics, and Artificial Intelligence, MRAI 2026

Conference

Conference2nd International Conference on Mechatronics, Robotics, and Artificial Intelligence, MRAI 2026
Country/TerritoryChina
CityChangsha
Period22/05/2624/05/26

Keywords

  • Attention mechanism
  • Feature fusion
  • Lightweight detection
  • Ship detection
  • UAV aerial imagery

Fingerprint

Dive into the research topics of 'A Lightweight Attention-Driven Multi-Scale Feature Fusion Network for Ship Detection in UAV Aerial Imagery'. Together they form a unique fingerprint.

Cite this