initial commit

README.md

@@ -0,0 +1,258 @@
+---
+language: multilingual
+thumbnail:
+tags:
+- audio-classification
+- speechbrain
+- embeddings
+- Language
+- Identification
+- pytorch
+- ECAPA-TDNN
+- TDNN
+- VoxLingua107
+license: "apache-2.0"
+datasets:
+- VoxLingua107
+metrics:
+- Accuracy
+widget:
+- label: English Sample
+  src: https://cdn-media.huggingface.co/speech_samples/LibriSpeech_61-70968-0000.flac
+---
+
+# VoxLingua107 ECAPA-TDNN Spoken Language Identification Model (CE)
+
+## Model description
+
+This is a spoken language recognition model trained on the VoxLingua107 dataset using SpeechBrain.
+The model uses the ECAPA-TDNN architecture that has previously been used for speaker recognition. However, it uses
+more fully connected hidden layers after the embedding layer, and cross-entropy loss was used for training. 
+We observed that this improved the performance of extracted utterance embeddings for downstream tasks.
+
+The model can classify a speech utterance according to the language spoken.
+It covers 107 different languages (
+Abkhazian, 
+Afrikaans, 
+Amharic, 
+Arabic, 
+Assamese, 
+Azerbaijani, 
+Bashkir, 
+Belarusian, 
+Bulgarian, 
+Bengali, 
+Tibetan, 
+Breton, 
+Bosnian, 
+Catalan, 
+Cebuano, 
+Czech, 
+Welsh, 
+Danish, 
+German, 
+Greek, 
+English, 
+Esperanto, 
+Spanish, 
+Estonian, 
+Basque, 
+Persian, 
+Finnish, 
+Faroese, 
+French, 
+Galician, 
+Guarani, 
+Gujarati, 
+Manx, 
+Hausa, 
+Hawaiian, 
+Hindi, 
+Croatian, 
+Haitian, 
+Hungarian, 
+Armenian, 
+Interlingua, 
+Indonesian, 
+Icelandic, 
+Italian, 
+Hebrew, 
+Japanese, 
+Javanese, 
+Georgian, 
+Kazakh, 
+Central Khmer, 
+Kannada, 
+Korean, 
+Latin, 
+Luxembourgish, 
+Lingala, 
+Lao, 
+Lithuanian, 
+Latvian, 
+Malagasy, 
+Maori, 
+Macedonian, 
+Malayalam, 
+Mongolian, 
+Marathi, 
+Malay, 
+Maltese, 
+Burmese, 
+Nepali, 
+Dutch, 
+Norwegian Nynorsk, 
+Norwegian, 
+Occitan, 
+Panjabi, 
+Polish, 
+Pushto, 
+Portuguese, 
+Romanian, 
+Russian, 
+Sanskrit, 
+Scots, 
+Sindhi, 
+Sinhala, 
+Slovak, 
+Slovenian, 
+Shona, 
+Somali, 
+Albanian, 
+Serbian, 
+Sundanese, 
+Swedish, 
+Swahili, 
+Tamil, 
+Telugu, 
+Tajik, 
+Thai, 
+Turkmen, 
+Tagalog, 
+Turkish, 
+Tatar, 
+Ukrainian, 
+Urdu, 
+Uzbek, 
+Vietnamese, 
+Waray, 
+Yiddish, 
+Yoruba, 
+Mandarin Chinese).
+
+## Intended uses & limitations
+
+The model has two uses:
+
+  - use 'as is' for spoken language recognition
+  - use as an utterance-level feature (embedding) extractor, for creating a dedicated language ID model on your own data
+  
+The model is trained on automatically collected YouTube data. For more 
+information about the dataset, see [here](http://bark.phon.ioc.ee/voxlingua107/).
+
+
+#### How to use
+
+```python
+import torchaudio
+from speechbrain.pretrained import EncoderClassifier
+language_id = EncoderClassifier.from_hparams(source="speechbrain/lang-id-voxlingua107-ecapa", savedir="tmp")
+# Download Thai language sample from Omniglot and cvert to suitable form
+signal = language_id.load_audio("https://omniglot.com/soundfiles/udhr/udhr_th.mp3")
+prediction =  language_id.classify_batch(signal)
+print(prediction)
+  (tensor([[-2.8646e+01, -3.0346e+01, -2.0748e+01, -2.9562e+01, -2.2187e+01,
+         -3.2668e+01, -3.6677e+01, -3.3573e+01, -3.2545e+01, -2.4365e+01,
+         -2.4688e+01, -3.1171e+01, -2.7743e+01, -2.9918e+01, -2.4770e+01,
+         -3.2250e+01, -2.4727e+01, -2.6087e+01, -2.1870e+01, -3.2821e+01,
+         -2.2128e+01, -2.2822e+01, -3.0888e+01, -3.3564e+01, -2.9906e+01,
+         -2.2392e+01, -2.5573e+01, -2.6443e+01, -3.2429e+01, -3.2652e+01,
+         -3.0030e+01, -2.4607e+01, -2.2967e+01, -2.4396e+01, -2.8578e+01,
+         -2.5153e+01, -2.8475e+01, -2.6409e+01, -2.5230e+01, -2.7957e+01,
+         -2.6298e+01, -2.3609e+01, -2.5863e+01, -2.8225e+01, -2.7225e+01,
+         -3.0486e+01, -2.1185e+01, -2.7938e+01, -3.3155e+01, -1.9076e+01,
+         -2.9181e+01, -2.2160e+01, -1.8352e+01, -2.5866e+01, -3.3636e+01,
+         -4.2016e+00, -3.1581e+01, -3.1894e+01, -2.7834e+01, -2.5429e+01,
+         -3.2235e+01, -3.2280e+01, -2.8786e+01, -2.3366e+01, -2.6047e+01,
+         -2.2075e+01, -2.3770e+01, -2.2518e+01, -2.8101e+01, -2.5745e+01,
+         -2.6441e+01, -2.9822e+01, -2.7109e+01, -3.0225e+01, -2.4566e+01,
+         -2.9268e+01, -2.7651e+01, -3.4221e+01, -2.9026e+01, -2.6009e+01,
+         -3.1968e+01, -3.1747e+01, -2.8156e+01, -2.9025e+01, -2.7756e+01,
+         -2.8052e+01, -2.9341e+01, -2.8806e+01, -2.1636e+01, -2.3992e+01,
+         -2.3794e+01, -3.3743e+01, -2.8332e+01, -2.7465e+01, -1.5085e-02,
+         -2.9094e+01, -2.1444e+01, -2.9780e+01, -3.6046e+01, -3.7401e+01,
+         -3.0888e+01, -3.3172e+01, -1.8931e+01, -2.2679e+01, -3.0225e+01,
+         -2.4995e+01, -2.1028e+01]]), tensor([-0.0151]), tensor([94]), ['th'])
+# The scores in the prediction[0] tensor can be interpreted as log-likelihoods that
+# the given utterance belongs to the given language (i.e., the larger the better)
+# The linear-scale likelihood can be retrieved using the following:
+print(prediction[1].exp())
+  tensor([0.9850])
+# The identified language ISO code is given in prediction[3]
+print(prediction[3])
+  ['th']
+  
+# Alternatively, use the utterance embedding extractor:
+emb =  language_id.encode_batch(signal)
+print(emb.shape)
+  torch.Size([1, 1, 256])
+```
+To perform inference on the GPU, add  `run_opts={"device":"cuda"}`  when calling the `from_hparams` method.
+
+
+#### Limitations and bias
+
+Since the model is trained on VoxLingua107, it has many limitations and biases, some of which are:
+
+ - Probably it's accuracy on smaller languages  is quite limited
+ - Probably it works worse on female speech than male speech (because YouTube data includes much more male speech)
+ - Based on subjective experiments, it doesn't work well on speech with a foreign accent
+ - Probably it doesn't work well on children's speech and on persons with speech disorders
+
+
+## Training data
+
+The model is trained on [VoxLingua107](http://bark.phon.ioc.ee/voxlingua107/).
+
+VoxLingua107 is a speech dataset for training spoken language identification models. 
+The dataset consists of short speech segments automatically extracted from YouTube videos and labeled according the language of the video title and description, with some post-processing steps to filter out false positives.
+
+VoxLingua107 contains data for 107 languages. The total amount of speech in the training set is 6628 hours. 
+The average amount of data per language is 62 hours. However, the real amount per language varies a lot. There is also a seperate development set containing 1609 speech segments from 33 languages, validated by at least two volunteers to really contain the given language.
+
+## Training procedure
+
+See the [SpeechBrain recipe](https://github.com/speechbrain/speechbrain/tree/voxlingua107/recipes/VoxLingua107/lang_id).
+
+## Evaluation results
+
+Error rate: 6.7% on the VoxLingua107 development dataset
+
+#### Referencing SpeechBrain
+```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}
+}
+```
+
+### Referencing VoxLingua107
+
+```bibtex
+@inproceedings{valk2021slt,
+  title={{VoxLingua107}: a Dataset for Spoken Language Recognition},
+  author={J{\"o}rgen Valk and Tanel Alum{\"a}e},
+  booktitle={Proc. IEEE SLT Workshop},
+  year={2021},
+}
+```
+
+#### About SpeechBrain
+SpeechBrain is an open-source and all-in-one speech toolkit. It is designed to be simple, extremely flexible, and user-friendly. Competitive or state-of-the-art performance is obtained in various domains.
+Website: https://speechbrain.github.io/
+GitHub: https://github.com/speechbrain/speechbrain

classifier.ckpt

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

embedding_model.ckpt

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

hyperparams.yaml

@@ -0,0 +1,57 @@
+pretrained_path: speechbrain/lang-id-voxlingua107-ecapa
+
+
+# Feature parameters
+n_mels: 60
+left_frames: 0
+right_frames: 0
+deltas: false
+
+# Number of speakers
+out_n_neurons: 107
+
+# Functions
+compute_features: !new:speechbrain.lobes.features.Fbank
+  n_mels: 60
+  left_frames: 0
+  right_frames: 0
+  deltas: false
+
+embedding_model: !new:speechbrain.lobes.models.ECAPA_TDNN.ECAPA_TDNN
+  input_size: 60
+  channels: [1024, 1024, 1024, 1024, 3072]
+  kernel_sizes: [5, 3, 3, 3, 1]
+  dilations: [1, 2, 3, 4, 1]
+  attention_channels: 128
+  lin_neurons: 256
+
+
+classifier: !new:speechbrain.lobes.models.Xvector.Classifier
+  input_shape: [null, null, 256]
+  activation: !name:torch.nn.LeakyReLU
+  lin_blocks: 1
+  lin_neurons: 512
+  out_neurons: !ref <out_n_neurons>
+
+
+mean_var_norm: !new:speechbrain.processing.features.InputNormalization
+  norm_type: sentence
+  std_norm: false
+
+modules:
+    compute_features: !ref <compute_features>
+    mean_var_norm: !ref <mean_var_norm>
+    embedding_model: !ref <embedding_model>
+    classifier: !ref <classifier>
+
+label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
+
+pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
+    loadables:
+        embedding_model: !ref <embedding_model>
+        classifier: !ref <classifier>
+        label_encoder: !ref <label_encoder>
+    paths:
+        embedding_model: !ref <pretrained_path>/embedding_model.ckpt
+        classifier: !ref <pretrained_path>/classifier.ckpt
+        label_encoder: !ref <pretrained_path>/label_encoder.txt

label_encoder.txt

@@ -0,0 +1,109 @@
+'ab' => 0
+'af' => 1
+'am' => 2
+'ar' => 3
+'as' => 4
+'az' => 5
+'ba' => 6
+'be' => 7
+'bg' => 8
+'bn' => 9
+'bo' => 10
+'br' => 11
+'bs' => 12
+'ca' => 13
+'ceb' => 14
+'cs' => 15
+'cy' => 16
+'da' => 17
+'de' => 18
+'el' => 19
+'en' => 20
+'eo' => 21
+'es' => 22
+'et' => 23
+'eu' => 24
+'fa' => 25
+'fi' => 26
+'fo' => 27
+'fr' => 28
+'gl' => 29
+'gn' => 30
+'gu' => 31
+'gv' => 32
+'ha' => 33
+'haw' => 34
+'hi' => 35
+'hr' => 36
+'ht' => 37
+'hu' => 38
+'hy' => 39
+'ia' => 40
+'id' => 41
+'is' => 42
+'it' => 43
+'iw' => 44
+'ja' => 45
+'jw' => 46
+'ka' => 47
+'kk' => 48
+'km' => 49
+'kn' => 50
+'ko' => 51
+'la' => 52
+'lb' => 53
+'ln' => 54
+'lo' => 55
+'lt' => 56
+'lv' => 57
+'mg' => 58
+'mi' => 59
+'mk' => 60
+'ml' => 61
+'mn' => 62
+'mr' => 63
+'ms' => 64
+'mt' => 65
+'my' => 66
+'ne' => 67
+'nl' => 68
+'nn' => 69
+'no' => 70
+'oc' => 71
+'pa' => 72
+'pl' => 73
+'ps' => 74
+'pt' => 75
+'ro' => 76
+'ru' => 77
+'sa' => 78
+'sco' => 79
+'sd' => 80
+'si' => 81
+'sk' => 82
+'sl' => 83
+'sn' => 84
+'so' => 85
+'sq' => 86
+'sr' => 87
+'su' => 88
+'sv' => 89
+'sw' => 90
+'ta' => 91
+'te' => 92
+'tg' => 93
+'th' => 94
+'tk' => 95
+'tl' => 96
+'tr' => 97
+'tt' => 98
+'uk' => 99
+'ur' => 100
+'uz' => 101
+'vi' => 102
+'war' => 103
+'yi' => 104
+'yo' => 105
+'zh' => 106
+================
+'starting_index' => 0

normalizer.ckpt

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