用极简模型实现接近大模型的音频事件检测效果
Exploring Performance-Complexity Trade-Offs in Sound Event Detection Models
- 轻量化卷积网络配合序列模块,实现帧级预测
- 仅需5%参数量,性能媲美顶尖Transformer模型
- 适配不同设备需求,适合边缘计算场景
本文致力于开发适用于声音事件检测任务的低复杂度神经网络。目标是在计算资源大幅降低的前提下,仍能与当前最先进的大型模型性能相当。研究发现,可通过调整卷积步长、移除全局池化层,并在分类头前加入序列模型,将此前用于音频标记的轻量卷积模型有效迁移至事件检测任务(需帧级预测)。系统性实验表明,最优序列模型的选择取决于具体应用中对复杂度指标的优先级。此外,还探讨了知识蒸馏等增强训练策略的影响。最终结果表明,在优化训练策略支持下,模型性能可与先进Transformer模型相媲美,同时参数量仅为后者的约5%。所有预训练模型与代码已开源,以促进低复杂度声音事件检测领域的后续研究。
原文摘要 · Abstract (English)
We target the problem of developing new low-complexity networks for the sound event detection task. Our goal is to meticulously analyze the performance-complexity trade-off, aiming to be competitive with the large state-of-the-art models, at a fraction of the computational requirements. We find that low-complexity convolutional models previously proposed for audio tagging can be effectively adapted for event detection (which requires frame-wise prediction) by adjusting convolutional strides, removing the global pooling, and, importantly, adding a sequence model before the (now frame-wise) classification heads. Systematic experiments reveal that the best choice for the sequence model type depends on which complexity metric is most important for the given application. We also investigate the impact of enhanced training strategies such as knowledge distillation. In the end, we show that combined with an optimized training strategy, we can reach event detection performance comparable to state-of-the-art transformers while requiring only around 5% of the parameters. We release all our pre-trained models and the code for reproducing this work to support future research in low-complexity sound event detection at https://github.com/theMoro/EfficientSED.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。