license: apache-2.0
language:
- vi
metrics:
- wer
- cer
base_model:
- openai/whisper-small
pipeline_tag: automatic-speech-recognition
tags:
- vietnamese
- lyrics
- alt
- song
- pytorch
- whisper
- transformers
Vietnamese Automatic Lyrics Transcription
This project aims to perform automatic lyrics transcription on Vietnamese songs, the pre-trained model used for this task is Whisper from Robust Speech Recognition via Large-Scale Weak Supervision.
Fine-Tuning
The model is fine-tuned on 8,000 Vietnamese songs scraped from zingmp3.vn (Vietnamese version of Spotify). The average song duration is 4.7 minutes, with a word per minute of 90.7.
7,000 Songs are used as training and 1,000 songs are used as validation. The reported metrics below are for the 1,000 validation songs.
Evaluation
| Model | WER (%) Case-Sensitive |
WER (%) Lowercase |
CER (%) Case-Sensitive |
CER (%) Lowercase |
|---|---|---|---|---|
| whisper-small | 34.91 | 30.73 | 24.82 | 23.65 |
| whisper-medium | 26.42 | 23.15 | 17.03 | 17.01 |
| whisper-large-v2 | 24.61 | 20.52 | 17.14 | 16.09 |
Lyrics Transcription
To generate the transcription for a song, we can use the Transformers pipeline. Chunking is enabled by setting chunk_length_s=30 when instantiating the pipeline. With chunking enabled, the pipeline can be run with batched inference. It can also be extended to predict sequence level timestamps by passing return_timestamps=True: In the following example we are passing return_timestamps="word" that provides precise timestamps for when each individual word in the audio starts and ends.
>>> from transformers import pipeline
>>> asr_pipeline = pipeline(
>>> "automatic-speech-recognition",
>>> model="xyzDivergence/whisper-small-vietnamese-lyrics-transcription", chunk_length_s=30, device='cuda',
>>> tokenizer="xyzDivergence/whisper-small-vietnamese-lyrics-transcription"
>>> )
>>> transcription = asr_pipeline("sample_audio.mp3", return_timestamps="word")
Training Data
The training dataset consists of 7,000 Vietnamese songs, in total of roughly 550 hours of audio, across various Vietnamese music genres, dialects and accents. Due to IP concerns, the data is not publicly available. Each song includes lyrics along with corresponding line-level timestamps, enabling precise mapping of audio segments to their respective lyrics based on the provided timestamp information.
Technical report coming soon. This project was made through equal contributions from: