A Branched Deep Neural Network for End-to-end Classification from ECGs with Varying Dimensions
Han Duan,Junchao Fan,Bin Xiao,Xiuli Bi,Junhui Zhang,Xu Ma
DOI: https://doi.org/10.22489/CinC.2021.013
2021-01-01
Abstract:Objective: Computerized electrocardiogram (ECG) interpretation plays a critical role in the clinical ECG workflow. For 12-lead ECGs, deep neural networks (DNNs) could match state-of-the-art algorithms when trained in openly available datasets (e.g., 2020 PhysioNet/CinC Challenge data). However, there is limited evidence to demonstrate the utility of reduced-lead ECGs for capturing a wide range of diagnostic information. Thus, in this work, we aimed to develop a deep learning-based algorithm to identify clinical diagnoses from twelve-lead, sixlead, four-lead, three-lead, and two-lead ECG recordings. Methods: We have employed a branched DNN that utilizes two CNNs (similar to the residual network, but adapted to signals with varying dimensions) with different filter sizes. This architecture can effectively capture abnormal patterns of diseases and suppress noise interference by introducing double branch and skip connections. Moreover, the Squeeze-and-Excitation block is added to every branch of the architecture, which allows the DNNs to learn more effective features from input signals. Results: Our code failed during the official phase and was not able to process the training data to complete the training process and therefore was not scored. Our (CQUPT Dontwant) method is evaluated using the twelvelead, six-lead, four-lead, three-lead, and two-lead of the datasets, and we received a 5-fold cross-validation Challenge metric score of 0.639, 0.610, 0.638, 0.619, and 0.625, respectively. However, the cross validated results on the training data are overly-optimistic, given the results reported by other teams on the test data.