Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Tang-Jia-Lu/Zeta

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Interpretable Multimodal Zero-Shot ECG Diagnosis via Structured Clinical Knowledge Alignment

1 Eindhoven University of Technology, Netherland

2Singapore Management University, Singapore; 3Maxima Medical Center, Netherland; 4Erasmus Medical Center, Netherland;

📝 Abstract

Electrocardiogram (ECG) interpretation is essential for cardiovascular disease diagnosis, but current automated systems often struggle with transparency and generalization to unseen conditions. To address this, we introduce ZETA, a zero-shot multimodal framework designed for interpretable ECG diagnosis aligned with clinical workflows. ZETA uniquely compares ECG signals against structured positive and negative clinical observations, which are curated through an LLM-assisted, expert-validated process, thereby mimicking differential diagnosis. Our approach leverages a pre-trained multimodal model to align ECG and text embeddings without disease-specific fine-tuning. Empirical evaluations demonstrate ZETA's competitive zero-shot classification performance and, importantly, provide qualitative and quantitative evidence of enhanced interpretability, grounding predictions in specific, clinically relevant positive and negative diagnostic features. ZETA underscores the potential of aligning ECG analysis with structured clinical knowledge for building more transparent, generalizable, and trustworthy AI diagnostic systems. We will release the curated observation dataset and code to facilitate future research.

📚 Three Features

  • 🩺 Clinical Grounding: Structured positive/negative observations emulate differential diagnosis, enhancing medical reasoning.
  • 🧠 Expert-in-the-Loop Curation: LLM-guided + expert-validated annotation pipeline ensures trustworthy and clinically relevant supervision.
  • 📊 Transparent Evaluation: Interpretability metrics and qualitative diagnostics support actionable insights beyond accuracy.

🔧 Requirements

Dataset

Datasets we used:

  • PTB-XL: We downloaded the PTB-XL dataset which consisting four subsets, Superclass, Subclass, Form, Rhythm.

  • CPSC2018: We downloaded the CPSC2018 dataset which consisting three training sets.

  • CSN(Chapman-Shaoxing-Ningbo): We downloaded the CSN dataset.

💡 Running scripts

cd ./ZETA
python main.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages