Weight Averaging: A Simple Yet Effective Method to Overcome Catastrophic Forgetting in Automatic Speech Recognition
About
Adapting a trained Automatic Speech Recognition (ASR) model to new tasks results in catastrophic forgetting of old tasks, limiting the model's ability to learn continually and to be extended to new speakers, dialects, languages, etc. Focusing on End-to-End ASR, in this paper, we propose a simple yet effective method to overcome catastrophic forgetting: weight averaging. By simply taking the average of the previous and the adapted model, our method achieves high performance on both the old and new tasks. It can be further improved by introducing a knowledge distillation loss during the adaptation. We illustrate the effectiveness of our method on both monolingual and multilingual ASR. In both cases, our method strongly outperforms all baselines, even in its simplest form.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Automatic Speech Recognition | Libri-Adapt Exp. 2 | WER (LIB Group 1)7.5 | 22 | |
| Automatic Speech Recognition | Common Voice English Accents Exp. 1 | WER (US Accent)15.9 | 15 | |
| Multilingual Automatic Speech Recognition | Common Voice Exp. 3 (test) | WER (US)19.9 | 13 | |
| Automatic Speech Recognition Adaptation | CGN Exp. 4 (test) | WER (CV/E)14.8 | 11 | |
| Automatic Speech Recognition | Common Voice (ENG, DEU, ESP) and CGN (NL, VL) OWSM v3.2 (test) | WER (ENG)16.4 | 11 | |
| Automatic Speech Recognition | Common Voice (ENG, DEU, ESP), CGN (VL), and GCND (DVL) OWSM v3.2 (test) | Word Error Rate (ENG)17.3 | 9 | |
| Continual Learning for Automatic Speech Recognition | Common Voice Exp. 1 (test) | WER (US)15.9 | 7 |