First Push!

hyperparams.yaml

@@ -0,0 +1,67 @@
+# ############################################################################
+# Model: WAV2VEC XLSR model for Accent Recognition (Spanish)
+# see paper: https://arxiv.org/abs/2305.18283
+# ############################################################################
+
+# Hparams NEEDED
+HPARAMS_NEEDED: ["encoder_dim", "out_n_neurons", "accent_encoder", "softmax"]
+# Modules Needed
+MODULES_NEEDED: ["wav2vec2", "avg_pool", "output_mlp"]
+
+# Feature parameters
+# wav2vec2_hub: facebook/wav2vec2-base
+wav2vec2_hub: "facebook/wav2vec2-large-xlsr-53"
+
+# Pretrain folder (HuggingFace)
+pretrained_path: Jzuluaga/accent-id-commonaccent_xlsr-spanish
+# URL for the biggest Fairseq english wav2vec2 model.
+
+# parameters
+encoder_dim: 1024
+out_n_neurons: 6
+
+wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
+    source: !ref <wav2vec2_hub>
+    output_norm: True
+    freeze: True
+    save_path: wav2vec2_checkpoints
+
+# Mean and std normalization of the input features
+mean_var_norm_input: !new:speechbrain.processing.features.InputNormalization
+    norm_type: sentence
+    std_norm: False
+
+avg_pool: !new:speechbrain.nnet.pooling.StatisticsPooling
+    return_std: False
+
+output_mlp: !new:speechbrain.nnet.linear.Linear
+    input_size: !ref <encoder_dim>
+    n_neurons: !ref <out_n_neurons>
+    bias: False
+
+model: !new:torch.nn.ModuleList
+    - [!ref <output_mlp>]
+
+modules:
+    mean_var_norm_input: !ref <mean_var_norm_input>
+    wav2vec2: !ref <wav2vec2>
+    output_mlp: !ref <output_mlp>
+    avg_pool: !ref <avg_pool>
+
+softmax: !new:speechbrain.nnet.activations.Softmax
+        
+accent_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
+
+
+pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
+    loadables:
+        mean_var_norm_input: !ref <mean_var_norm_input>
+        wav2vec2: !ref <wav2vec2>
+        model: !ref <model>
+        accent_encoder: !ref <accent_encoder>
+    paths:
+        mean_var_norm_input: !ref <pretrained_path>/normalizer_input.ckpt
+        wav2vec2: !ref <pretrained_path>/wav2vec2.ckpt
+        model: !ref <pretrained_path>/model.ckpt
+        accent_encoder: !ref <pretrained_path>/accent_encoder.txt
+

README.md

@@ -1,3 +1,174 @@
 ---
-license: mit
+language:
+- en
+thumbnail:
+tags:
+- audio-classification
+- speechbrain
+- embeddings
+- Accent Identification
+- pytorch
+- wav2vec2
+- XLSR
+- CommonAccent
+license: "mit"
+datasets:
+- CommonVoice
+metrics:
+- Accuracy
+widget:
+- example_title: Caribe-Colombia-Cuba
+  src: https://huggingface.co/Jzuluaga/accent-id-commonaccent_xlsr-spanish/resolve/main/data/caribe-cuba-colombia.wav
+- example_title: Andino
+  src: https://huggingface.co/Jzuluaga/accent-id-commonaccent_xlsr-spanish/resolve/main/data/andino.wav
+- example_title: Mexico
+  src: https://huggingface.co/Jzuluaga/accent-id-commonaccent_xlsr-spanish/resolve/main/data/mexico.wav
+- example_title: Spain
+  src: https://huggingface.co/Jzuluaga/accent-id-commonaccent_xlsr-spanish/resolve/main/data/spain.wav
 ---
