sukju/cardiologent_ecg_specialist

Cardiologent — ECG Specialist

Two-stage ECG waveform specialist for arrhythmia description, trained on CardioBench.

• stage1_encoder.ckpt — self-supervised time-series encoder (TimeMAE, peak-weighted masked reconstruction).
• stage2_llm.ckpt — LLaVA-1.5-7B with the frozen encoder + a perceiver bridge, finetuned on per-window ECG QA (answer-only cross-entropy).

Given an ECG window it produces a free-text arrhythmia read over the 9-class taxonomy (Normal, AF, MAT, SVT, VT, PVC, PAC, AVB, SND). Used as the grounded specialist inside the Cardiologent diagnostic agent.

Training / inference code: see the CardioBench GitHub repo (specialist/). Data: sukju/cardiologent_dataset.