A novel method for reducing arrhythmia classification from 12-lead ECG signals to single-lead ECG with minimal loss of accuracy through teacher-student knowledge distillation

Abstract

Deep learning models developed through multi-lead electrocardiogram (ECG) signals are considered the leading methods for the automated detection of arrhythmia on computer systems. However, due to the amplitudes of input signals, these models generate too many parameters for practical use. Therefore, they are rarely used on devices with limited computational resources in the newly-emerged technology of the Internet of medical things (IoMT). Knowledge distillation was utilized in this paper to propose a method for bridging the gap between the arrhythmia classification model with multi-lead ECG signals and the arrhythmia classification model with single-lead ECG signals by minimizing the performance decline. The proposed method consists of a teacher model with advanced architecture and a student model with simple architecture. The teacher model was already developed through multi-lead ECG signals, whereas the student model was developed through single-lead signals under the supervision of the teacher. Despite its simplicity, the student model receives the dark knowledge of multi-lead ECG signals from the teacher by imitating the teacher’s behavior in the development process. According to the results, the student model was nearly 262.18 times more compressed than its teacher. Moreover, the student experienced approximately 0.81% of accuracy decline in Chapman ECG with 10646 patients.