Computer Science ›› 2024, Vol. 51 ›› Issue (12): 174-180.doi: 10.11896/jsjkx.230800083

• Database & Big Data & Data Science • Previous Articles     Next Articles

Multitask Classification Algorithm of ECG Signals Based on Radient Magnitude Direction Adjustment

ZHANG Xue1, TIAN Lan1, ZENG Ming1, LIU Junhui 1, ZONG Shaoguo2   

  1. 1 School of Microelectronics, Shandong University, Jinan 250101, China
    2 Shandong Houde Measurement and Control Technology Co., LTD, Jinan 250000, China
  • Received:2023-08-15 Revised:2024-03-01 Online:2024-12-15 Published:2024-12-10
  • About author:ZHANG Xue,born in 2000,postgra-duate,is a member of CCF(No.T3347G).Her main research interests include biological signal processing and intelligent healthcare.
    TIAN Lan,born in 1965,professor,Ph.D supervisor.Her main research interests include brain-computer interface,brain-like computing and deep learning,biological signal processing and light regulation of neural activity.
  • Supported by:
    Natural Science Foundation of Shandong Province,China(ZR2021MF065,ZR2021ZD40).

Abstract: Cardiovascular diseases are posing more and more serious threats to human health and safety.ECG signals can be used to diagnose and classify related diseases.Most existing ECG classification algorithms adopt single-task learning model,which can not make comprehensive use of complementary features in multiple tasks.However,multi-task learning model can learn multiple related tasks at the same time,share related task features,and help improve the classification performance of multiple tasks.Combining deep learning and multi-task learning,a multi-task classification algorithm for ECG signals based on loss optimization is proposed.The multi-classification task of ECG signals is decomposed into multiple binary classification tasks,and loss optimization is carried out from the aspects of the amplitude and direction of task gradient,so as to avoid the negative transfer caused by manual setting of task loss weights and the cancellation of task losses.The performance of ECG signal multi-classification task is improved.The model uses PTB-XL database to decompose 23 classification tasks into 23 binary classification tasks to evaluate the proposed algorithm.Experimental results show that the average area under the macro curve(AUC) reaches 0.950,the accuracy reaches 96.50%,the tag-based F1 score reaches 0.583,and the sample-based F1 score reaches 0.777.Compared with the single-task learning algorithm,the proposedalgorithm shows good performance in the multi-classification of ECG signals.

Key words: ECG signal classification, Multi-task learning, Loss optimization

CLC Number: 

  • TP391.7
[1]BENJAMIN E J,VIRANI S S,CALLAWAY C W,et al.Heart Disease and Stroke Statistics-2018 Update:A Report From the American Heart Association[J].Circulation,2018,137(12):E67-E492.
[2]LUZ E J D,SCHWARTZ W R,CAMARA-CHAVEZ G,et al.ECG-based heartbeat classification for arrhythmia detection:A survey[J].Computer Methods and Programs in Biomedicine,2016,127:144-164.
[3]BERKAYA S K,UYSAL A K,GUNAL E S,et al.A survey on ECG analysis[J].Biomedical Signal Processing and Control,2018,43:216-235.
[4]ZHAO R X,XU P J,LIU Y.ECG-based Atrial Fibrillation Detection Based on Deep Convolutional Residual Neural Network[J].Computer Science,2022,49(5):186-193.
[5]HANNUN A Y,RAJPURKAR P,HAGPANAHI M,et al.Car-diologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network[J].Nature Medicine,2019,25(1):65-69.
[6]AHMED A A,ALI W,ABDULLAH T A A,et al.Classifying Cardiac Arrhythmia from ECG Signal Using 1D CNN Deep Learning Model[J].Mathematics,2023,11,562:1-16.
[7]ACHARYA U R,OH S L,HAGIWARA Y,et al.A deep con-volutional neural network model to classify heartbeats[J].Computers in Biology and Medicine,2017,89:389-396.
[8]YILDIRIM O,PLAWIA P,TAN R S,et al.Arrhythmia detection using deep convolutional neural network with long duration ECG signals[J].Computers in Biology and Medicine,2018,102:411-420.
[9]ULLAH H,HEYAT M,AKHTAR F,et al.An Automatic Premature Ventricular Contraction Recognition System Based on Imbalanced Dataset and Pre-Trained Residual Network Using Transfer Learning on ECG Signal[J].Diagnostics,2023,13(1):1-20.
[10]CARUANA R.Multitask learning[J].Machine Learning,1997,28(1):41-75.
[11]ZHANG Y,YANG Q.An overview of multi-task learning[J].National Science Review,2018,5(1):30-43.
[12]SHAHI M,OO E,AHMED B,et al.Adversarial Multi-TaskLearning for Robust End-to-End ECG-based Heartbeat Classification[C]//Proceedings of the 42nd Annual International Conference of the IEEE Engineering in Medicine and Biology Society.Montreal,Canada,2020:341-344.
[13]ZHANG Y,YANG Q.A Survey on Multi-Task Learning[J].IEEE Transactions on Knowledge and Data Engineering,2022,34(12):1-20.
[14]HUANG Z C,LI W,XIA X G,et al.A Novel Nonlocal-Aware Pyramid and Multiscale Multitask Refinement Detector for Object Detection in Remote Sensing Images[J].IEEE Transactions on Geoscience and Remote Sensing,2022,60:1-20.
[15]LU G Q,ZHAO X S,YIN J,et al.Multi-task learning using va-riational auto-encoder for sentiment classification[J].Pattern Recognition Letters,2020,132:115-122.
[16]LU G Q,GAN J Z,YIN J,et al.Multi-task learning using a hybrid representation for text classification[J].Neural Computing and Applications,2020,32(11):6467-6480.
[17]BANIAT L H,PARK S,PARK S B.A Neural Machine Translation Model for Arabic Dialects That Utilizes Multitask Lear-ning(MTL)[J].Computational Intelligence and Neuroscience,2018,2018:1-11.
[18]JI J L,CHEN X H,LUO C Q,et al.A Deep Multi-task Learning Approach for ECG Data Analysis[C]//Proceedings of the 2018 IEEE EMBS International Conference on Biomedical and Health Informatics.Las Vegas,USA,2018:124-127.
[19]LIU Y,LI Q C,WANG K Q,et al.Automatic Multi-Label ECG Classification with Category Imbalance and Cost-Sensitive Thresholding[J].Biosensors-Basel,2021,11(11):1-17.
[20]LIU J H,ZENG M,SHAN K,et al.Loss Optimization Based Algorithm for Multi-classification of ECG Signal[C]//ISAIMS 2022:2022 3rd International Symposium on Artificial Intelligence for Medicine Sciences.2022:537-541.
[21]UBEYLI E D.ECG beats classification using multiclass support vector machines with error correcting output codes[J].Digital Signal Processing,2007,17(3):675-684.
[22]OBERMEYE Z,EMANUEL E J.Predicting the Future-Big Data,Machine Learning,and Clinical Medicine[J].New England Journal of Medicine,2016,375(13):1216-1219.
[23]GUO M,HAQUE A,HUANG D A,et al.Dynamic Task Prioritization for Multitask Learning[C]//Proceedings of the 15th European Conference on Computer Vision.2018:282-299.
[24]YU T H,KUMAR S,GUPT A,et al.Gradient surgery formulti-task learning[C]//Proceedings of the 34th Conference on Neural Information Processing Systems.2020:1-27.
[25]WAGNER P,STRODTHOFF N,BOUSSELJO R-D,et al.PTB-XL,a large publicly available electrocardiography dataset[J].Scientific Data,2020,7(1):1-15.
[26]STRODTHOFF N,WAGNER P,et al.Deep Learning for ECG Analysis:Benchmarks and Insights from PTB-XL[J].IEEE Journal Biomedical and Health Informatics,2021,25(5):1519-1528.
[27]LOSHCHILO I,HUTTER F.Fixing Weight Decay Regularization in ADAM[C]//Proceedings of the 2018 International Conference on Learning Representations.2018:1-10.
[28]CHEN Z,BADRINARAYANA V,LEE C Y,et al.GradNorm:Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks[C]//Proceedings of the 35th International Conference on Machine Learning.Stockholm,Sweden,2018:1-27.
[29]LIU S K,JOHNS E,DAVISON A J.End-To-End Multi-Task Learning with Attention[C]//Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition.Long Beach,USA,2019:1871-1880.
[30]CENG Q C,LIU H,GAO T L,et al.An ECG Classification Method Based on Multi-Task Learning and CoT Attention Mechanism[J].Healthcare,2023,11(7):1-13.
[1] ZHANG Haoyan, DUAN Liguo, WANG Qinchen, GAO Hao. Long Text Multi-entity Sentiment Analysis Based on Multi-task Joint Training [J]. Computer Science, 2024, 51(6): 309-316.
[2] LIU Zeyu, LIU Jianwei. Video and Image Salient Object Detection Based on Multi-task Learning [J]. Computer Science, 2024, 51(4): 217-228.
[3] FU Mingrui, LI Weijiang. Multi-task Emotion-Cause Pair Extraction Method Based on Position-aware Interaction Network [J]. Computer Science, 2024, 51(11A): 231000086-9.
[4] WANG Kunyang, LIU Yang, YE Ning, ZHANG Kai. Road Extraction from Complex Urban Remote Sensing Images Based on Multi-task Learning [J]. Computer Science, 2024, 51(11A): 240300095-8.
[5] XU Bei, XU Peng. Emotion Elicited Question Generation Model in Dialogue Scenarios [J]. Computer Science, 2024, 51(11): 265-272.
[6] ZHANG Xiaoyun, ZHAO Hui. Study on Multi-task Student Emotion Recognition Methods Based on Facial Action Units [J]. Computer Science, 2024, 51(10): 105-111.
[7] LUO Huilan, YE Ju. Study of Multi-task Learning with Joint Semantic Segmentation and Depth Estimation [J]. Computer Science, 2023, 50(6A): 220100111-10.
[8] ZHEN Tiange, SONG Mingyang, JING Liping. Incorporating Multi-granularity Extractive Features for Keyphrase Generation [J]. Computer Science, 2023, 50(4): 181-187.
[9] DU Li-jun, TANG Xi-lu, ZHOU Jiao, CHEN Yu-lan, CHENG Jian. Alzheimer's Disease Classification Method Based on Attention Mechanism and Multi-task Learning [J]. Computer Science, 2022, 49(6A): 60-65.
[10] ZHAO Kai, AN Wei-chao, ZHANG Xiao-yu, WANG Bin, ZHANG Shan, XIANG Jie. Intracerebral Hemorrhage Image Segmentation and Classification Based on Multi-taskLearning of Shared Shallow Parameters [J]. Computer Science, 2022, 49(4): 203-208.
[11] YANG Xiao-yu, YIN Kang-ning, HOU Shao-qi, DU Wen-yi, YIN Guang-qiang. Person Re-identification Based on Feature Location and Fusion [J]. Computer Science, 2022, 49(3): 170-178.
[12] ZHENG Shun-yuan, HU Liang-xiao, LYU Xiao-qian, SUN Xin, ZHANG Sheng-ping. Edge Guided Self-correction Skin Detection [J]. Computer Science, 2022, 49(11): 141-147.
[13] SONG Long-ze, WAN Huai-yu, GUO Sheng-nan, LIN You-fang. Multi-task Spatial-Temporal Graph Convolutional Network for Taxi Idle Time Prediction [J]. Computer Science, 2021, 48(7): 112-117.
[14] LIU Xiao-long, HAN Fang, WANG Zhi-jie. Joint Question Answering Model Based on Knowledge Representation [J]. Computer Science, 2021, 48(6): 241-245.
[15] ZHOU Xiao-jin, XU Chen-ming, RUAN Tong. Multi-granularity Medical Entity Recognition for Chinese Electronic Medical Records [J]. Computer Science, 2021, 48(4): 237-242.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!