MaskSR: Masked Language Model for Full-band Speech Restoration

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Li, Xu, Wang, Qirui, Liu, Xiaoyu
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909216570081280
author Li, Xu
Wang, Qirui
Liu, Xiaoyu
author_facet Li, Xu
Wang, Qirui
Liu, Xiaoyu
contents Speech restoration aims at restoring high quality speech in the presence of a diverse set of distortions. Although several deep learning paradigms have been studied for this task, the power of the recently emerging language models has not been fully explored. In this paper, we propose MaskSR, a masked language model capable of restoring full-band 44.1 kHz speech jointly considering noise, reverb, clipping, and low bandwidth. MaskSR works with discrete acoustic tokens extracted using a pre-trained neural codec. During training, MaskSR is optimized to predict randomly masked tokens extracted from the high quality target speech, conditioned on the corrupted speech with various distortions. During inference, MaskSR reconstructs the target speech tokens with efficient iterative sampling. Extensive experiments show that MaskSR obtains competitive results on both the full-band speech restoration task and also on sub-tasks compared with a wide range of models.
format Preprint
id arxiv_https___arxiv_org_abs_2406_02092
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MaskSR: Masked Language Model for Full-band Speech Restoration
Li, Xu
Wang, Qirui
Liu, Xiaoyu
Sound
Artificial Intelligence
Machine Learning
Audio and Speech Processing
Signal Processing
Speech restoration aims at restoring high quality speech in the presence of a diverse set of distortions. Although several deep learning paradigms have been studied for this task, the power of the recently emerging language models has not been fully explored. In this paper, we propose MaskSR, a masked language model capable of restoring full-band 44.1 kHz speech jointly considering noise, reverb, clipping, and low bandwidth. MaskSR works with discrete acoustic tokens extracted using a pre-trained neural codec. During training, MaskSR is optimized to predict randomly masked tokens extracted from the high quality target speech, conditioned on the corrupted speech with various distortions. During inference, MaskSR reconstructs the target speech tokens with efficient iterative sampling. Extensive experiments show that MaskSR obtains competitive results on both the full-band speech restoration task and also on sub-tasks compared with a wide range of models.
title MaskSR: Masked Language Model for Full-band Speech Restoration
topic Sound
Artificial Intelligence
Machine Learning
Audio and Speech Processing
Signal Processing
url https://arxiv.org/abs/2406.02092