Skip to content
 
 

Latest commit

 

History

40 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Voxlect: A Speech Foundation Model Benchmark for Modeling Dialects and Regional Languages Around the Globe

📄 [Preprint Paper]  |  🤗 [Whisper-small Models]   |   [Whisper-Large v3 Models]   |   [MMS-LID-256 Models]  

This repo presents Voxlect, a benchmark that predicts dialects and regional languages worldwide using speech foundation models.

This repository is a Hyperion-maintained fork of the original Voxlect project. The fork adapts dependency constraints and packaging so Voxlect can coexist with the Hyperion toolkit. It is not an official release, endorsement, or representation of the original authors.

The published distribution is hyperion-voxlect. The existing Python source namespace remains src, so existing imports such as src.model.dialect.whisper_dialect remain unchanged.

We report benchmark evaluations on dialects and regional language varieties in English, Arabic, Mandarin and Cantonese, Tibetan, Indic languages and Indian English, Thai, Spanish, French, German, Brazilian Portuguese, and Italian. Our study used over 2 million training utterances from 30 publicly available speech corpora that are provided with dialectal (or accent) information.

Labeling Scheme

In Voxlect, we experiments with the following datasets to predict dialects or regional languages. Our labeling is described below:

Our training data filters output audio shorter than 3 seconds (unreliable predictions) and longer than 15 seconds (computation limitation), so you need to cut your audio to a maximum of 15 seconds, 16kHz and mono channel.

We also subsample datasets like IndicVoices and CORRA, so you may see noticeable smaller training sample size than original datasets.

Benchmark Performance

Compared to Vox-Profile, we use additional English speech data in ParaSpeechCaps that gains improved classification performance.

Error Confusions - Whisper-Large v3 model predictions

We observe that geographic proximity is a "main" source of confusion, while this indicates the evolution of languages and dialects.

Mandarin Dialects and Cantonese

Spanish Dialects

Download Repo

git clone git@github.com:hyperion-ml/voxlect.git

Installation

python -m pip install hyperion-voxlect

Python 3.10 or newer is required. PyTorch and TorchAudio may need a platform-specific installation command for the target CPU, CUDA, or ROCm environment; install that build first when necessary.

The package includes model architecture and adapter code only. Model weights, datasets, audio, caches, and other large external assets are not included. Model checkpoints are downloaded separately from Hugging Face; review the license and terms for each checkpoint and its underlying pretrained model.

The relaxed dependency ranges are intentional: this fork avoids fixed CUDA, PyTorch, NumPy, and transitive dependency pins so it can be installed in the same environment as Hyperion. SpeechBrain is not required by this fork.

Quick Example 1 - Whisper Large Mandarin Dialect and Cantonese Classification

# Load libraries
import torch
import torch.nn.functional as F
from src.model.dialect.whisper_dialect import WhisperWrapper

# Label List
dialect_label_list = [
    "Jiang-Huai",
    "Jiao-Liao",
    "Ji-Lu",
    "Lan-Yin",
    "Mandarin",
    "Southwestern",
    "Zhongyuan",
    "Cantonese"
]
    
# Find device
device = torch.device("cuda") if torch.cuda.is_available() else "cpu"

# Load model from Huggingface
whisper_model = WhisperWrapper.from_pretrained("tiantiaf/voxlect-mandarin-cantonese-dialect-whisper-large-v3").to(device)
whisper_model.eval()

# Load data, here just zeros as the example
# Our training data filters output audio shorter than 3 seconds (unreliable predictions) and longer than 15 seconds (computation limitation)
# So you need to prepare your audio to a maximum of 15 seconds, 16kHz and mono channel
max_audio_length = 15 * 16000
data = torch.zeros([1, 16000]).float().to(device)[:, :max_audio_length]
whisper_logits, whisper_embeddings = whisper_model(data, return_feature=True)
    
# Probability and output
whisper_prob = F.softmax(whisper_logits, dim=1)
print(dialect_label_list[torch.argmax(whisper_prob).detach().cpu().item()])

For example, the Sichuan speech dialects generated from the CosyVoice2:

example_audios/Sichuan_1000238_4_0.mp3

Using the Voxlect, we obtain the following probability

Dialect: Jiang-Huai       Probability: 0.001
Dialect: Jiao-Liao        Probability: 0.001
Dialect: Ji-Lu            Probability: 0.009
Dialect: Lan-Yin          Probability: 0.000
Dialect: Mandarin         Probability: 0.002
Dialect: Southwestern     Probability: 0.981 (Target dialect)
Dialect: Zhongyuan        Probability: 0.006
Dialect: Yue              Probability: 0.000

For example, the Tianjin speech dialects generated from the CosyVoice2:

example_audios/Tianjin_1002906_0_0.wav

Using the Voxlect, we obtain the following probability

