Biomarker-Based Pretraining for Chagas Disease Screening in Electrocardiograms
Organizations: Medical Technology & E-health, Akershus University Hospital, Lørenskog, Norway · Faculty of Medicine, University of Oslo, Oslo, Norway
Abstract
Chagas disease screening via ECGs is limited by scarce and noisy labels in existing datasets. We propose a biomarker-based pretraining approach, where an ECG feature extractor is first trained to predict percentile-binned blood biomarkers from the MIMIC-IV-ECG dataset. The pretrained model is then fine-tuned on Brazilian datasets for Chagas detection. Our 5-model ensemble, developed by the Ahus AIM team, achieved a challenge score of 0.269 on the hidden test set, ranking 5th in Detection of Chagas Disease from the ECG: The George B. Moody PhysioNet Challenge 2025. Source code and the model are shared on GitHub: github.com/Ahus-AIM/physionet-challenge-2025
Figures & tables
| Biomarker | Clinical domain |
|---|---|
| Albumin | Liver function |
| Calcium, Total | Electrolytes |
| Creatinine | Renal |
| Hematocrit | Hematology |
| Hemoglobin | Hematology |
| INR(PT) | Coagulation |
| Dataset | # ECGs | # Patients | Step |
|---|---|---|---|
| MIMIC-IV-ECG | 523,275 | 102,511 | Pretraining |
| CODE15% | 345,779 | 233,770 | Fine-tuning |
| SaMi-Trop | 1,959 | 1,959 | Fine-tuning |
| Component | Specification |
|---|---|
| System | Debian 12 |
| CPU | Intel Core i9-14900KF |
| RAM | 2 48 GB; 4800 MT/s |
| GPU | NVIDIA GeForce RTX 5090 |
| CUDA version | 12.9 |
| Programming language | Python 3.12 |
| Parameter | Value |
|---|---|
| Batch size | 64 |
| Optimizer | Muon and Adam |
| Muon momentum | 0.95 |
| Learning rate | 0.0037 |
| Loss function | Cross-entropy |
| Parameter | Value |
|---|---|
| Batch size | 128 |
| Optimizer | Muon and Adam |
| Muon momentum | 0.95 |
| Learning rate | 0.001 |
| Loss function | Binary cross-entropy |
| Parameter | Value |
|---|---|
| Number of blocks | 6 |
| Kernel sizes | 9, 19, 39 |
| Number of filters | 32 |
| Bottleneck channels | 32 |
| Team | Challenge score | Rank |
|---|---|---|
| Ahus AIM | 0.269 | 5 |
Electrocardiogram Classification
Electrocardiogram classification uses machine learning to identify heart rhythms and cardiac conditions from ECG waveforms for screening, diagnosis, and monitoring.
See all 106 papers