Downloads · 30 days
27
17% of all-time downloads
twangodev/rasr-parakeet-v2
rasr-parakeet-v2 is a automatic speech recognition model from twangodev. Use it when you need speech turned into text. It is set up for nemo. The card lists the license as llama3.2.
ATC ASR model targeting US air traffic control. A continued finetune of twangodev/rasr-parakeet-v1 on real US ATC audio from the TartanAviation corpus (CMU AirLab). v2 of the rasr toolkit.
Downloads · 30 days
27
17% of all-time downloads
All-time downloads
161
Public
Repo size
2.5 GB
Likes
1
Public
Click a slice to open those files.
.nemo2.5 GB · 100%
From the Hugging Face model README
ATC ASR model targeting US air traffic control. A continued finetune of twangodev/rasr-parakeet-v1 on real US ATC audio from the TartanAviation corpus (CMU AirLab). v2 of the rasr toolkit.
v1 was strong on European ATC but weak on US ATC, since its only real-audio anchor was European. v2 adds about 29 hours of real US tower audio to close that gap.
| Metric | v2 | v1 |
|---|---|---|
| liveATC WER (US) | 0.251 | 0.311 |
| liveATC CER (US) | 0.174 | 0.207 |
| ATCO2 val WER (EU) | 0.128 | 0.125 |
| ATCO2 val CER (EU) | 0.084 | 0.078 |
About 19% relative WER reduction on US ATC (liveATC) over v1, with European performance held flat. Both rows use rasr eval run on the same references, so the comparison is direct. One caveat: readback numeric WER regressed (see Limitations).
import nemo.collections.asr as nemo_asr
model = nemo_asr.models.ASRModel.from_pretrained("twangodev/rasr-parakeet-v2")
result = model.transcribe(["atc_clip.wav"])
print(result[0].text)
Or via the rasr eval toolkit:
pip install rasr
rasr eval run \
-m nemo:hf://twangodev/rasr-parakeet-v2 \
-d hf:SAadettin-BERber/liveATC_merged:train \
--language en --batch-size 16
nvidia/parakeet-tdt-0.6b-v3 (FastConformer encoder, TDT decoder, 0.6B params)twangodev/rasr-parakeet-v1 (this model is a continued finetune of v1)v2 starts from v1's weights and continues training on the mix below. v1's own training (synthetic transcripts plus a European real anchor) is inherited in the weights; see the v1 card.
| Source | Type | Role |
|---|---|---|
| TartanAviation, high inter-model agreement subset | Real US ATC (KAGC tower) | US training audio: 35,240 clips, 28.8 h |
jlvdoorn/atco2-asr-atcosim (train) | Real EU ATC plus simulator | European rehearsal |
jlvdoorn/atco2-asr (train) | Real EU ATC | European rehearsal |
TartanAviation ships no transcripts. The training targets are pseudo labels from a weighted ROVER fusion of three ASR models (nvidia/parakeet-tdt-0.6b-v2, nvidia/canary-qwen-2.5b, jlvdoorn/whisper-large-v3-atco2-asr) plus ADS-B callsign snapping, filtered to the highest inter-model agreement (twangodev/tartanaviation-atc-labels). On a human-reviewed sample this subset is about 98% acceptable, but these are not human labels. Evaluation clips are held out by date.
TartanAviation components used:
twangodev/tartanaviation-atc-adsb-utterances: segmented US ATC audio (16 kHz, with per-clip ADS-B tracks).twangodev/tartanaviation-atc-labels: the pseudo-label transcripts and agreement scores used as supervised targets.twangodev/tartanaviation-atc-adsb: ADS-B trajectory data, used for the callsign snapping in the labels.This model is "Built with Llama" under the Llama 3.2 Community License. v2 is a continued finetune of v1, whose supervised targets were ATC dialogue transcripts generated by Llama 3.2; v2 is therefore a derivative of those Llama-trained weights. The new US targets in v2 were not generated by Llama (they are the pseudo labels described above).
US training audio is from the TartanAviation dataset (CMU AirLab, CC-BY-4.0). Please credit CMU AirLab and cite:
@article{patrikar2024tartanaviation,
title={TartanAviation: Image, Speech, and ADS-B Trajectory Datasets for Terminal Airspace Operations},
author={Jay Patrikar and Joao Dantas and Brady Moon and Milad Hamidi and Sourish Ghosh and Nikhil Keetha and Ian Higgins and Atharva Chandak and Takashi Yoneyama and Sebastian Scherer},
year={2024},
eprint={2403.03372},
archivePrefix={arXiv},
primaryClass={cs.LG},
url={https://arxiv.org/pdf/2403.03372.pdf}
}
Continued finetune from v1 (rasr-parakeet-v1.nemo). Recipe: training_recipe.yaml (this repo).
| Hyperparameter | Value |
|---|---|
| Initialize from | twangodev/rasr-parakeet-v1 (continued finetune, not from base) |
| Optimizer | AdamW, betas (0.9, 0.98), weight_decay 1e-3 |
| Learning rate | 5e-5 |
| Schedule | CosineAnnealing, warmup 300 steps, min_lr 1e-6 |
| Batch size | 48 |
| Precision | bf16-mixed |
| Max steps | 10,000 |
| Augmentation | SpecAugment (default), speed perturb 0.95 to 1.05 |
| Max audio duration | 18.0 s |
| Mixing | weighted manifest concat (TartanAviation x1, ATCO2 train x1, ATCO2+ATCOSIM train x1) |
| Hardware | NVIDIA RTX PRO 6000 Blackwell (96 GB) |
| Wall clock | about 1 hour |
rasr eval run). v1's strength was numeric accuracy, and v2 trades some of it for the overall gain. A second normalization pipeline showed numeric roughly flat rather than worse, so the magnitude is uncertain, but it did not improve. For digit-critical use, evaluate v1 as well and layer hot-word or grammar constraints on number fields.change_decoding_strategy()).@software{rasr,
author = {Ding, James},
title = {rasr: ATC ASR finetuning toolkit},
url = {https://github.com/twangodev/rasr},
year = {2026}
}
Base model:
@misc{parakeet-tdt,
author = {NVIDIA},
title = {Parakeet-TDT-0.6B-v3},
url = {https://huggingface.co/nvidia/parakeet-tdt-0.6b-v3}
}
US training audio (TartanAviation) and Llama 3.2 (inherited from v1): see the attribution sections above.
Released under the Llama 3.2 Community License ("Built with Llama"). This is the binding upstream license: v2 is a continued finetune of v1, whose training transcripts were generated by Llama 3.2, so the weights are treated as a derivative work of Llama Materials.
This model also inherits attribution and use requirements from its other parents:
To redistribute or deploy:
This is not legal advice. If you are deploying this model commercially or at scale, consult a lawyer regarding the interaction of the upstream licenses.