compulsi0n/heart-failure-audio
Viewer • Updated • 1.77k • 18
How to use compulsi0n/whisper-hf-lora with PEFT:
from peft import PeftModel
from transformers import AutoModelForSeq2SeqLM
base_model = AutoModelForSeq2SeqLM.from_pretrained("openai/whisper-large-v3-turbo")
model = PeftModel.from_pretrained(base_model, "compulsi0n/whisper-hf-lora")This model is a fine-tuned version of openai/whisper-large-v3-turbo on compulsion/heart-failure-audio. It achieves the following results on the evaluation set:
A PEFT LoRA adapter of whisper-large-v3-turbo finetuned on heart failure audio data that is conversational, longitudinal, and focused on chronic illness management and care coordination in a community-based healthcare setting.
To be used in ASR tasks specifically in the heart failure domain.
Normalized for PHI redactions and throught Transformer's BasicTextNormalizer.
| Model | Raw WER (%) | Normalised WER (%) |
|---|---|---|
| Baseline | 35.00 | 26.71 |
| LoRA | 28.61 | 22.68 |
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Wer |
|---|---|---|---|---|
| 2.5146 | 1.0 | 92 | 2.3205 | 0.2746 |
| 2.4798 | 2.0 | 184 | 2.1513 | 0.2734 |
| 2.1105 | 3.0 | 276 | 1.7245 | 0.2647 |
| 1.6642 | 4.0 | 368 | 1.2785 | 0.2463 |
| 1.2627 | 5.0 | 460 | 1.0579 | 0.2395 |
| 1.076 | 6.0 | 552 | 0.9416 | 0.2771 |
| 1.0108 | 7.0 | 644 | 0.8642 | 0.2746 |
| 0.9166 | 8.0 | 736 | 0.7572 | 0.2624 |