从大模型中提取轻量专用子网,实现高效音频任务部署。
Keep what you need : extracting efficient subnetworks from large audio representation models
- 用可学习的二值掩码在预训练模型层间筛选关键路径。
- 在下游任务上训练后,模型压缩率达60%以上,精度损失<2%。
- 无需重训主干,适合资源受限设备实时应用。
近年来,音频基础模型在复杂下游任务上取得显著进展,但模型规模与复杂度随之激增,导致难以在消费级设备上部署,限制了实时应用。为解决这一矛盾,本文提出一种简单而有效的方法:在预训练表示模型的层间引入可学习的二值掩码,通过稀疏性正则化损失,在不更新主干权重的前提下,端到端训练出仅针对单一任务的紧凑子网络。训练完成后,可移除被掩码的计算单元,实现显著性能提升。我们在三种基于不同骨干架构的主流音频基础模型上验证方法有效性,涵盖语音、音乐和通用音频任务,结果表明该方法在保持高精度的同时,实现60%以上的模型压缩率。代码与网页支持已开源。
原文摘要 · Abstract (English)
Recently, research on audio foundation models has witnessed notable advances, as illustrated by the ever improving results on complex downstream tasks. Subsequently, those pretrained networks have quickly been used for various audio applications. These improvements have however resulted in a considerable increase both in size and complexity of these models. Along the environmental concerns this issue raises, this prevents the deployment of such networks on consumer-level devices, and precludes their use for real-time applications. Moreover, this appears contradictory with the specificity of the tasks for which these models are used, which are often simpler compared to extracting a rich, multi-purpose representation from any type of audio data. In this paper, we address this issue with a simple, yet effective method to extract lightweight specialist subnetworks from large foundation models. Specifically, we introduce learnable binary masks in-between the layers of a pretrained representation model. When training the end-to-end model on a downstream task, we add a sparsity-inducing loss to the overall objective, hence learning a compact subnetwork specialized on a single task. Importantly, the weights of the foundation model are kept frozen, resulting into low additional training costs. Once trained, the masked computational units can then be removed from the network, implying significant performance gains. We assess our method on three widespread audio foundation models, each based on a different backbone architecture, and illustrate its effectiveness on common audio representation evaluation tasks, as well as its versatility on both speech, music, and general audio. Code for reproducing the results and supporting webpage are available at https://github.com/gnvIRCAM/Audio-representation-trimming
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。