基于人工智能的心率失常分类算法性能评价方法

    Performance Evaluation Method for Arrhythmia Classification Algorithms Based on Artificial Intelligence

    • 摘要: 心率失常的心拍分类对心血管疾病预测至关重要。目前有很多基于人工智能的心拍分类算法的研究,但是尚未建立算法指标评估体系来评价算法性能。基于心电信号的心拍分类,建立了一种心拍分类算法评价方法,用于评估算法模型的性能。在MIT-BIH标准数据库上,采用传统机器学习、神经网络、集成学习模型等11种心拍算法进行训练验证。基于实验研究建立了一套评价方法,包含12项指标,用于评估算法性能。结果表明,集成学习的模型整体表现较好(准确度大于98%),其中Light GBM表现最好,准确度、AUC、精确度和特异性都在98.8%之上,且拟合和泛化能力较好,训练时间相对较短(6.3s);此外,在同一模型上,AUC、精确度、敏感性、F1分数与准确度表现出很强的线性关系。建立的评价方法能够客观、全面地评估心率失常心拍分类算法,具有优化诊断效果和推动临床应用的意义。

       

      Abstract: Arrhythmia heartbeat classification is crucial for predicting cardiovascular diseases. Currently, artificial intelligence-based heartbeat classification algorithms are relatively well developed. However, a comprehensive evaluation method to assess the performance of these algorithms has yet to be established. This study introduces a framework for evaluating heartbeat classification algorithms, aimed at assessing model performance. Using the MIT-BIH standard database, we trained and validated eleven different heartbeat algorithms, including traditional machine learning, neural networks, and ensemble learning models. Based on a literature review, we developed an evaluation method comprising 12 direct metrics to assess algorithm performance. The results show that ensemble learning models performed best overall, achieving an accuracy greater than 98%, with LightGBM standing out as the top performer. Its accuracy, AUC, precision, and specificity all exceeded 98.8%, with strong fitting and generalization capabilities and a relatively short training time (6.3 seconds). Furthermore, metrics such as AUC, Precision, Sensitivity, F1 Score, and Accuracy demonstrated strong linear relationships. The evaluation framework in this study objectively assesses arrhythmic heartbeat classification algorithms, enhancing diagnostic accuracy and advancing clinical applications.

       

    /

    返回文章
    返回