WavTokenizer: an Efficient Acoustic Discrete Codec Tokenizer for Audio Language Modeling
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , , , , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2024
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866910843631828992 |
|---|---|
| author | Ji, Shengpeng Jiang, Ziyue Wang, Wen Chen, Yifu Fang, Minghui Zuo, Jialong Yang, Qian Cheng, Xize Wang, Zehan Li, Ruiqi Zhang, Ziang Yang, Xiaoda Huang, Rongjie Jiang, Yidi Chen, Qian Zheng, Siqi Zhao, Zhou |
| author_facet | Ji, Shengpeng Jiang, Ziyue Wang, Wen Chen, Yifu Fang, Minghui Zuo, Jialong Yang, Qian Cheng, Xize Wang, Zehan Li, Ruiqi Zhang, Ziang Yang, Xiaoda Huang, Rongjie Jiang, Yidi Chen, Qian Zheng, Siqi Zhao, Zhou |
| contents | Language models have been effectively applied to modeling natural signals, such as images, video, speech, and audio. A crucial component of these models is the codec tokenizer, which compresses high-dimensional natural signals into lower-dimensional discrete tokens. In this paper, we introduce WavTokenizer, which offers several advantages over previous SOTA acoustic codec models in the audio domain: 1)extreme compression. By compressing the layers of quantizers and the temporal dimension of the discrete codec, one-second audio of 24kHz sampling rate requires only a single quantizer with 40 or 75 tokens. 2)improved subjective quality. Despite the reduced number of tokens, WavTokenizer achieves state-of-the-art reconstruction quality with outstanding UTMOS scores and inherently contains richer semantic information. Specifically, we achieve these results by designing a broader VQ space, extended contextual windows, and improved attention networks, as well as introducing a powerful multi-scale discriminator and an inverse Fourier transform structure. We conducted extensive reconstruction experiments in the domains of speech, audio, and music. WavTokenizer exhibited strong performance across various objective and subjective metrics compared to state-of-the-art models. We also tested semantic information, VQ utilization, and adaptability to generative models. Comprehensive ablation studies confirm the necessity of each module in WavTokenizer. The related code, demos, and pre-trained models are available at https://github.com/jishengpeng/WavTokenizer. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2408_16532 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | WavTokenizer: an Efficient Acoustic Discrete Codec Tokenizer for Audio Language Modeling Ji, Shengpeng Jiang, Ziyue Wang, Wen Chen, Yifu Fang, Minghui Zuo, Jialong Yang, Qian Cheng, Xize Wang, Zehan Li, Ruiqi Zhang, Ziang Yang, Xiaoda Huang, Rongjie Jiang, Yidi Chen, Qian Zheng, Siqi Zhao, Zhou Audio and Speech Processing Machine Learning Multimedia Sound Signal Processing Language models have been effectively applied to modeling natural signals, such as images, video, speech, and audio. A crucial component of these models is the codec tokenizer, which compresses high-dimensional natural signals into lower-dimensional discrete tokens. In this paper, we introduce WavTokenizer, which offers several advantages over previous SOTA acoustic codec models in the audio domain: 1)extreme compression. By compressing the layers of quantizers and the temporal dimension of the discrete codec, one-second audio of 24kHz sampling rate requires only a single quantizer with 40 or 75 tokens. 2)improved subjective quality. Despite the reduced number of tokens, WavTokenizer achieves state-of-the-art reconstruction quality with outstanding UTMOS scores and inherently contains richer semantic information. Specifically, we achieve these results by designing a broader VQ space, extended contextual windows, and improved attention networks, as well as introducing a powerful multi-scale discriminator and an inverse Fourier transform structure. We conducted extensive reconstruction experiments in the domains of speech, audio, and music. WavTokenizer exhibited strong performance across various objective and subjective metrics compared to state-of-the-art models. We also tested semantic information, VQ utilization, and adaptability to generative models. Comprehensive ablation studies confirm the necessity of each module in WavTokenizer. The related code, demos, and pre-trained models are available at https://github.com/jishengpeng/WavTokenizer. |
| title | WavTokenizer: an Efficient Acoustic Discrete Codec Tokenizer for Audio Language Modeling |
| topic | Audio and Speech Processing Machine Learning Multimedia Sound Signal Processing |
| url | https://arxiv.org/abs/2408.16532 |