Skip to content

NabJa/calibrating-ecg-models

Repository files navigation

Monte Carlo Sampling for Classification

Monte Carlo Sampling improves uncertainty estimation and calibration in ECG classification tasks.

This repository contains code to perform Monte Carlo sampling for uncertainty estimation in ECG classification tasks. The implementation is based on the work by Yarin Gal and Zoubin Ghahramani.

News

This work has been accepted in the European Heart Journal – Digital Health (2025): From Clinic to Couch: An Uncertainty-Aware Deep Learning Approach for ECG Analysis Across Modalities

  • 👉 Link to article (to be added)

Results

MC-Dropout improves model calibration

Calibration Plot

Small models are better calibrated than large models

Uncertainty Plot

About

Calibration and uncertainty estimation for ECG classifcation models

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages