Multi-Label Diagnosis of Arrhythmias Based on a Modified Two-Category Cross-Entropy Loss Function
The 12-lead resting electrocardiogram (ECG) is commonly used in hospitals to assess heart health. The ECG can reflect a variety of cardiac abnormalities, requiring multi-label classification. However, the diagnosis results in previous studies have been imprecise. For example, in some previous studie...
| Main Authors: | , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2023-12-01
|
| Series: | Electronics |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2079-9292/12/24/4976 |
| _version_ | 1797381334037954560 |
|---|---|
| author | Junjiang Zhu Cheng Ma Yihui Zhang Hao Huang Dongdong Kong Wangjin Ni |
| author_facet | Junjiang Zhu Cheng Ma Yihui Zhang Hao Huang Dongdong Kong Wangjin Ni |
| author_sort | Junjiang Zhu |
| collection | DOAJ |
| description | The 12-lead resting electrocardiogram (ECG) is commonly used in hospitals to assess heart health. The ECG can reflect a variety of cardiac abnormalities, requiring multi-label classification. However, the diagnosis results in previous studies have been imprecise. For example, in some previous studies, some cardiac abnormalities that cannot coexist often appeared in the diagnostic results. In this work, we explore how to realize the effective multi-label diagnosis of ECG signals and prevent the prediction of cardiac arrhythmias that cannot coexist. In this work, a multi-label classification method based on a convolutional neural network (CNN), long short-term memory (LSTM), and an attention mechanism is presented for the multi-label diagnosis of cardiac arrhythmia using resting ECGs. In addition, this work proposes a modified two-category cross-entropy loss function by introducing a regularization term to avoid the existence of arrhythmias that cannot coexist. The effectiveness of the modified cross-entropy loss function is validated using a 12-lead resting ECG database collected by our team. Using traditional and modified cross-entropy loss functions, three deep learning methods are employed to classify six types of ECG signals. Experimental results show the modified cross-entropy loss function greatly reduces the number of non-coexisting label pairs while maintaining prediction accuracy. Deep learning methods are effective in the multi-label diagnosis of ECG signals, and diagnostic efficiency can be improved by using the modified cross-entropy loss function. In addition, the modified cross-entropy loss function helps prevent diagnostic models from outputting two arrhythmias that cannot coexist, further reducing the false positive rate of non-coexisting arrhythmic diseases, thereby demonstrating the potential value of the modified loss function in clinical applications. |
| first_indexed | 2024-03-08T20:49:56Z |
| format | Article |
| id | doaj.art-8e5d479cb1814e69a169bb56cb4d775d |
| institution | Directory Open Access Journal |
| issn | 2079-9292 |
| language | English |
| last_indexed | 2024-03-08T20:49:56Z |
| publishDate | 2023-12-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Electronics |
| spelling | doaj.art-8e5d479cb1814e69a169bb56cb4d775d2023-12-22T14:05:07ZengMDPI AGElectronics2079-92922023-12-011224497610.3390/electronics12244976Multi-Label Diagnosis of Arrhythmias Based on a Modified Two-Category Cross-Entropy Loss FunctionJunjiang Zhu0Cheng Ma1Yihui Zhang2Hao Huang3Dongdong Kong4Wangjin Ni5College of Mechanical and Electrical Engineering, China Jiliang University, Hangzhou 310018, ChinaCollege of Mechanical and Electrical Engineering, China Jiliang University, Hangzhou 310018, ChinaCollege of Mechanical and Electrical Engineering, China Jiliang University, Hangzhou 310018, ChinaCollege of Mechanical and Electrical Engineering, China Jiliang University, Hangzhou 310018, ChinaSchool of Mechatronic Engineering and Automation, Shanghai University, Shanghai 200444, ChinaCollege of Mechanical and Electrical Engineering, China Jiliang University, Hangzhou 310018, ChinaThe 12-lead resting electrocardiogram (ECG) is commonly used in hospitals to assess heart health. The ECG can reflect a variety of cardiac abnormalities, requiring multi-label classification. However, the diagnosis results in previous studies have been imprecise. For example, in some previous studies, some cardiac abnormalities that cannot coexist often appeared in the diagnostic results. In this work, we explore how to realize the effective multi-label diagnosis of ECG signals and prevent the prediction of cardiac arrhythmias that cannot coexist. In this work, a multi-label classification method based on a convolutional neural network (CNN), long short-term memory (LSTM), and an attention mechanism is presented for the multi-label diagnosis of cardiac arrhythmia using resting ECGs. In addition, this work proposes a modified two-category cross-entropy loss function by introducing a regularization term to avoid the existence of arrhythmias that cannot coexist. The effectiveness of the modified cross-entropy loss function is validated using a 12-lead resting ECG database collected by our team. Using traditional and modified cross-entropy loss functions, three deep learning methods are employed to classify six types of ECG signals. Experimental results show the modified cross-entropy loss function greatly reduces the number of non-coexisting label pairs while maintaining prediction accuracy. Deep learning methods are effective in the multi-label diagnosis of ECG signals, and diagnostic efficiency can be improved by using the modified cross-entropy loss function. In addition, the modified cross-entropy loss function helps prevent diagnostic models from outputting two arrhythmias that cannot coexist, further reducing the false positive rate of non-coexisting arrhythmic diseases, thereby demonstrating the potential value of the modified loss function in clinical applications.https://www.mdpi.com/2079-9292/12/24/4976multi-label diagnosisdeep learningcross-entropy loss function |
| spellingShingle | Junjiang Zhu Cheng Ma Yihui Zhang Hao Huang Dongdong Kong Wangjin Ni Multi-Label Diagnosis of Arrhythmias Based on a Modified Two-Category Cross-Entropy Loss Function Electronics multi-label diagnosis deep learning cross-entropy loss function |
| title | Multi-Label Diagnosis of Arrhythmias Based on a Modified Two-Category Cross-Entropy Loss Function |
| title_full | Multi-Label Diagnosis of Arrhythmias Based on a Modified Two-Category Cross-Entropy Loss Function |
| title_fullStr | Multi-Label Diagnosis of Arrhythmias Based on a Modified Two-Category Cross-Entropy Loss Function |
| title_full_unstemmed | Multi-Label Diagnosis of Arrhythmias Based on a Modified Two-Category Cross-Entropy Loss Function |
| title_short | Multi-Label Diagnosis of Arrhythmias Based on a Modified Two-Category Cross-Entropy Loss Function |
| title_sort | multi label diagnosis of arrhythmias based on a modified two category cross entropy loss function |
| topic | multi-label diagnosis deep learning cross-entropy loss function |
| url | https://www.mdpi.com/2079-9292/12/24/4976 |
| work_keys_str_mv | AT junjiangzhu multilabeldiagnosisofarrhythmiasbasedonamodifiedtwocategorycrossentropylossfunction AT chengma multilabeldiagnosisofarrhythmiasbasedonamodifiedtwocategorycrossentropylossfunction AT yihuizhang multilabeldiagnosisofarrhythmiasbasedonamodifiedtwocategorycrossentropylossfunction AT haohuang multilabeldiagnosisofarrhythmiasbasedonamodifiedtwocategorycrossentropylossfunction AT dongdongkong multilabeldiagnosisofarrhythmiasbasedonamodifiedtwocategorycrossentropylossfunction AT wangjinni multilabeldiagnosisofarrhythmiasbasedonamodifiedtwocategorycrossentropylossfunction |