Detection of multiple cardiac abnormalities using Convolution, Positional Encoder and Transformer on 12-lead ECG recordings
Ford, A.; Lan, J.; Ng, K.
Show abstract
ObjectivesFirstly, we aimed to develop a system capable of detecting multiple cardiac abnormalities simultaneously from 12-lead ECG recordings. Secondly, we tried to improve the detection by analyzing the relationship between imbalanced datasets and optimal classification thresholds. MethodsA novel fusion of Convolution Positional Encoder and Transformer Encoder was used to solve the multi-label classification problem. We used a proper evaluation metric called area under the precision-recall curve (AUPRC) that enabled us to analyze the precision-recall trade-off and find the optimal thresholds. ResultsHaving outperformed other popular deep networks, the model achieved the highest AUPRC of 0.96 and f1-score of 0.90 on 42511-sample datasets. We also found the negative correlation coefficient of -0.68 be-tween optimal thresholds and the proportion of positive samples. Significance &ConclusionThis study compared the performances of different deep learning architectures on a medical problem and showed the potential of advanced techniques in capturing spatial, temporal features alongside attention mechanisms. It also introduced how to reduce the impact of imbalanced datasets and find optimal classification thresholds.
Matching journals
The top 6 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Epileptic Spike Detection Using Neural Networks with Linear-Phase Convolutions 95%
- Evaluating Explanations from AI Algorithms for Clinical Decision-Making: A Social Science-based Approach 94%
- BertNDA: a Model Based on Graph-Bert and Multi-scale Information Fusion for ncRNA-disease Association Prediction 93%
Similar papers in this journal
- Multiple Instance Learning Framework can Facilitate Explainability in Murmur Detection 96%
- Uncovering the effects of model initialization on deep model generalization: A study with adult and pediatric chest X-ray images 96%
- A recurrent neural network and parallel hidden Markov model algorithm to segment and detect heart murmurs in phonocardiograms 95%
Similar papers in this journal
- SingleChannelNet: A Model for Automatic Sleep Stage Classification with Raw Single-Channel EEG 95%
- Hilbert-Envelope Features for Cardiac Disease Classification from Noisy Phonocardiograms 95%
- Graph connection Laplacian allows for enhanced outcomes of consumer camera based photoplethysmography imaging 93%
Similar papers in this journal
"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.