arXiv:2502.12672cs.CLcs.AI2025-02被引 6

提出新方法让语音模型在微调后仍保持跨任务泛化能力

Speech-FT: Merging Pre-trained And Fine-Tuned Speech Representation Models For Cross-Task Generalization

  • 两阶段微调:先减小表征漂移,再用预训练模型插值恢复通用性
  • 在SUPERB基准上,语音识别错误率降低至3.94%,说话人识别准确率提升至84.11%
  • 适合需要兼顾特定任务性能与通用性的语音模型优化场景

微调语音表示模型虽能提升特定任务表现,但常损害其跨任务泛化能力,原因在于表征过度变化导致预训练信息丢失。现有方法如权重正则化难以维持足够的特征相似性。为此,我们提出Speech-FT,一种两阶段微调框架:首先通过定制化微调减少表征漂移,随后在权重空间进行预训练模型插值以恢复跨任务泛化能力。在HuBERT、wav2vec 2.0、DeCoAR 2.0和WavLM Base+上的大量实验表明,Speech-FT在监督、无监督及多任务微调场景中均持续提升性能。相比显式约束权重变化的基线(如权重空间正则化、LoRA),Speech-FT在允许更大权重更新的同时,保持更高特征相似性,展现出更优的跨任务泛化能力。例如,在HuBERT自动语音识别任务中,电话错误率从5.17%降至3.94%,词错误率从6.38%降至5.75%,说话人识别准确率从81.86%升至84.11%。

原文摘要 · Abstract (English)

Fine-tuning speech representation models can enhance performance on specific tasks but often compromises their cross-task generalization ability. This degradation is often caused by excessive changes in the representations, making it difficult to retain information learned during pre-training. Existing approaches, such as regularizing weight changes during fine-tuning, may fail to maintain sufficiently high feature similarity with the pre-trained model, and thus could possibly lose cross-task generalization. To address this issue, we propose Speech-FT, a novel two-stage fine-tuning framework designed to maintain cross-task generalization while benefiting from fine-tuning. Speech-FT first applies fine-tuning specifically designed to reduce representational drift, followed by weight-space interpolation with the pre-trained model to restore cross-task generalization. Extensive experiments on HuBERT, wav2vec 2.0, DeCoAR 2.0, and WavLM Base+ demonstrate that Speech-FT consistently improves performance across a wide range of supervised, unsupervised, and multitask fine-tuning scenarios. Moreover, Speech-FT achieves superior cross-task generalization compared to fine-tuning baselines that explicitly constrain weight changes, such as weight-space regularization and LoRA fine-tuning. Our analysis reveals that Speech-FT maintains higher feature similarity to the pre-trained model compared to alternative strategies, despite allowing larger weight-space updates. Notably, Speech-FT achieves significant improvements on the SUPERB benchmark. For example, when fine-tuning HuBERT on automatic speech recognition, Speech-FT is able to reduce phone error rate from 5.17% to 3.94%, lower word error rate from 6.38% to 5.75%, and increase speaker identification accuracy from 81.86% to 84.11%. Speech-FT provides a simple yet powerful solution for further refining speech representation models after pre-training.

语音表示微调泛化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。