arrow
Return

Knowledge-shot learning: An interpretable deep model for classifying imbalanced electrocardiography data

delete2020-12-01
delete14
PRE
AI
Y
Yen-Hsiu Chou
S
Shenda Hong
周雨熙 cover
周雨熙 (Yuxi Zhou)
J
Junyuan Shang
M
Moxian Song
H
Hongyan Li *
DOI:10.1016/j.neucom.2020.07.045delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
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.
Keywords:
Heartbeats classification
Imbalance problem
Neural network
Interpretable Model
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Neurocomputing cover
Neurocomputing
IF:
6.5
Papers:
2.5W
Citations:
6.5W

Organization

U
university system of georgia
Scholars:
7.3W
Papers: 6.5W
Citations: 101
P
peking university
Scholars:
11.8W
Papers: 8.7W
Citations: 146