CoolFace
Modelpublic

yuriyvnv/parakeet-tdt-0.6b-EN-Medical

sourceHugging Facecc-by-4.0updated 4mo agoView on Hugging Face
0likes22downloads
Model Card

๐Ÿฉบ Parakeet-TDT-0.6B English Medical ๐Ÿ‡ฌ๐Ÿ‡ง

A fine-tune of `nvidia/parakeet-tdt-0.6b-v3` on the English subset of MultiMed mixed with Common Voice 17 English (train + validation). The mix is the trick: it pushes the model toward medical vocabulary (TAVI, intervertebral disc herniation, drug names, dosing instructions) while keeping the everyday English it already knew.

Outputs cased English text with punctuation. Drop-in for the base Parakeet: same NeMo API, same long-form support, same timestamps.

๐Ÿ”ฅ Quick start

bash
pip install nemo_toolkit[asr]
python
import nemo.collections.asr as nemo_asr

asr = nemo_asr.models.ASRModel.from_pretrained("yuriyvnv/parakeet-tdt-0.6b-EN-Medical")
print(asr.transcribe(["audio.wav"])[0].text)

๐Ÿ“Š Results

One model, one training mix (MultiMed-en train + Common Voice 17-en train + validation, concatenated and shuffled per epoch โ€” same .nemo for every row below). The two rows are the same checkpoint evaluated on two different held-out test sets: one in-domain (medical) and one out-of-domain (general English). Neither test set was seen during training.

The zero-shot column is the unmodified `nvidia/parakeet-tdt-0.6b-v3`, measured on the same test set with the same evaluator. All numbers are normalized (lowercase + strip punctuation), the standard protocol used by the MultiMed paper and the Open ASR Leaderboard, so they are directly comparable to other published results.

Test set (held out)SamplesZero-shot WER**Fine-tuned WER**ยนฮ” WERZero-shot CER**Fine-tuned CER**ยน
๐Ÿฉบ MultiMed English (test)7,56719.2214.31-4.91 (-25.5%)14.1610.37
๐Ÿ—ฃ๏ธ Common Voice 17 EN (test)16,3938.589.11+0.53 (+6.2%)4.014.10

For reference, the MultiMed paper's best published result is Whisper-Small multilingual fine-tune at 16.62% WER (arXiv 2409.14074, Table 6). This model beats that by 2.31 absolute / 14% relative, at 0.6B params.

ยน Both fine-tuned numbers are this single model โ€” the one trained on the MultiMed-en + CV17-en mix.

Two things worth noting:

  • โ€”๐ŸŽฏ Medical: 25% relative WER reduction on MultiMed-en test (19.22% โ†’ 14.31%), beating the published MultiMed paper SOTA at 0.6B params.
  • โ€”๐Ÿ›ก๏ธ General English is essentially preserved. On CV17-en test the same model lands at 9.11% WER vs the base Parakeet's 8.58% โ€” a +0.53 absolute / +6.2% relative change. That's the total cost of medical specialisation. A previous medical-only version of this model (trained on MultiMed alone, no CV) scored 13.53% normalized WER on CV17-en โ€” a real catastrophic-forgetting hit. Including CV17-en in the training mix is what made the trade-off this cheap.

๐Ÿ› ๏ธ Training

PropertyValue
Base modelnvidia/parakeet-tdt-0.6b-v3
ArchitectureFastConformer-TDT (~600M params)
Training dataCV17-en (train + validation) + MultiMed-en (train), concatenated, shuffled per-epoch
Total train clips~1.07M (CV17-en ~97.5% / MultiMed-en ~2.5%)
ValidationMultiMed-en eval (~2,816 clips) โ€” drives early stopping
OptimiserAdamW (ฮฒ=0.9/0.98, weight decay 1e-3)
Learning rate5e-5 (cosine annealing, min 1e-6)
Warmup10% of total steps
Batch size32
Precisionbf16-mixed
Gradient clipping1.0
Max clip duration30 s
Early stoppingval_wer plateau, patience 10
Best epoch31 (val WER 22.84%)
HardwareSingle NVIDIA H100

The natural CV17:MultiMed ratio (~97.5% : ~2.5%) means each batch of 32 contains on average ~0.8 medical clips and the rest general English โ€” strong CV gradient keeps the base distribution intact while medical clips slowly steer the model into the clinical domain.

๐ŸŽฏ Intended use

  • โ€”Transcription of English medical speech: clinical consultations, surgical procedures, patient narratives, medical podcasts, healthcare interviews.
  • โ€”A drop-in replacement for nvidia/parakeet-tdt-0.6b-v3 in pipelines targeting medical English โ€” picks up clinical vocabulary without breaking general transcription.

๐Ÿ“š Training data

๐Ÿ™ Acknowledgements

  • โ€”NVIDIA NeMo team for releasing Parakeet-TDT-0.6B-v3 and the NeMo toolkit.
  • โ€”Khai Le-Duc and the MultiMed authors for releasing the MultiMed multilingual medical ASR dataset.
  • โ€”The Mozilla Common Voice community for the crowd-sourced English speech corpus used as the general-English anchor in the training mix.