arXiv:2412.02801cs.AI2024-12被引 29

用粒子群优化改进Transformer,心病预测准确率达96.5%。

Optimization of Transformer heart disease prediction model based on particle swarm optimization algorithm

  • 用粒子群算法优化Transformer模型参数
  • 心病预测准确率提升至96.5%,比随机森林高4.3个百分点
  • 适合医疗AI、模型优化方向的研究者参考

针对最新粒子群优化算法,本文提出一种改进的Transformer模型,以提升心脏病预测准确率并提供新算法思路。首先采用决策树、随机森林和XGBoost三种主流机器学习分类算法进行对比,结果显示随机森林在心脏病分类预测中表现最佳,准确率为92.2%。随后,将基于粒子群优化(PSO)的Transformer模型应用于相同数据集进行分类实验,结果表明该模型分类准确率达到96.5%,较随机森林高出4.3个百分点,验证了PSO在优化Transformer模型中的有效性。研究表明,粒子群优化显著提升了Transformer在心脏病预测中的性能。提高心脏病预测能力是全球优先事项,有助于提升公共卫生水平、优化医疗资源配置、降低医疗成本,推动更高效健康管理和更健康、更具韧性的全球社会建设。

原文摘要 · Abstract (English)

Aiming at the latest particle swarm optimization algorithm, this paper proposes an improved Transformer model to improve the accuracy of heart disease prediction and provide a new algorithm idea. We first use three mainstream machine learning classification algorithms - decision tree, random forest and XGBoost, and then output the confusion matrix of these three models. The results showed that the random forest model had the best performance in predicting the classification of heart disease, with an accuracy of 92.2%. Then, we apply the Transformer model based on particle swarm optimization (PSO) algorithm to the same dataset for classification experiment. The results show that the classification accuracy of the model is as high as 96.5%, 4.3 percentage points higher than that of random forest, which verifies the effectiveness of PSO in optimizing Transformer model. From the above research, we can see that particle swarm optimization significantly improves Transformer performance in heart disease prediction. Improving the ability to predict heart disease is a global priority with benefits for all humankind. Accurate prediction can enhance public health, optimize medical resources, and reduce healthcare costs, leading to healthier populations and more productive societies worldwide. This advancement paves the way for more efficient health management and supports the foundation of a healthier, more resilient global community.

心脏病预测Transformer粒子群优化医疗AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。