用小样本验证集动态调整训练轮次,防止语音识别模型个性化时遗忘通用知识。
ValSub: Subsampling Validation Data to Mitigate Forgetting during ASR Personalization
- 从大验证集抽样生成小而有效的验证集,用于监测模型遗忘程度。
- 在3种遗忘阈值下,误差比随机采样低20%以上,最差仅3.78。
- 适合移动端语音识别模型个性化,无需额外计算资源。
自动语音识别(ASR)广泛应用于手机等消费设备。近期研究表明,针对目标用户语音进行模型微调可提升对生僻词或口音的识别性能。然而,这种微调可能导致模型遗忘原始训练数据分布的知识(灾难性遗忘),从而降低通用识别能力。现有方法通过代表源域分布的验证集来衡量遗忘,但此类验证集过大,难以在移动设备上使用。为此,我们提出一种新方法,将大规模验证集高效子采样为更小的集合,同时保持遗忘估计能力。实验表明,该方法能动态确定最优微调轮次:与50倍大的基准验证集相比,平均绝对误差为3.39,优于相同规模的随机采样(3.78-8.65)。在三种遗忘阈值下,我们的方法始终稳定跟踪基准行为。
原文摘要 · Abstract (English)
Automatic Speech Recognition (ASR) is widely used within consumer devices such as mobile phones. Recently, personalization or on-device model fine-tuning has shown that adaptation of ASR models towards target user speech improves their performance over rare words or accented speech. Despite these gains, fine-tuning on user data (target domain) risks the personalized model to forget knowledge about its original training distribution (source domain) i.e. catastrophic forgetting, leading to subpar general ASR performance. A simple and efficient approach to combat catastrophic forgetting is to measure forgetting via a validation set that represents the source domain distribution. However, such validation sets are large and impractical for mobile devices. Towards this, we propose a novel method to subsample a substantially large validation set into a smaller one while maintaining the ability to estimate forgetting. We demonstrate the efficacy of such a dataset in mitigating forgetting by utilizing it to dynamically determine the number of ideal fine-tuning epochs. When measuring the deviations in per user fine-tuning epochs against a 50x larger validation set (oracle), our method achieves a lower mean-absolute-error (3.39) compared to randomly selected subsets of the same size (3.78-8.65). Unlike random baselines, our method consistently tracks the oracle's behaviour across three different forgetting thresholds.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。