Continual Learning with Embedding Layer Surgery and Task-wise Beam Search using Whisper

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Kwok, Chin Yuen, Yip, Jia Qi, Chng, Eng Siong
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929675206393856
author Kwok, Chin Yuen
Yip, Jia Qi
Chng, Eng Siong
author_facet Kwok, Chin Yuen
Yip, Jia Qi
Chng, Eng Siong
contents Current Multilingual ASR models only support a fraction of the world's languages. Continual Learning (CL) aims to tackle this problem by adding new languages to pre-trained models while avoiding the loss of performance on existing languages, also known as Catastrophic Forgetting (CF). However, existing CL methods overlook the adaptation of the token embedding lookup table at the decoder, despite its significant contribution to CF. We propose Embedding Layer Surgery where separate copies of the token embeddings are created for each new languages, and one of the copies is selected to replace the old languages embeddings when transcribing the corresponding new language. Unfortunately, this approach means LID errors also cause incorrect ASR embedding selection. Our Task-wise Beam Search allows self-correction for such mistakes. By adapting Whisper to 10 hours of data for each of 10 unseen languages from Common Voice, results show that our method reduces the Average WER (AWER) of pre-trained languages from 14.2% to 11.9% compared with Experience Replay, without compromising the AWER of the unseen languages.
format Preprint
id arxiv_https___arxiv_org_abs_2501_07875
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Continual Learning with Embedding Layer Surgery and Task-wise Beam Search using Whisper
Kwok, Chin Yuen
Yip, Jia Qi
Chng, Eng Siong
Computation and Language
Artificial Intelligence
Current Multilingual ASR models only support a fraction of the world's languages. Continual Learning (CL) aims to tackle this problem by adding new languages to pre-trained models while avoiding the loss of performance on existing languages, also known as Catastrophic Forgetting (CF). However, existing CL methods overlook the adaptation of the token embedding lookup table at the decoder, despite its significant contribution to CF. We propose Embedding Layer Surgery where separate copies of the token embeddings are created for each new languages, and one of the copies is selected to replace the old languages embeddings when transcribing the corresponding new language. Unfortunately, this approach means LID errors also cause incorrect ASR embedding selection. Our Task-wise Beam Search allows self-correction for such mistakes. By adapting Whisper to 10 hours of data for each of 10 unseen languages from Common Voice, results show that our method reduces the Average WER (AWER) of pre-trained languages from 14.2% to 11.9% compared with Experience Replay, without compromising the AWER of the unseen languages.
title Continual Learning with Embedding Layer Surgery and Task-wise Beam Search using Whisper
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2501.07875