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

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Berg, Andrew P., Zhang, Qian, Wang, Mia Y.
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915445682995200
author Berg, Andrew P.
Zhang, Qian
Wang, Mia Y.
author_facet Berg, Andrew P.
Zhang, Qian
Wang, Mia Y.
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2506_11049
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle 15,500 Seconds: Lean UAV Classification Using EfficientNet and Lightweight Fine-Tuning
Berg, Andrew P.
Zhang, Qian
Wang, Mia Y.
Machine Learning
Artificial Intelligence
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.
title 15,500 Seconds: Lean UAV Classification Using EfficientNet and Lightweight Fine-Tuning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2506.11049