CoAt-Mixer: Self-attention deep learning framework for left ventricular hypertrophy using electrocardiography

Abstract

Left ventricular hypertrophy is a significant independent risk factor for all-cause mortality and morbidity, and an accurate diagnosis at an early stage of heart change is clinically significant. Electrocardiography is the most convenient, economical, and non-invasive method for screening in primary care. However, the coincidence rate of the actual left ventricular hypertrophy and diagnostic findings was low, consequently increasing the interest in algorithms using big data and deep learning. We attempted to diagnose left ventricular hypertrophy using big data and deep learning algorithms, and aimed to confirm its diagnostic power according to the differences between males and females. This retrospective study used electrocardiographs obtained at Yonsei University Wonju Severance Christian Hospital, Wonju, Korea, from October 2010 to February 2020. Binary classification was performed for primary screening for left ventricular hypertrophy. Three datasets were used for the experiment: the male, female, and entire dataset. A cutoff for binary classification was defined as the meaningful as a screening test (<132 g/m2 vs. ≥132 g/m2, <109 g/m2 vs. ≥109 g/m2). Six types of input were used for the classification tasks. We attempted to determine whether electrocardiography had predictive power for left ventricular hypertrophy diagnosis. For the entire dataset, the model achieved an area under the receiver operating characteristic (AUROC) curve of 0.836 (95% CI, 0.833-838) with a sensitivity of 78.37% (95% CI, 76.79-79.95). For the male dataset, the AUROC was 0.826 (95% CI, 0.822-830) with a sensitivity of 76.73% (95% CI, 75.14-78.33). For the female dataset, the AUROC was 0.772 (95% CI, 0.769-775) with a sensitivity of 72.90% (95% CI, 70.33-75.46). Our model confirmed that left ventricular hypertrophy can be classified to some extent using electrocardiography, demographics, and electrocardiography features. In particular, a learning environment that considered gender differences was constructed. Consequently, the difference in diagnostic power between men and women was confirmed. Our model will help patients with suspected left ventricular hypertrophy to undergo screening tests at a low cost. In addition, our research and attempts will show the expected effect that gender-consideration approaches can help with various currently proposed diagnostic methods.