Accurate overall, uneven by patient: a benchmark and demographic audit of deep learning for 12 lead ECG classification on PTB-XL
Rehman, A. D.; Nazir, S.
Show abstract
Deep learning reads 12 lead electrocardiograms at close to expert level on public benchmarks, yet most reports give one accuracy figure for the whole test set and stop there. We trained three architectures that are standard in this field, a 1D ResNet, a convolutional network with a bidirectional LSTM, and a convolutional network with a bidirectional LSTM followed by a transformer encoder, on the PTB-XL dataset to classify the five diagnostic superclasses, and then looked at how each one performed across sex and age. On the held out fold all three reached a macro AUC near 0.92, in line with the strongest published results on this benchmark, and the simplest model, the 1D ResNet, was marginally the best at 0.9241. The averages hid a steady pattern. Every model scored lower for female patients than for male patients, and every model scored lowest for patients aged 80 and over, where the 1D ResNet fell to 0.8878 and the transformer to 0.8693. Adding complexity did not close either gap and slightly widened the gap by age. Overall accuracy on PTB-XL is close to solved for these model families, but the benefit is not shared evenly, and a single headline number hides the patients a model serves worst. We release the full stratified evaluation to support fairness aware reporting.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.