Title
A Weighted Graph Attention Network Based Method For Multi-Label Classification Of Electrocardiogram Abnormalities
Abstract
The multi-label electrocardiogram (ECG) classification is to automatically predict a set of concurrent cardiac abnormalities in an ECG record, which is significant for clinical diagnosis. Modeling the cardiac abnormality dependencies is the key to improving classification performance. To capture the dependencies, we proposed a multi-label classification method based on the weighted graph attention networks. In the study, a graph taking each class as a node was mapped and the class dependencies were represented by the weights of graph edges. A novel weights generation method was proposed by combining the self-attentional weights and the prior learned co-occurrence knowledge of classes. The algorithm was evaluated on the dataset of the Hefei Hi-tech Cup ECG Intelligent Competition for 34 kinds of ECG abnormalities classification. And the micro-f1 and the macro-f1 of cross validation respectively were 91.45% and 44.48%. The experiment results show that the proposed method can model class dependencies and improve classification performance.
Year
DOI
Venue
2020
10.1109/EMBC44109.2020.9175981
42ND ANNUAL INTERNATIONAL CONFERENCES OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY: ENABLING INNOVATIVE TECHNOLOGIES FOR GLOBAL HEALTHCARE EMBC'20
DocType
Volume
ISSN
Conference
2020
1557-170X
Citations 
PageRank 
References 
0
0.34
0
Authors
8
Name
Order
Citations
PageRank
Hongmei Wang13113.44
wei zhao27128.69
Zhenqi Li343.11
Dongya Jia444.80
Cong Yan532.09
jing hu62213.68
Jiansheng Fang764.82
Ming Yang8114.10