Boosting Code-Switching ASR with Mixture of Experts Enhanced Speech-Conditioned LLM

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Fengrun, Geng, Wang, Huang, Hukai, Shan, Yahui, Yi, Cheng, Qu, He
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914999223451648
author Zhang, Fengrun
Geng, Wang
Huang, Hukai
Shan, Yahui
Yi, Cheng
Qu, He
author_facet Zhang, Fengrun
Geng, Wang
Huang, Hukai
Shan, Yahui
Yi, Cheng
Qu, He
contents In this paper, we introduce a speech-conditioned Large Language Model (LLM) integrated with a Mixture of Experts (MoE) based connector to address the challenge of Code-Switching (CS) in Automatic Speech Recognition (ASR). Specifically, we propose an Insertion and Deletion of Interruption Token (IDIT) mechanism for better transfer text generation ability of LLM to speech recognition task. We also present a connecter with MoE architecture that manages multiple languages efficiently. To further enhance the collaboration of multiple experts and leverage the understanding capabilities of LLM, we propose a two-stage progressive training strategy: 1) The connector is unfrozen and trained with language-specialized experts to map speech representations to the text space. 2) The connector and LLM LoRA adaptor are trained with the proposed IDIT mechanism and all experts are activated to learn general representations. Experimental results demonstrate that our method significantly outperforms state-of-the-art models, including end-to-end and large-scale audio-language models.
format Preprint
id arxiv_https___arxiv_org_abs_2409_15905
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Boosting Code-Switching ASR with Mixture of Experts Enhanced Speech-Conditioned LLM
Zhang, Fengrun
Geng, Wang
Huang, Hukai
Shan, Yahui
Yi, Cheng
Qu, He
Sound
Artificial Intelligence
Audio and Speech Processing
In this paper, we introduce a speech-conditioned Large Language Model (LLM) integrated with a Mixture of Experts (MoE) based connector to address the challenge of Code-Switching (CS) in Automatic Speech Recognition (ASR). Specifically, we propose an Insertion and Deletion of Interruption Token (IDIT) mechanism for better transfer text generation ability of LLM to speech recognition task. We also present a connecter with MoE architecture that manages multiple languages efficiently. To further enhance the collaboration of multiple experts and leverage the understanding capabilities of LLM, we propose a two-stage progressive training strategy: 1) The connector is unfrozen and trained with language-specialized experts to map speech representations to the text space. 2) The connector and LLM LoRA adaptor are trained with the proposed IDIT mechanism and all experts are activated to learn general representations. Experimental results demonstrate that our method significantly outperforms state-of-the-art models, including end-to-end and large-scale audio-language models.
title Boosting Code-Switching ASR with Mixture of Experts Enhanced Speech-Conditioned LLM
topic Sound
Artificial Intelligence
Audio and Speech Processing
url https://arxiv.org/abs/2409.15905