arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.03424cs.LGcs.CL

面向自动语音识别的神经语言模型剪枝

Neural Language Model Pruning for Automatic Speech Recognition

Apple(苹果公司)

Leonardo Emili, Thiago Fraga-Silva, Ernest Pusateri, Markus Nußbaum-Thom, Youssef Oualil

AI总结:

该研究针对自动语音识别的Transformer语言模型剪枝框架的准则、方法、调度器展开深度分析,提出适配增量压缩的低秩近似变体,验证了数据驱动剪枝、增量剪枝及低秩近似的相关优势。

摘要:

我们研究了应用于自动语音识别任务的基于Transformer的神经网络语言模型的剪枝方法。我们从剪枝框架的三个维度——即准则、方法和调度器——展开探索,分析它们在准确率和推理速度方面的贡献。据我们所知,针对大规模识别系统的此类深度分析尚未在现有文献中被报道。此外,我们提出了一种适用于增量式压缩模型的低秩近似变体,可生成多种目标尺寸的模型。实验结果包括:a)数据驱动剪枝在多种场景下表现优于幅度驱动剪枝;b)与一次性剪枝相比,增量剪枝能取得更高准确率,尤其是在目标尺寸更小的情况下;c)对于中等程度的压缩,低秩近似在模型体积缩减和推理速度提升之间实现了最佳权衡。

英文摘要:

We study model pruning methods applied to Transformer-based neural network language models for automatic speech recognition. We explore three aspects of the pruning frame work, namely criterion, method and scheduler, analyzing their contribution in terms of accuracy and inference speed. To the best of our knowledge, such in-depth analyses on large-scale recognition systems has not been reported in the literature. In addition, we propose a variant of low-rank approximation suitable for incrementally compressing models, and delivering multiple models with varied target sizes. Among other results, we show that a) data-driven pruning outperforms magnitude-driven in several scenarios; b) incremental pruning achieves higher accuracy compared to one-shot pruning, especially when targeting smaller sizes; and c) low-rank approximation presents the best trade-off between size reduction and inference speed-up for moderate compression.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