Domain and Patient Adversarial Multi-Task Learning for Arrhythmia Classification.

Abstract

Manual screening of electrocardiograms (ECGs) for heart arrhythmias by clinicians is time-consuming and labor-intensive. A machine learning model for the automated diagnosis of heart arrhythmia from ECG signals can facilitate improved diagnosis, greater accessibility and earlier intervention for patients. The potential of such models is limited however by the small size of clinical datasets available for training. Methods that can be trained with multiple datasets to classify heart arrhythmia are needed to overcome this problem.In this paper, we propose using adversarial multi-task learning (AMTL) to extract domain and patient invariant features from two electrocardiogram databases. We further investigated the influence of beat segmentation location and beat normalization on domain invariance. Our proposed methods were tested on the MIT-BIH Arrhythmia and the St Petersburg INCART 12-lead Arrhythmia Databases. The domain adversarial models achieved a higher accuracy and average F1 score than their counterparts without domain adversarial learning. In particular, the patient and domain adversarial model improved the F1 scores on the two tested databases from 70% and 74% to 77% each.Clinical Relevance-This establishes that adversarial multitask learning with multiple datasets and multiple adversarial tasks can improve the F1 score of arrhythmia classification.