Back

Brugada ECG detection with self-supervised VICReg pre-training: a novel deep learning approach for rare cardiac diseases

Ronan, R.; Tarabanis, C.; Chinitz, L.; Jankelosn, L.

2024-03-30 cardiovascular medicine
10.1101/2024.03.29.24305072 medRxiv
Show abstract

Existing deep learning algorithms for electrocardiogram (ECG) classification rely on supervised training approaches requiring large volumes of reliably labeled data. This limits their applicability to rare cardiac diseases like Brugada syndrome (BrS), often lacking accurately labeled ECG examples. To address labeled data constraints and the resulting limitations of supervised training approaches, we developed a novel deep learning model for BrS ECG classification using the Variance-Invariance-Covariance Regularization (VICReg) architecture for self-supervised pre-training. The VICReg model outperformed a state-of-the-art neural network in all calculated metrics, achieving an area under the receiver operating and precision-recall curves of 0.88 and 0.82, respectively. We used the VICReg model to identify missed BrS cases and hence refine the previously underestimated institutional BrS prevalence and patient outcomes. Our results provide a novel approach to rare cardiac disease identification and challenge existing BrS prevalence estimates offering a framework for other rare cardiac conditions.

Matching journals

The top 4 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.