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="kelvinbksoh/whisper-small-vietnamese-lyrics-transcription", chunk_length_s=30, device='cuda',
>>> tokenizer="kelvinbksoh/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 copyright concerns, the raw data is not publicly available. However, the CSV files, which contain links to the songs and lyrics, can be used for downloading and are available in our repository. 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.
For more details, please refer to our paper: Vietnamese Automatic Lyrics Transcription. This project was made through equal contributions from: