English

15,500 Seconds: Lean UAV Classification Using EfficientNet and Lightweight Fine-Tuning

Machine Learning 2025-08-15 v4 Artificial Intelligence

Abstract

As unmanned aerial vehicles (UAVs) become increasingly prevalent in both consumer and defense applications, the need for reliable, modality-specific classification systems grows in urgency. This paper addresses the challenge of data scarcity in UAV audio classification by expanding on prior work through the integration of pre-trained deep learning models, parameter-efficient fine-tuning (PEFT) strategies, and targeted data augmentation techniques. Using a custom dataset of 3,100 UAV audio clips (15,500 seconds) spanning 31 distinct drone types, we evaluate the performance of transformer-based and convolutional neural network (CNN) architectures under various fine-tuning configurations. Experiments were conducted with five-fold cross-validation, assessing accuracy, training efficiency, and robustness. Results show that full fine-tuning of the EfficientNet-B0 model with three augmentations achieved the highest validation accuracy (95.95), outperforming both the custom CNN and transformer-based models like AST. These findings suggest that combining lightweight architectures with PEFT and well-chosen augmentations provides an effective strategy for UAV audio classification on limited datasets. Future work will extend this framework to multimodal UAV classification using visual and radar telemetry.

Keywords

Cite

@article{arxiv.2506.11049,
  title  = {15,500 Seconds: Lean UAV Classification Using EfficientNet and Lightweight Fine-Tuning},
  author = {Andrew P. Berg and Qian Zhang and Mia Y. Wang},
  journal= {arXiv preprint arXiv:2506.11049},
  year   = {2025}
}
R2 v1 2026-07-01T03:14:14.759Z