word_timestamps = output[0].timestamp['word'] # word level timestamps for first sample segment_timestamps = output[0].timestamp['segment'] # segment level timestamps
for stamp in segment_timestamps: print(f"{stamp['start']}s - {stamp['end']}s : {stamp['segment']}")
#### Translating with timestamps
To translate with timestamps:
```python
output = asr_model.transcribe(['2086-149220-0033.wav'], source_lang='en', target_lang='fr', timestamps=True)
segment_timestamps = output[0].timestamp['segment'] # only supports segment level timestamps for translation
for stamp in segment_timestamps:
print(f"{stamp['start']}s - {stamp['end']}s : {stamp['segment']}")
For translation task, please, refer to segment-level timestamps for getting intuitive and accurate alignment.
Note: If timestamps are not required for your work, you can reduce memory usage by restoring only the
.nemofile without the auxiliary CTC model. To do this, extract the.nemofile, remove any timestamps_asr_model files, then repackage it into a new.nemofile.
Runtime Engine(s):
Supported Hardware Microarchitecture Compatibility:
[Preferred/Supported] Operating System(s):
Hardware Specific Requirements: At least 6GB RAM for model to load.
Current version: Canary-1b-v2. Previous versions can be accessed here.
The model was trained using the NeMo toolkit [4], following a 3-stage training procedure:
For all the stages of training, both languages and corpora are weighted using temperature sampling (τ = 0.5).
Training script: speech_to_text_aed.py
Tokenizer script: process_asr_text_tokenizer.py
Canary-1b-v2 was trained on a massive multilingual speech recognition and translation dataset combining Nvidia's newly published Granary and in-house dataset NeMo ASR Set 3.0.
Granary Dataset [5] [6] with improved pseudo-labels and efficiently filtered versions of the following corpora:
Granary is now available on Hugging Face.
To read more about the pseudo-labeling technique and pipeline, please refer to the Granary Paper.
NeMo ASR Set 3.0 including human-labeled transcriptions from the following corpora:
Total training hours: 1.7M
All transcripts include punctuation and capitalization.
Data Collection Method by dataset
Labeling Method by dataset
Data Collection Method by dataset
Labeling Method by dataset
This section reports the evaluation results of the Canary-1b-v2 model across multiple tasks, including Automatic Speech Recognition (ASR), Speech Translation (AST), robustness to noise, and long-form transcription.
| WER ↓ | Fleurs-25 Langs | CoVoST-13 Langs | MLS - 6 Langs |
|---|---|---|---|
Canary-1b-v2 | 8.40% | 8.85% | 7.27% |
Note: Presented WERs do not include Punctuation and Capitalization errors.
| WER ↓ | RTFx | Mean | AMI | GigaSpeech | LS Clean | LS Other | Earnings22 | SPGISpech | Tedlium | Voxpopuli |
|---|---|---|---|---|---|---|---|---|---|---|
Canary-1b-v2 | 749 | 7.15 | 16.01 | 10.82 | 2.18 | 3.56 | 11.79 | 2.28 | 4.29 | 6.25 |
More details on evaluation can be found at HuggingFace ASR Leaderboard
| COMET ↑ | BLEU ↑ | |||
|---|---|---|---|---|
| Fleurs-24 Langs | CoVoST-13 Langs | Fleurs-24 Langs | CoVoST-13 Langs | |
Canary-1b-v2 | 79.30 | 77.48 | 29.08 | 40.48 |
| COMET ↑ | BLEU ↑ | |||
|---|---|---|---|---|
| Fleurs-24 Langs | CoVoST-5 Langs | Fleurs-24 Langs | CoVoST-5 Langs | |
Canary-1b-v2 | 84.56 | 80.29 | 29.4 | 32.33 |
Performance across different Signal-to-Noise Ratios (SNR) using MUSAN music and noise samples [16] on the LibriSpeech Clean test set. Metric: Word Error Rate (WER)
| SNR (dB) | 100 | 10 | 5 | 0 | -5 |
|---|---|---|---|---|---|
Canary-1b-v2 | 2.18% | 2.29% | 2.80% | 5.08% | 19.38% |
Number of characters per minute on MUSAN [16] 48 hrs eval set:
| # of character per minute ↓ | |
|---|---|
Canary-1b-v2 | 134.7 |
Canary-1b-v2 achieves strong performance on long-form transcription by using dynamic chunking with 1-second overlap between chunks, allowing for efficient parallel processing. This dynamic chunking feature is automatically enabled when calling .transcribe() on a single audio file, or when using batch_size=1 with multiple audio files that are longer than 40 seconds.
| Dataset | WER ↓ |
|---|---|
| Earnings-22 | 13.78% |
| This American Life | 9.87% |
Note: Presented WERs do not include Punctuation and Capitalization errors.
Engine:
Test Hardware:
NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their supporting model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.
For more detailed information on ethical considerations for this model, please see the Model Card++ Explainability, Bias, Safety & Security, and Privacy Subcards here.
Please report security vulnerabilities or NVIDIA AI Concerns here.
| Field | Response |
|---|---|
| Participation considerations from adversely impacted groups protected classes in model design and testing | None |
| Measures taken to mitigate against unwanted bias | None |
| Field | Response |
|---|---|
| Intended Domain | Speech to Text Transcription and Translation |
| Model Type | Attention Encoder-Decoder |
| Intended Users | This model is intended for developers, researchers, academics, and industries building conversational based applications. |
| Output | Text |
| Describe how the model works | Speech input is encoded into embeddings and passed into conformer-based model and output a text response. |
| Name the adversely impacted groups this has been tested to deliver comparable outcomes regardless of | Not Applicable |
| Technical Limitations & Mitigation | Transcripts and translations may be not 100% accurate. Accuracy varies based on source and target language and characteristics of input audio (Domain, Use Case, Accent, Noise, Speech Type, Context of speech, etc.) |
| Verified to have met prescribed NVIDIA quality standards | Yes |
| Performance Metrics | Word Error Rate (Speech Transcription) / BLEU score (Speech Translation) / COMET score (Speech Translation) |
| Potential Known Risks | If a word is not trained in the language model and not presented in vocabulary, the word is not likely to be recognized. Not recommended for word-for-word/incomplete sentences as accuracy varies based on the context of input text |
| Licensing | GOVERNING TERMS: Use of this model is governed by the CC-BY-4.0 license. |
| Field | Response |
|---|---|
| Generatable or reverse engineerable personal data? | None |
| Personal data used to create this model? | None |
| Is there provenance for all datasets used in training? | Yes |
| Does data labeling (annotation, metadata) comply with privacy laws? | Yes |
| Is data compliant with data subject requests for data correction or removal, if such a request was made? | No, not possible with externally-sourced data. |
| Applicable Privacy Policy | https://www.nvidia.com/en-us/about-nvidia/privacy-policy/ |
| Field | Response |
|---|---|
| Model Application(s) | Speech to Text Transcription |
| Describe the life critical impact | None |
| Use Case Restrictions | Abide by CC-BY-4.0 License |
| Model and dataset restrictions | The Principle of least privilege (PoLP) is applied limiting access for dataset generation and model development. Restrictions enforce dataset access during training, and dataset license constraints adhered to. |
[1] Granary: Speech Recognition and Translation Dataset in 25 European Languages
[2] NVIDIA Granary Dataset Card
[3] Fast Conformer with Linearly Scalable Attention for Efficient Speech Recognition
[5] Google Sentencepiece Tokenizer
[7] Youtube-Commons
[8] MOSEL: 950,000 Hours of Speech Data for Open-Source Speech Foundation Model Training on EU Languages
[9] YODAS: Youtube-Oriented Dataset for Audio and Speech
[10] FLEURS: Few-shot Learning Evaluation of Universal Representations of Speech
[11] MLS: A Large-Scale Multilingual Dataset for Speech Research
[12] CoVoST 2 and Massively Multilingual Speech-to-Text Translation
[13] HuggingFace Open ASR Leaderboard
[15] Speech Recognition and Multi-Speaker Diarization of Long Conversations