An interpretable and explainable neural network to classify sports-related cardiac arrhythmias in professional football athletes
Vanegas Mueller, E.; Harford, M.; He, L.; Banerjee, A.; Leeson, P.; Villarroel, M.
Show abstract
Sudden cardiac death risk is 2-3-fold higher in athletes than in non-athletes. We classify sports-related cardiac arrhythmias using a novel explainability framework comprising data analysis, model interpretability, post-hoc visualisation, and systematic assessment. Two neural networks--one with interpretable sinc convolution and one with standard convolution--were trained on general-population ECGs (PhysioNet, n=88,253, 30 arrhythmias, three continents) and tested on professional footballers (PF12RED, n=161) via domain adaptation for normal sinus rhythm (NSR), sinus bradycardia (SB), incomplete right bundle branch block (IRBBB), and T-wave inversion (TWI). Sinc convolution achieved superior NSR detection (AUROC 0.75 vs 0.70), whilst standard convolution excelled at SB (0.74 vs 0.73), IRBBB (0.66 vs 0.58), and TWI (0.59 vs 0.54). Gradient-weighted Class Activation Mapping revealed that sinc models focus on physiologically relevant ECG segments (the PR interval for NSR/SB and the T wave for TWI). We hypothesise that sinc convolution better captures periodic rhythms but struggles with complex morphological patterns, suggesting architectural choice should align with underlying cardiac pathophysiology. Graphical abstractAbbreviations: AI, artificial intelligence; AUPRC, area under the precision-recall curve; AUROC, area under the receiver operating characteristic curve; Conv, convolution; ECG, electrocardiogram; Grad-CAM, gradient-weighted class activation mapping; IAVB, first-degree atrioventricular block; IRBBB, incomplete right bundle branch block; LAD, left axis deviation; LBBB, left bundle branch block; LVH, left ventricular hypertrophy; NSR, normal sinus rhythm; QT, QT interval; RAD, right axis deviation; RBBB, right bundle branch block; RVH, right ventricular hypertrophy; SA, sinus arrhythmia; SB, sinus bradycardia; TWI, T-wave inversion; xAI, explainable artificial intelligence. O_FIG O_LINKSMALLFIG WIDTH=200 HEIGHT=123 SRC="FIGDIR/small/26346628v1_ufig1.gif" ALT="Figure 1"> View larger version (60K): org.highwire.dtl.DTLVardef@124f232org.highwire.dtl.DTLVardef@98ba6forg.highwire.dtl.DTLVardef@f7fdedorg.highwire.dtl.DTLVardef@14012b3_HPS_FORMAT_FIGEXP M_FIG C_FIG
Matching journals
The top 6 journals account for 50% of the predicted probability mass.