Title | Knowledge-shot learning: An interpretable deep model for classifying imbalanced electrocardiography data |
Authors | Chou, Yen-hsiu Hong, Shenda Zhou, Yuxi Shang, Junyuan Song, Moxian Li, Hongyan |
Affiliation | Peking Univ, Sch Elect Engn & Comp Sci, Beijing, Peoples R China Peking Univ, Minist Educ, Key Lab Machine Percept, Beijing, Peoples R China Georgia Inst Technol, Dept Computat Sci & Engn, Atlanta, GA 30332 USA |
Keywords | ECG CLASSIFICATION ATRIAL-FIBRILLATION NEURAL-NETWORK |
Issue Date | 5-Dec-2020 |
Publisher | NEUROCOMPUTING |
Abstract | Classification of Electrocardiogram (ECG) data has been an important research topic in the machine learning area for many years. Recently, deep learning methods have been used in classifying ECG data and have achieved superior results than traditional methods. However, in real-world applications, two challenges existing deep learning methods cannot handle well - imbalanced data and model interpretability. In this paper, we propose an ECG classification method named Knowledge-Shot Learning (KSL) that can handle with the above two challenges. KSL constructs a novel neural network architecture that can be effectively trained on imbalanced ECG data. Besides, KSL can also extract interpretable feature vectors and give support cases as result explanations. Moreover, KSL can even classify unseen diseases if provided with the necessary medical knowledge. Experiments on real-world ECG data show that KSL improves 10.00% of F-1-score on imbalanced classes, and 43.75% of F-1-score on unseen classes, compared with the second-best baseline. KSL also provides interpretable results that are consistent with medical domain knowledge. (C) 2020 Elsevier B.V. All rights reserved. |
URI | http://hdl.handle.net/20.500.11897/600801 |
ISSN | 0925-2312 |
DOI | 10.1016/j.neucom.2020.07.045 |
Indexed | SCI(E) |
Appears in Collections: | 信息科学技术学院 机器感知与智能教育部重点实验室 |