RobustDistiller压缩模型

压缩同时更抗噪

通过多任务学习蒸馏,使压缩后的语音表征对环境噪声和混响更鲁棒。

Heitor Rodrigues, Guimaraes · Arthur, Pimentel · Avila, Anderson R. · Rezagholizadeh, Mehdi · Boxing, Chen · Falk, Tiago H.

2023

参数信息

参数压缩率
4 ×
学生模型参数
24 M
教师模型参数
95 M

技术优势

小模型不输大模型

24M参数的学生模型在多种噪声类型、噪声水平和混响时间下达到与95M教师模型相当的性能。

蒸馏同时提升鲁棒性

多任务学习目标使压缩后的表征对环境噪声和混响更加鲁棒,优于其他基准方法。

应用场景