Generalization challenges in electrocardiogram deep learning: insights from dataset characteristics and attention mechanism.
Journal:
Future cardiology
Published Date:
Jun 5, 2024
Abstract
Deep learning's widespread use prompts heightened scrutiny, particularly in the biomedical fields, with a specific focus on model generalizability. This study delves into the influence of training data characteristics on the generalization performance of models, specifically in cardiac abnormality detection. Leveraging diverse electrocardiogram datasets, models are trained on subsets with varying characteristics and subsequently compared for performance. Additionally, the introduction of the attention mechanism aims to improve generalizability. Experiments reveal that using a balanced dataset, just 1% of a large dataset, leads to equal performance in generalization tasks, notably in detecting cardiology abnormalities. This balanced training data notably enhances model generalizability, while the integration of the attention mechanism further refines the model's ability to generalize effectively.