Variable Bitrate Residual Vector Quantization for Audio Coding

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chae, Yunkee, Choi, Woosung, Takida, Yuhta, Koo, Junghyun, Ikemiya, Yukara, Zhong, Zhi, Cheuk, Kin Wai, Martínez-Ramírez, Marco A., Lee, Kyogu, Liao, Wei-Hsiang, Mitsufuji, Yuki
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915260697411584
author Chae, Yunkee
Choi, Woosung
Takida, Yuhta
Koo, Junghyun
Ikemiya, Yukara
Zhong, Zhi
Cheuk, Kin Wai
Martínez-Ramírez, Marco A.
Lee, Kyogu
Liao, Wei-Hsiang
Mitsufuji, Yuki
author_facet Chae, Yunkee
Choi, Woosung
Takida, Yuhta
Koo, Junghyun
Ikemiya, Yukara
Zhong, Zhi
Cheuk, Kin Wai
Martínez-Ramírez, Marco A.
Lee, Kyogu
Liao, Wei-Hsiang
Mitsufuji, Yuki
contents Recent state-of-the-art neural audio compression models have progressively adopted residual vector quantization (RVQ). Despite this success, these models employ a fixed number of codebooks per frame, which can be suboptimal in terms of rate-distortion tradeoff, particularly in scenarios with simple input audio, such as silence. To address this limitation, we propose variable bitrate RVQ (VRVQ) for audio codecs, which allows for more efficient coding by adapting the number of codebooks used per frame. Furthermore, we propose a gradient estimation method for the non-differentiable masking operation that transforms from the importance map to the binary importance mask, improving model training via a straight-through estimator. We demonstrate that the proposed training framework achieves superior results compared to the baseline method and shows further improvement when applied to the current state-of-the-art codec.
format Preprint
id arxiv_https___arxiv_org_abs_2410_06016
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Variable Bitrate Residual Vector Quantization for Audio Coding
Chae, Yunkee
Choi, Woosung
Takida, Yuhta
Koo, Junghyun
Ikemiya, Yukara
Zhong, Zhi
Cheuk, Kin Wai
Martínez-Ramírez, Marco A.
Lee, Kyogu
Liao, Wei-Hsiang
Mitsufuji, Yuki
Sound
Machine Learning
Audio and Speech Processing
Recent state-of-the-art neural audio compression models have progressively adopted residual vector quantization (RVQ). Despite this success, these models employ a fixed number of codebooks per frame, which can be suboptimal in terms of rate-distortion tradeoff, particularly in scenarios with simple input audio, such as silence. To address this limitation, we propose variable bitrate RVQ (VRVQ) for audio codecs, which allows for more efficient coding by adapting the number of codebooks used per frame. Furthermore, we propose a gradient estimation method for the non-differentiable masking operation that transforms from the importance map to the binary importance mask, improving model training via a straight-through estimator. We demonstrate that the proposed training framework achieves superior results compared to the baseline method and shows further improvement when applied to the current state-of-the-art codec.
title Variable Bitrate Residual Vector Quantization for Audio Coding
topic Sound
Machine Learning
Audio and Speech Processing
url https://arxiv.org/abs/2410.06016