+
+
+<iframe src="https://ghbtns.com/github-btn.html?user=speechbrain&repo=speechbrain&type=star&count=true&size=large&v=2" frameborder="0" scrolling="0" width="170" height="30" title="GitHub"></iframe>
+<br/><br/>
+
+# CommonAccent: Exploring Large Acoustic Pretrained Models for Accent Classification Based on Common Voice
+
+
+**Abstract**: 
+Despite the recent advancements in Automatic Speech Recognition (ASR), the recognition of accented speech still remains a dominant problem. In order to create more inclusive ASR systems, research has shown that the integration of accent information, as part of a larger ASR framework, can lead to the mitigation of accented speech errors. We address multilingual accent classification through the ECAPA-TDNN and Wav2Vec 2.0/XLSR architectures which have been proven to perform well on a variety of speech-related downstream tasks. We introduce a simple-to-follow recipe aligned to the SpeechBrain toolkit for accent classification based on Common Voice 7.0 (English) and Common Voice 11.0 (Italian, German, and Spanish). Furthermore, we establish new state-of-the-art for English accent classification with as high as 95% accuracy. We also study the internal categorization of the Wav2Vev 2.0 embeddings through t-SNE, noting that there is a level of clustering based on phonological similarity. 
+
+
+This repository provides all the necessary tools to perform accent identification from speech recordings with [SpeechBrain](https://github.com/speechbrain/speechbrain).
+The system uses a model pretrained on the CommonAccent dataset in Spanish (6 accents). This system is based on the CommonLanguage Recipe located here: https://github.com/speechbrain/speechbrain/tree/develop/recipes/CommonLanguage
+
+
+The provided system can recognize the following 16 accents from short speech recordings in Spanish (ES):
+
+```
+- ESPANA SUR PENINSULAR - ANDALUCIA EXTREMADURA MURCIA
+- MEXICO
+- ANDINOPACIFICO COLOMBIA PERU ECUADOR OESTE DE BOLIVIA Y VENEZUELA ANDINA
+- CARIBE CUBA VENEZUELA PUERTO RICO REPUBLICA DOMINICANA PANAMA COLOMBIA CARIBENA MEXICO CARIBENO COSTA DEL GOLFO DE MEXICO
+- RIOPLATENSE ARGENTINA URUGUAY ESTE DE BOLIVIA PARAGUAY
+- CHILENO CHILE CUYO
+```
+
+<a href="https://github.com/JuanPZuluaga/accent-recog-slt2022"> <img alt="GitHub" src="https://img.shields.io/badge/GitHub-Open%20source-green"> </a> Github repository link: https://github.com/JuanPZuluaga/accent-recog-slt2022
+
+
+For a better experience, we encourage you to learn more about
+[SpeechBrain](https://speechbrain.github.io). The given model performance on the test set is:
+
+| Release (dd/mm/yyyy) | Accuracy (%)
+|:-------------:|:--------------:|
+| 01-08-2023 (this model) | 68.5 | 
+
+
+## Pipeline description
+This system is composed of a fine-tuned XLSR model coupled with statistical pooling. A classifier, trained with NLL Loss, is applied on top of that.
+
+The system is trained with recordings sampled at 16kHz (single channel).
+The code will automatically normalize your audio (i.e., resampling + mono channel selection) when calling *classify_file* if needed. Make sure your input tensor is compliant with the expected sampling rate if you use *encode_batch* and *classify_batch*.
+
+## Install SpeechBrain
+
+First of all, please install SpeechBrain with the following command:
+
+```
+pip install speechbrain
+```
+
+Please notice that we encourage you to read our tutorials and learn more about
+[SpeechBrain](https://speechbrain.github.io).
+
+### Perform Accent Identification from Speech Recordings
+
+```python
+import torchaudio
+from speechbrain.pretrained import EncoderClassifier
+classifier = EncoderClassifier.from_hparams(source="Jzuluaga/accent-id-commonaccent_xlsr-spanish", savedir="pretrained_models/accent-id-commonaccent_xlsr-spanish")
+# Cuban Accent Example
+out_prob, score, index, text_lab = classifier.classify_file('Jzuluaga/accent-id-commonaccent_xlsr-spanish/data/mexico.wav')
+print(text_lab)
+
+# Caribean Example
+out_prob, score, index, text_lab = classifier.classify_file('Jzuluaga/accent-id-commonaccent_xlsr-spanish/data/caribe-cuba-colombia.wav')
+print(text_lab)
+```
+
+### Inference on GPU
+To perform inference on the GPU, add  `run_opts={"device":"cuda"}`  when calling the `from_hparams` method.
+
+### Training
+
+The model was trained with SpeechBrain.
+
+To train it from scratch follow these steps:
+
+1. Clone SpeechBrain:
+```bash
+git clone https://github.com/speechbrain/speechbrain/
+```
+
+2. Install it:
+```bash
+cd speechbrain
+pip install -r requirements.txt
+pip install -e .
+```
+
+3. Clone our repository in https://github.com/JuanPZuluaga/accent-recog-slt2022:
+
+```bash
+git clone https://github.com/JuanPZuluaga/accent-recog-slt2022
+cd CommonAccent/accent_id
+python train_w2v2.py hparams/train_w2v2.yaml
+```
+
+You can find our training results (models, logs, etc) in this repository's `Files and versions` page.
+
+### Limitations
+The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
+
+
+
+#### Cite our work: CommonAccent
+
+If you find useful this work, please cite our work as: 
+
+```
+@article{zuluaga2023commonaccent,
+  title={CommonAccent: Exploring Large Acoustic Pretrained Models for Accent Classification Based on Common Voice},
+  author={Zuluaga-Gomez, Juan and Ahmed, Sara and Visockas, Danielius and Subakan, Cem},
+  journal={Interspeech 2023},
+  url={https://arxiv.org/abs/2305.18283},
+  year={2023}
+}
+```
+
+#### Cite XLSR model
+
+```@article{conneau2020unsupervised,
+  title={Unsupervised cross-lingual representation learning for speech recognition},
+  author={Conneau, Alexis and Baevski, Alexei and Collobert, Ronan and Mohamed, Abdelrahman and Auli, Michael},
+  journal={arXiv preprint arXiv:2006.13979},
+  year={2020}
+}
+```
+
+
+# **Cite SpeechBrain**
+Please, cite SpeechBrain if you use it for your research or business.
+
+
+```bibtex
+@misc{speechbrain,
+  title={{SpeechBrain}: A General-Purpose Speech Toolkit},
+  author={Mirco Ravanelli and Titouan Parcollet and Peter Plantinga and Aku Rouhe and Samuele Cornell and Loren Lugosch and Cem Subakan and Nauman Dawalatabad and Abdelwahab Heba and Jianyuan Zhong and Ju-Chieh Chou and Sung-Lin Yeh and Szu-Wei Fu and Chien-Feng Liao and Elena Rastorgueva and François Grondin and William Aris and Hwidong Na and Yan Gao and Renato De Mori and Yoshua Bengio},
+  year={2021},
+  eprint={2106.04624},
+  archivePrefix={arXiv},
+  primaryClass={eess.AS},
+  note={arXiv:2106.04624}
+}
+```

accent_encoder.txt

@@ -0,0 +1,8 @@
+'ESPANA SUR PENINSULAR (ANDALUCIA EXTREMADURA MURCIA)' => 0
+'MEXICO' => 1
+'ANDINOPACIFICO COLOMBIA PERU ECUADOR OESTE DE BOLIVIA Y VENEZUELA ANDINA' => 2
+'CARIBE CUBA VENEZUELA PUERTO RICO REPUBLICA DOMINICANA PANAMA COLOMBIA CARIBENA MEXICO CARIBENO COSTA DEL GOLFO DE MEXICO' => 3
+'RIOPLATENSE ARGENTINA URUGUAY ESTE DE BOLIVIA PARAGUAY' => 4
+'CHILENO CHILE CUYO' => 5
+================
+'starting_index' => 0

config.json

@@ -0,0 +1,3 @@
+{
+    "speechbrain_interface": "EncoderClassifier"
+}

model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4d01453d1630748add9fb21725404fef576bac0b98951a049b402908bf866f8
+size 8423259

normalizer_input.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ea5fe8b249fd7932f9e6804b89618f519fac248273d00f0a36470b085c48d7c
+size 1127