一个模型同时搞定语音识别、音频分类和说话人验证
MT2KD: Towards A General-Purpose Encoder for Speech, Speaker, and Audio Events
- 两阶段训练:先用知识蒸馏对齐三个专用模型特征,再微调
- 仅用66M参数,在三项任务上表现接近专用模型
- 适合需要轻量通用音频编码器的场景
随着深度学习发展,端到端语音与音频处理单任务模型性能持续提升。但构建在多个任务上表现优异的通用模型仍具挑战,因不同任务通常需不同数据、输入特征或模型结构。本文提出MT2KD,一种两阶段多任务学习框架,旨在构建一个能联合完成自动语音识别(ASR)、音频标注(AT)和说话人验证(SV)的通用编码器。第一阶段采用多教师知识蒸馏,利用相同无标签数据将三个高性能单任务教师编码器的特征空间对齐至单一学生编码器;第二阶段通过从第一阶段初始化模型,并在各任务独立标注数据上进行多任务监督微调。实验表明,该多任务训练流程显著优于从零开始的多任务训练基线。最终系统在三项任务上均表现良好:ASR词错误率仅增加不足4%,AT平均精度下降1.9,SV等错误率仅高0.23%,且总参数量仅为66M。
原文摘要 · Abstract (English)
With the advances in deep learning, the performance of end-to-end (E2E) single-task models for speech and audio processing has been constantly improving. However, it is still challenging to build a general-purpose model with high performance on multiple tasks, since different speech and audio processing tasks usually require different training data, input features, or model architectures to achieve optimal performance. In this work, MT2KD, a novel two-stage multi-task learning framework is proposed to build a general-purpose speech and audio encoder that jointly performs three fundamental tasks: automatic speech recognition (ASR), audio tagging (AT) and speaker verification (SV). In the first stage, multi-teacher knowledge distillation (KD) is applied to align the feature spaces of three single-task high-performance teacher encoders into a single student encoder using the same unlabelled data. In the second stage, multi-task supervised fine-tuning is carried out by initialising the model from the first stage and training on the separate labelled data of each single task. Experiments demonstrate that the proposed multi-task training pipeline significantly outperforms a baseline model trained with multi-task learning from scratch. The final system achieves good performance on ASR, AT and SV: with less than 4% relative word-error-rate increase on ASR, only 1.9 lower mean averaged precision on AT and 0.23% absolute higher equal error rate on SV compared to the best-performing single-task encoders, using only a 66M total model parameters.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。