发现语音模型中变换器层高度冗余,可大幅压缩而不损失性能。
How Redundant Is the Transformer Stack in Speech Representation Models?
- 用三种相似性度量分析层间冗余,发现分块相似结构。
- 剪枝40%层仍保持95%以上预测能力,无需再训练。
- 知识蒸馏使模型缩小95%-98%,推理提速达94%。
自监督语音表示模型,尤其是基于Transformer架构的模型,在语音识别、说话人识别和情绪检测等任务中表现优异。近期研究发现变换器层间存在高冗余性,且具备显著剪枝潜力。本文针对基于Transformer的语音表示模型,采用余弦相似度、中心核对齐和互最近邻对齐三种度量方法,深入分析层间相似性。结果揭示出具有高相似性的块状结构,表明存在两个主要处理阶段,且层间冗余显著。我们证明了无需后训练即可有效剪枝该类模型,最多可减少40%的变压器层,同时保持超过95%的模型预测能力。此外,通过知识蒸馏方法,以模仿层替代整个变压器堆栈,使网络规模缩小95%-98%,推理时间缩短高达94%。这一显著的计算负载降低未带来明显性能损失,表明对于下游应用而言,变压器堆栈几乎完全冗余。
原文摘要 · Abstract (English)
Self-supervised speech representation models, particularly those leveraging transformer architectures, have demonstrated remarkable performance across various tasks such as speech recognition, speaker identification, and emotion detection. Recent studies on transformer models revealed a high redundancy between layers and the potential for significant pruning, which we will investigate here for transformer-based speech representation models. We perform a detailed analysis of layer similarity in speech representation models using three similarity metrics: cosine similarity, centered kernel alignment, and mutual nearest-neighbor alignment. Our findings reveal a block-like structure of high similarity, suggesting two main processing steps and significant redundancy of layers. We demonstrate the effectiveness of pruning transformer-based speech representation models without the need for post-training, achieving up to 40% reduction in transformer layers while maintaining over 95% of the model's predictive capacity. Furthermore, we employ a knowledge distillation method to substitute the entire transformer stack with mimicking layers, reducing the network size 95-98% and the inference time by up to 94%. This substantial decrease in computational load occurs without considerable performance loss, suggesting that the transformer stack is almost completely redundant for downstream applications of speech representation models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。