计算机应用与软件2025，Vol.42Issue(5)：247-254,8.DOI:10.3969/j.issn.1000-386x.2025.05.034
基于GoogLeNet-ViT模型的心律不齐多标签诊断算法
MULTI-LABEL DIAGNOSIS ALGORITHM OF ARRHYTHMIA VIA GOOGLENET-ViT MODEL
摘要
Abstract
Electrocardiogram(ECG)has been proved to be the most common and effective approach to investigate arrhythmia.The automatic diagnosis algorithm of arrhythmia can be seen as a multi-label classification problem.The vision transformer(ViT)model has a good performance on classification problems.However,when it is directly applied to ECG classification,it will destroy the shape features inside the ECG signal,resulting in lower model accuracy.To this end,a multi-label classification algorithm for arrhythmia based on the GoogLeNet-VIT model is proposed.The algorithm used the pre-trained GoogLeNet to extract features instead of directly segmenting the ECG signal,and only used a single Transformer Encoder to complete the construction of the global relationship of features,and finally inputted the fully connected layer to complete multi-label classification.20 409 cases of clinical ECG data were selected for testing.The results show that the average F1 value of the algorithm reached 0.862 3,the average accuracy rate is 97.68％,and the proportion of diagnostic labels that are completely correct is 83.14％.Compared with the ViT model and the conventional CNN network,the proposed algorithm has clear advantages.关键词
心电信号/心律失常/深度学习/Vision TransformerKey words
Electrocardiogram/Arrhythmia/Deep learning/Vision transformer分类
信息技术与安全科学引用本文复制引用
黄浩,朱俊江..基于GoogLeNet-ViT模型的心律不齐多标签诊断算法[J].计算机应用与软件,2025,42(5):247-254,8.基金项目
国家自然科学基金项目(61801454). （61801454）
