An interpretable machine learning method for heart beat classification / 中国胸心血管外科临床杂志

ABSTRACT

@#Objective    To explore the application of Tsetlin Machine (TM) in heart beat classification. Methods    TM was used to classify the normal beats, premature ventricular contraction (PVC) and supraventricular premature beats (SPB) in the 2020 data set of China Physiological Signal Challenge. This data set consisted of the single-lead electro-cardiogram data of 10 patients with arrhythmia. One patient with atrial fibrillation was excluded, and finally data of the other 9 patients were included in this study. The classification results were then analyzed. Results    The classification results showed that the average recognition accuracy of TM was 84.3%, and the basis of classification could be shown by the bit pattern interpretation diagram. Conclusion    TM can explain the classification results when classifying heart beats. The reasonable interpretation of classification results can increase the reliability of the model and facilitate people's review and understanding.