Identifying Hypertrophic or Dilated Cardiomyopathy: Development and Validation of a Fine-Tuned ResNet50 Model Based

Jiayu Xu1, Bo Chen2, Weiyang Liu1

  • 1Graduate School, Chinese People's Liberation Army General Hospital, Beijing 100853, China.

Insights

A novel deep learning model can identify hypertrophic cardiomyopathy (HCM) and dilated cardiomyopathy (DCM) using standard electrocardiogram (ECG) images. This AI tool shows promise for inexpensive and early detection of these heart conditions.

Area of Science:

  • Cardiology
  • Artificial Intelligence
  • Medical Imaging

Background:

  • Hypertrophic cardiomyopathy (HCM) and dilated cardiomyopathy (DCM) lack established non-invasive diagnostic tools.
  • Early and accurate detection of HCM and DCM is crucial for patient management and outcomes.

Purpose of the Study:

  • To develop and validate a deep learning model for identifying HCM and DCM using 12-lead electrocardiogram (ECG) images.
  • To assess the model's performance and identify key ECG regions contributing to predictions.

Main Methods:

  • A ResNet50 deep learning architecture was fine-tuned using 2849 ECG images (171 HCM, 364 DCM, 2314 controls).
  • Stratified five-fold cross-validation was employed for robust model training and testing.
  • Gradient-weighted Class Activation Mapping (Grad-CAM) was used for model interpretability.

Main Results:

  • The model achieved high accuracy in distinguishing DCM (AUROC 0.996, AUPRC 0.940) and HCM (AUROC 0.980, AUPRC 0.953).
  • Temporal validation confirmed the model's prospective stability.
  • Grad-CAM analysis highlighted anterior and anteroseptal leads as critical for predictions.

Conclusions:

  • A fine-tuned ResNet50 model effectively identifies HCM and DCM from standard ECGs.
  • This AI-driven approach offers a promising, cost-effective method for early detection of these cardiomyopathies.
  • The model's interpretability provides insights into ECG markers for HCM and DCM.