Dialect: Jiang-Huai       Probability: 0.002
Dialect: Jiao-Liao        Probability: 0.051
Dialect: Ji-Lu            Probability: 0.169 (Target dialect)
Dialect: Lan-Yin          Probability: 0.001
Dialect: Mandarin         Probability: 0.765
Dialect: Southwestern     Probability: 0.000 
Dialect: Zhongyuan        Probability: 0.013
Dialect: Yue              Probability: 0.000

If you are a Mandarin speaker, you will notice that the first 2 words are spoken likely to be Tianjin dialect, but after that, the generated speech sounds more like standard Mandarin, and our model captures this correctly.

Quick Example 2 - MMS-LID-256 Spanish Dialect

# Load libraries
import torch
import torch.nn.functional as F
from src.model.dialect.mms_dialect import MMSWrapper

# Label List
spanish_dialect_list = [
    "Andino-Pacífico", 
    "Caribe and Central", 
    "Chileno",
    "Mexican", 
    "Penisular", 
    "Rioplatense", 
]
    
# Find device
device = torch.device("cuda") if torch.cuda.is_available() else "cpu"

# Load model from Huggingface
mms_model = MMSWrapper.from_pretrained("tiantiaf/voxlect-spanish-dialect-mms-lid-256").to(device)
mms_model.eval()

# Load data, here just zeros as the example
# Our training data filters output audio shorter than 3 seconds (unreliable predictions) and longer than 15 seconds (computation limitation)
# So you need to prepare your audio to a maximum of 15 seconds, 16kHz and mono channel
max_audio_length = 15 * 16000
data = torch.zeros([1, 16000]).float().to(device)[:, :max_audio_length]
mms_logits, mms_embeddings = mms_model(data, return_feature=True)
    
# Probability and output
mms_prob = F.softmax(mms_logits, dim=1)
print(spanish_dialect_list[torch.argmax(mms_prob).detach().cpu().item()])

Given that the Voxlect Benchmark paper is still under peer-review, we provide limited set of models and model weights before the review is concluded. But below are the models we currently put out.

MMS-LID-256 Models

Model Name Data Pre-trained Model Use LoRa LoRa Rank Size Output Example Code
tiantiaf/voxlect-english-dialect-mms-lid-256 12 Datasets mms-lid-256 Yes 64 16 English Varieties
tiantiaf/voxlect-spanish-dialect-mms-lid-256 CommonVoice+Latin American Spanish mms-lid-256 Yes 64 Penisular, Mexican, Chileno, Andino-Pacífico, Central America and Caribbean, Rioplatense
tiantiaf/voxlect-mandarin-cantonese-dialect-mms-lid-256 KeSpeech+CommonVoice-yue+CommonVoice-hk mms-lid-256 Yes 64 Jiang-Huai,Jiao-Liao,Ji-Lu,Lan-Yin,Standard Mandarin,Southwestern,Zhongyuan,Cantonese
tiantiaf/voxlect-indic-lid-mms-lid-256 IndicVoices+CommonVoice-en mms-lid-256 Yes 64 22 Indic Languages (e.g. Hindi, Urdu, Telegu, Tamil) and Indian English
tiantiaf/voxlect-thai-dialect-mms-lid-256 Thai-Dialect-Corpus mms-lid-256 Yes 64 Thai Central, Khummuang, Korat, Pattani
tiantiaf/voxlect-french-dialect-mms-lid-256 CommonVoice-fr+African Accented French mms-lid-256 Yes 64 France, Africa, Canada, Swiss/Belgium/German
tiantiaf/voxlect-german-dialect-mms-lid-256 CommonVoice-de mms-lid-256 Yes 64 German-Non-NRW Area, German-NRW, Austria, Swiss, Other

Whisper-Large V3 Models

Model Name Data Pre-trained Model Use LoRa LoRa Rank Size Output Example Code
tiantiaf/voxlect-english-dialect-whisper-large-v3 12 Datasets whisper-large-v3 Yes 64 16 English Varieties
tiantiaf/voxlect-spanish-dialect-whisper-large-v3 CommonVoice+Latin American Spanish whisper-large-v3 Yes 64 Penisular, Mexican, Chileno, Andino-Pacífico, Central America and Caribbean, Rioplatense
tiantiaf/voxlect-mandarin-cantonese-dialect-whisper-large-v3 KeSpeech+CommonVoice-yue+CommonVoice-hk whisper-large-v3 Yes 64 Jiang-Huai,Jiao-Liao,Ji-Lu,Lan-Yin,Standard Mandarin,Southwestern,Zhongyuan,Cantonese
tiantiaf/voxlect-indic-lid-whisper-large-v3 IndicVoices+CommonVoice-en whisper-large-v3 Yes 64 22 Indic Languages (e.g. Hindi, Urdu, Telegu, Tamil) and Indian English
tiantiaf/voxlect-thai-dialect-whisper-large-v3 Thai-Dialect-Corpus whisper-large-v3 Yes 64 Thai Central, Khummuang, Korat, Pattani
tiantiaf/voxlect-french-dialect-whisper-large-v3 CommonVoice-fr+African Accented French whisper-large-v3 Yes 64 France, Africa, Canada, Swiss/Belgium/German
tiantiaf/voxlect-german-dialect-whisper-large-v3 CommonVoice-de whisper-large-v3 Yes 64 German-Non-NRW Area, German-NRW, Austria, Swiss, Other

Whisper-small Models

Model Name Data Pre-trained Model Use LoRa LoRa Rank Size Output Example Code
tiantiaf/voxlect-english-dialect-whisper-small 12 Datasets whisper-small Yes 64 16 English Varieties
tiantiaf/voxlect-spanish-dialect-whisper-small CommonVoice+Latin American Spanish whisper-small Yes 64 Penisular, Mexican, Chileno, Andino-Pacífico, Central America and Caribbean, Rioplatense
tiantiaf/voxlect-mandarin-cantonese-dialect-whisper-small KeSpeech+CommonVoice-yue+CommonVoice-hk whisper-small Yes 64 Jiang-Huai,Jiao-Liao,Ji-Lu,Lan-Yin,Standard Mandarin,Southwestern,Zhongyuan,Cantonese
tiantiaf/voxlect-indic-lid-whisper-small IndicVoices+CommonVoice-en whisper-small Yes 64 22 Indic Languages (e.g. Hindi, Urdu, Telegu, Tamil) and Indian English
tiantiaf/voxlect-thai-dialect-whisper-small Thai-Dialect-Corpus whisper-small Yes 64 Thai Central, Khummuang, Korat, Pattani
tiantiaf/voxlect-french-dialect-whisper-small CommonVoice-fr+African Accented French whisper-small Yes 64 France, Africa, Canada, Swiss/Belgium/German
tiantiaf/voxlect-german-dialect-whisper-small CommonVoice-de whisper-small Yes 64 German-Non-NRW Area, German-NRW, Austria, Swiss, Other

Responsible Use: Users should respect the privacy and consent of the data subjects, and adhere to the relevant laws and regulations in their jurisdictions when using Voxlect.

Out-of-Scope Use

  • Clinical or diagnostic applications
  • Surveillance
  • Privacy-invasive applications
  • No commercial use

The source code remains distributed under the complete Responsible AI Source Code License v1.1 (RAIL), included in LICENSE. The license contains specific use restrictions, notice requirements, and termination provisions. Read it before using or redistributing this software; this README does not replace the license. Model checkpoints, pretrained models, datasets, and external assets may have separate licenses and terms.

If you like our work or use the models in your work, kindly cite the following. We appreciate your recognition!

@article{feng2025voxlect,
  title={Voxlect: A Speech Foundation Model Benchmark for Modeling Dialects and Regional Languages Around the Globe},
  author={Feng, Tiantian and Huang, Kevin and Xu, Anfeng and Shi, Xuan and Lertpetchpun, Thanathai and Lee, Jihwan and Lee, Yoonjeong and Byrd, Dani and Narayanan, Shrikanth},
  journal={arXiv preprint arXiv:2508.01691},
  year={2025}
}

Maintainer publishing setup

The GitHub Actions workflow publishes only when a GitHub Release is published from main. It uses PyPI Trusted Publishing (OIDC), not a long-lived API token.

For the first release, configure a PyPI GitHub Actions Trusted Publisher with:

  • PyPI project: hyperion-voxlect
  • Owner: hyperion-ml
  • Repository: voxlect
  • Workflow filename: .github/workflows/python-publish.yml
  • GitHub environment: pypi

Create the pypi environment in GitHub and restrict deployment to authorized maintainers; required reviewers are recommended. Do not add a PyPI token secret. Then create a version tag and a published GitHub Release targeting main. Ordinary pushes, pull requests, and draft releases do not publish.

After the workflow succeeds, verify the wheel and source archive, metadata, dependencies, Python requirement, and included LICENSE on PyPI.

Related work

@article{feng2025vox,
  title={Vox-Profile: A Speech Foundation Model Benchmark for Characterizing Diverse Speaker and Speech Traits},
  author={Feng, Tiantian and Lee, Jihwan and Xu, Anfeng and Lee, Yoonjeong and Lertpetchpun, Thanathai and Shi, Xuan and Wang, Helin and Thebaud, Thomas and Moro-Velazquez, Laureano and Byrd, Dani and others},
  journal={arXiv preprint arXiv:2505.14648},
  year={2025}
}

About

[KDD 2026] Voxlect: A Speech Foundation Model Benchmark for Modeling Dialects and Regional Languages Around the Globe

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages