Towards Interpretable Arrhythmia Classification With Human-Machine Collaborative Knowledge Representation
Jilong Wang, Rui Li, Renfa Li, Bin Fu, Chunxia Xiao, Danny Z. Chen
Abstract
Arrhythmia detection and classification is a crucial step for diagnosing cardiovascular diseases. However, deep learning models that are commonly used and trained in end-to-end fashion are not able to provide good interpretability. In this paper, we address this deficiency by proposing the first novel interpretable arrhythmia classification approach based on a human-machine collaborative knowledge representation. Our approach first employs an AutoEncoder to encode electrocardiogram signals into two parts: hand-encoding knowledge and machine-encoding knowledge. A classifier then takes as input the encoded knowledge to classify arrhythmia heartbeats with or without human in the loop (HIL). Experiments and evaluation on the MIT-BIH Arrhythmia Database demonstrate that our new approach not only can effectively classify arrhythmia while offering interpretability, but also can improve the classification accuracy by adjusting the hand-encoding knowledge with our HIL mechanism.