Source-linked AI summary
The Unbearable Weight: Scaling Models and Methods for UAV Audio Classification
Andrew P. Berg, Qian Zhang, Mia Y. Wang
TL;DR
To address the compute and memory burden of large pretrained models in resource-constrained UAV settings, the paper compares architectures and fine-tuning methods for UAV audio classification. On its 3,100-clip, 31-class dataset, fine-tuned CNNs outperform transformers in both accuracy and efficiency.
Problem
The paper asks when large pretrained models’ computational and memory burden is justified for UAV audio classification in resource-constrained settings.
Source: §1 · pp. 2–3Method
The study compares CNN and transformer backbones across full, classifier-only, and four parameter-efficient tuning methods, measuring accuracy, training time, trainable-parameter share, and inference memory on 3,100 clips from 31 classes.
Source: §1 · p. 2Results
Across the sweep, fine-tuned CNNs outperform transformers in accuracy and efficiency; EfficientNet-B7 with batch-norm tuning and three-fold augmentation attains 97.65% ± 0.30 validation accuracy.
Source: §5 · pp. 16–17Takeaways & Limitations
For UAV audio classification under data scarcity, scaling the fine-tuning method rather than the model gives the best balance of accuracy, training efficiency, and inference-time footprint.
Source: §5 · p. 18Evidence is scoped to a 3,100-clip UAV dataset, with ViT using an ImageNet-pretrained checkpoint rather than audio pretraining.
Source: §3.7 · p. 10
Abstract
from arXiv · showhide
As unmanned aerial vehicles (UAVs) become increasingly prevalent in consumer and defense settings, classifying them reliably from limited, modality-specific data is an urgent challenge. The dominant approach, large pretrained networks fully fine-tuned on task data, carries a substantial computational and memory weight that is hard to bear in resource-constrained UAV deployments, where edge inference and rapid retraining for emerging platforms are both required. This paper systematically scales across both model architectures and fine-tuning methods for UAV audio classification, asking when that weight is justified and when lighter alternatives prevail. Using a custom dataset of 3,100 audio clips spanning 31 drone classes, we evaluate transformer (ViT, AST) and convolutional (custom CNN, ResNet-18/152, MobileNet-V3-S/L, EfficientNet-B0/B7) backbones under full fine-tuning, classifier-only fine-tuning, and four parameter-efficient fine-tuning (PEFT) methods: SSF, IA3, OFT, and selective batch-norm tuning. All configurations are evaluated with 5-fold cross-validation across accuracy, training time, trainable-parameter share, and inference-time memory footprint. Selective batch-norm fine-tuning of EfficientNet-B7 with three-fold augmentations achieves the highest validation accuracy (97.65% +- 0.30) while updating under 0.5% of model parameters. Across the sweep, lightweight CNNs consistently outperform transformers on both accuracy and efficiency. For UAV audio classification under data scarcity, scaling the method outperforms scaling the model.
1 Introduction
UAV audio can identify a drone platform from its acoustic signature, but data scarcity and the cost of adapting large pretrained models make reliable classification challenging. The paper compares model architectures and fine-tuning methods to assess accuracy and efficiency trade-offs on a 31-class UAV audio task.
1 Introduction
- Audio captures platform-specific rotor, motor, and aerodynamic signatures that can distinguish UAV models, while outdoor recordings also include environmental noise.Unlike telemetry or mission content, the recordings represent the drone itself.Source: §1 · p. 2
- Limited, costly-to-collect UAV audio makes training large networks from scratch impractical, while full fine-tuning is computationally heavy and slow to retrain.The interaction between model scale, parameter-efficient fine-tuning, and targeted augmentation on small-data audio tasks is not well-characterized.Source: §1 · p. 2
- The study compares CNN and transformer backbones at multiple sizes under full, classifier-only, and four parameter-efficient fine-tuning methods.It reports accuracy, training time, trainable-parameter share, and inference-time memory footprint on a 3,100-clip, 31-class dataset.Source: §1 · p. 2
- The central question is when large pretrained models’ compute and memory costs are justified versus when smaller models or lighter fine-tuning are better engineering choices.The introduction frames research questions around CNN-versus-transformer performance and accuracy-to-efficiency trade-offs under data scarcity.Source: §1 · p. 3
2 Related Work
Related work motivates UAV audio as an accessible classification modality and situates this study among prior UAV-audio, deep-learning, and transfer-learning research. Prior findings include a small-dataset comparison favoring a custom CNN and broader evidence for both CNN and transformer approaches.
Previous Work
- The prior UAV-audio study used a 9-class dataset and found the custom CNN slightly outperformed the pretrained AST; the dataset later grew to 31 classes.The current dataset comprises 15,500 seconds of audio, enabling a look at scaling for UAV classification.Source: §2 · p. 3
UAV Classification
Deep Learning & Audio Classification
- Prior work identifies CNNs as common in audio classification, while another comparison reports transformers can significantly outperform CNNs and supports transfer learning.The paper narrows its model selection to CNNs and transformers.Source: §2 · p. 4
Transfer Learning
- Transfer learning initializes a related task with weights from a model pretrained on a large dataset, and cross-modal transfer is an alternative when domain-specific pretrained networks are unavailable [8].The review describes it as reducing training and dataset-scarcity constraints.Source: §2 · p. 4
3 Methodology
The study collects a balanced, multi-environment UAV audio dataset and converts recordings into log-scaled mel-spectrograms, augmenting training data while keeping validation unchanged. It compares pretrained and custom backbones under full, classifier-only, and parameter-efficient fine-tuning strategies.
3.1 Data Collection
- 3,100 five-second recordings from 31 UAVs form a perfectly balanced dataset collected across indoor and outdoor environments in three U.S. locations.Recordings capture rotor, motor, and aerodynamic signatures during flight.Source: §3.1 · pp. 4–5
3.2 Technical Environment
- Experiments ran on a server with two RTX 4090 GPUs, but used one GPU because distributed runs took longer.The authors report code and training runs as publicly available.Source: §3.2 · p. 5
3.3 Feature Extraction
- Mel-spectrograms represent frequency content over time on a human-perception scale; extraction uses a 16,000 Hz sampling rate, 1,024-point FFT window, 128 hop length, and 128 mel bands.They are computed during training time.Source: §3.3 · p. 6
3.4 Data Augmentations
- Three deterministic compositions of time stretch and harmonic distortion are generated per training sample, tripling training data while leaving validation unaugmented.The authors note augmented data may not fully reflect real-world audio.Source: §3.4 · p. 7
3.5 Training
- Full fine-tuning updates all model parameters, whereas classifier fine-tuning trains only the final classifier block as an ablation of pretrained generalization.The paper describes classifier fine-tuning as not reliably large enough to generalize effectively.Source: §3.5 · pp. 7–8
3.6 PEFT
- SSF scales and shifts features across model families, IA3 and OFT target transformers, and batch-norm fine-tuning targets CNNs; the custom CNN is randomly initialized and evaluated only with full fine-tuning.Classifier fine-tuning is an ablation for the other backbones, while full fine-tuning is reported as an adaptation upper bound.Source: §3.6 · pp. 8–9
3.7 Models
- The model sweep includes a three-layer custom CNN, ResNet-18/152, MobileNet-V3 small/large, EfficientNet-B0/B7, ViT, and AST backbones.ViT uses an ImageNet-fine-tuned checkpoint, while the AST checkpoint was trained on AudioSet.Source: §3.7 · pp. 10, 11, 13
4 Experiments & Results
Experiments compare model architectures and fine-tuning configurations under cross-validation, tracking accuracy and efficiency measures. EfficientNet-B7 with batch-norm fine-tuning and three-fold augmentation has the highest validation accuracy, while CNNs are more efficient than transformers.
4 Experiments & Results
- All models used five-fold cross-validation and the same computational budget; training time, trainable-parameter share, and approximate memory footprint were tracked alongside accuracy.Each fold used an 80% training and 20% test split.Source: §4 · p. 13
- CNNs scale evenly with time to performance, whereas AST and ViT train longer with lower accuracy, making them less efficient.Training-time comparisons are shown in Fig. 8.Source: §4 · p. 14, Figure 8 · p. 16
- 97.65%±0.30 validation accuracy was achieved by EfficientNet-B7 with batch-norm fine-tuning and three-fold augmentation.Table 1 reports the five-fold cross-validation results.Source: §4 · p. 13, Table 1 · p. 15
5 Discussion
The discussion finds that CNNs, especially EfficientNet-B7 with selective batch-norm tuning, offer the strongest accuracy-efficiency balance in this evaluation. It attributes the CNN advantage to spectrogram locality and limited data, while identifying continual adaptation and larger-scale evaluation as future directions.
RQ1: CNNs vs. transformers
- 97.65% ± 0.30 for EfficientNet-B7 exceeds AST with OFT at 96.74%±0.24; the CNN trains in roughly one-seventh the wall time per fold and uses 243 MB versus 333 MB.Source: §5 · p. 16
RQ2: best accuracy-to-efficiency configuration
- 97.65% ± 0.30 validation accuracy, 0.49% of model parameters trained, and a 243 MB inference-time memory footprint make EfficientNet-B7 with batch-norm fine-tuning and three-fold augmentation the strongest configuration.The configuration updates fewer than 0.5% of parameters.Source: §5 · pp. 16–17
RQ3: within-family backbone scaling
- Larger backbones make PEFT effective across families, while smaller backbones require full fine-tuning to remain competitive; MobileNetV3 is an exception to the scaling pattern.EfficientNet-B7 peaks under batch-norm fine-tuning, while EfficientNet-B0 peaks under full fine-tuning.Source: §5 · p. 17
Why CNNs outperform transformers
- The discussion attributes CNN performance to locality suited to mel-spectrograms, weak alignment between inherited pretraining and UAV audio, and batch-norm adaptation of channel statistics.The reported custom CNN result is 94.32% ± 1.09% with three-fold augmentation, illustrating the boundary case without a strong pretrained signal.Source: §5 · p. 17
Future directions
- Future work targets continual adaptation, multimodal fallback, larger-scale transformer crossover tests, and separating feature transfer from distribution-level adaptation.The paper says these questions remain open, including how PEFT methods depend on data scale.Source: §5 · p. 18
Conclusion
- For UAV audio classification under data scarcity, the conclusion favors lightweight CNN backbones with selective fine-tuning over large transformers with full fine-tuning.The paper characterizes scaling the method, rather than the model, as giving the best balance of accuracy, training efficiency, and inference-time footprint.Source: §5 · p. 18
6 Declarations
The declarations report funding, data and code availability, no conflict of interest, and that ethical approval was not applicable.
- The work is supported by the South Carolina Research Authority (SCRA).Source: §6 · p. 18
- The code and model training logs are available online, and the authors declare no conflict of interest.Source: §6 · p. 18
- The authors state that ethical approval was not applicable.Source: §6 · p. 18
Appendix A Data Collection
Appendix A identifies a dataset table describing the 31 UAV audio classes and their collection sites.
- Table A1 is described as listing the 31 UAV audio dataset classes with collection sites.The appendix description also names collection site, manufacturer, and model name as table fields.Source: Appendix A · p. 22, Table A1 · p. 23
Appendix B Memory Footprint
The appendix measures model-specific inference-time memory footprint. It reports little correlation between model accuracy and memory footprint, while additive PEFT methods slightly raise the footprint.
Appendix B Memory Footprint
- The reported memory footprint is an inference-time measure for each model.The paper identifies it as relevant to edge devices and overall scale.Source: Appendix B · p. 24
- The memory estimate uses bytes per parameter; for float32, the paper specifies four bytes per parameter.Source: Appendix B · p. 24
- Model accuracy and memory footprint show little correlation, and additive PEFT methods such as SSF, IA3, and OFT slightly raise the footprint.Fig. B1 illustrates the comparison, and Table B2 provides the model and PEFT breakdown.Source: Appendix B · p. 24, Figure B1 · p. 24
Appendix C Training Time vs. Augmentations
Data augmentations significantly increase training times, as shown in Fig. C2.
Appendix C Training Time vs. Augmentations
- Data augmentations significantly increase training times.Source: Appendix C · p. 24
Appendix D Training Time Table
Table D3 reports average training runtime across models and PEFT configurations. Blank entries mark PEFT incompatibility.
Appendix D Training Time Table
- Table D3 reports average training runtime in minutes for different models under various PEFT configurations.Source: Appendix D · p. 24
- Blank entries in Table D3 indicate PEFT incompatibility.Source: Appendix D · p. 24
Appendix E Trainable Parameter Percentage Table
Table E4 presents trainable-parameter percentages for each model and PEFT configuration. Full fine-tuning trains all parameters and is therefore 100%.
Appendix E Trainable Parameter Percentage Table
- Full fine-tuning trains all parameters, corresponding to 100%.Source: Appendix E · p. 25
- Table E4 reports trainable-parameter percentages for each model and PEFT configuration.Source: Appendix E · p. 25, Table E4 · p. 27
Related papers
This paper cites
- Attention Is All You Need2017
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale2020
- PyTorch: An Imperative Style, High-Performance Deep Learning Library2019
- MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications2017
- Deep Residual Learning for Image Recognition2015