Establishing a generalizable and universal fine-tuning framework for pre-trained vision networks in the ECG classification for Cardiovascular Disease(CVD) is crucial for advancing reliability and effectiveness in medical imaging. This paper proposes deep feature extraction, optimization, and uncertainty analysis along with transfer learning approaches for categorization of ECG images with pre-trained convolutional neural networks (CNN). On a labeled dataset of 1,376 ECG images split into four diagnostic classes, five well-known architectures, ResNet50, EfficientNetB0, DenseNet121, VGG16, and MobileNetV2, were systematically assessed, implying uncertainty quantification for enhanced explainability and calibration, and the framework was further analyzed on several datasets. The two-stage approach used in the classification pipeline is the first feature extraction using frozen base layers, followed by the selective fine-tuning of the deeper layers. The confusion matrix and the ROC-AUC studies support ResNet50’s exceptional classification performance among all assessed models, attesting to a validation accuracy of 99.05% and a balanced accuracy of 0.99 with uncertainty quantification (Mean Total Uncertainty: 0.0612, Mean Epistemic Uncertainty: 0.0090, Mean Aleatoric Uncertainty: 0.0523, Mean Confidence: 0.9867, Calibration Error: 0.0064, Uncertainty Separation: 0.6499). Especially in situations limited by limited annotated data, the results highlight the efficiency of using pre-trained visual networks for medical picture categorization. Using deep transfer learning, this framework lays the groundwork for scalable, high-precision ECG-based diagnostic systems.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Deep Feature Extraction and Optimization Framework for ECG Classification via Pretrained Vision Networks

  • Md. Rahat,
  • Md. Minhaj Ul Islam,
  • Md Zahid Akon,
  • Dr. Rahat Hossain Faisal

摘要

Establishing a generalizable and universal fine-tuning framework for pre-trained vision networks in the ECG classification for Cardiovascular Disease(CVD) is crucial for advancing reliability and effectiveness in medical imaging. This paper proposes deep feature extraction, optimization, and uncertainty analysis along with transfer learning approaches for categorization of ECG images with pre-trained convolutional neural networks (CNN). On a labeled dataset of 1,376 ECG images split into four diagnostic classes, five well-known architectures, ResNet50, EfficientNetB0, DenseNet121, VGG16, and MobileNetV2, were systematically assessed, implying uncertainty quantification for enhanced explainability and calibration, and the framework was further analyzed on several datasets. The two-stage approach used in the classification pipeline is the first feature extraction using frozen base layers, followed by the selective fine-tuning of the deeper layers. The confusion matrix and the ROC-AUC studies support ResNet50’s exceptional classification performance among all assessed models, attesting to a validation accuracy of 99.05% and a balanced accuracy of 0.99 with uncertainty quantification (Mean Total Uncertainty: 0.0612, Mean Epistemic Uncertainty: 0.0090, Mean Aleatoric Uncertainty: 0.0523, Mean Confidence: 0.9867, Calibration Error: 0.0064, Uncertainty Separation: 0.6499). Especially in situations limited by limited annotated data, the results highlight the efficiency of using pre-trained visual networks for medical picture categorization. Using deep transfer learning, this framework lays the groundwork for scalable, high-precision ECG-based diagnostic systems.