Application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation
Abstract Background Atrial fibrillation is a paroxysmal heart disease without any obvious symptoms for most people during the onset. The electrocardiogram (ECG) at the time other than the onset of this disease is not significantly different from that of normal people, which makes it difficult to det...
Guardado en:
Autores principales: | , , , , , , , , |
---|---|
Formato: | article |
Lenguaje: | EN |
Publicado: |
BMC
2021
|
Materias: | |
Acceso en línea: | https://doaj.org/article/dc335e28fd3c4e2a84533e1c01858bbd |
Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
id |
oai:doaj.org-article:dc335e28fd3c4e2a84533e1c01858bbd |
---|---|
record_format |
dspace |
spelling |
oai:doaj.org-article:dc335e28fd3c4e2a84533e1c01858bbd2021-11-14T12:13:05ZApplication of artificial intelligence ensemble learning model in early prediction of atrial fibrillation10.1186/s12859-021-04000-21471-2105https://doaj.org/article/dc335e28fd3c4e2a84533e1c01858bbd2021-11-01T00:00:00Zhttps://doi.org/10.1186/s12859-021-04000-2https://doaj.org/toc/1471-2105Abstract Background Atrial fibrillation is a paroxysmal heart disease without any obvious symptoms for most people during the onset. The electrocardiogram (ECG) at the time other than the onset of this disease is not significantly different from that of normal people, which makes it difficult to detect and diagnose. However, if atrial fibrillation is not detected and treated early, it tends to worsen the condition and increase the possibility of stroke. In this paper, P-wave morphology parameters and heart rate variability feature parameters were simultaneously extracted from the ECG. A total of 31 parameters were used as input variables to perform the modeling of artificial intelligence ensemble learning model. Results This paper applied three artificial intelligence ensemble learning methods, namely Bagging ensemble learning method, AdaBoost ensemble learning method, and Stacking ensemble learning method. The prediction results of these three artificial intelligence ensemble learning methods were compared. As a result of the comparison, the Stacking ensemble learning method combined with various models finally obtained the best prediction effect with the accuracy of 92%, sensitivity of 88%, specificity of 96%, positive predictive value of 95.7%, negative predictive value of 88.9%, F1 score of 0.9231 and area under receiver operating characteristic curve value of 0.911. Conclusion In feature extraction, this paper combined P-wave morphology parameters and heart rate variability parameters as input parameters for model training, and validated the value of the proposed parameters combination for the improvement of the model’s predicting effect. In the calculation of the P-wave morphology parameters, the hybrid Taguchi-genetic algorithm was used to obtain more accurate Gaussian function fitting parameters. The prediction model was trained using the Stacking ensemble learning method, so that the model accuracy had better results, which can further improve the early prediction of atrial fibrillation.Cai WuMaxwell HwangTian-Hsiang HuangYen-Ming J. ChenYiu-Jen ChangTsung-Han HoJian HuangKao-Shing HwangWen-Hsien HoBMCarticleAtrial fibrillationElectrocardiogramArtificial intelligenceEnsemble learningComputer applications to medicine. Medical informaticsR858-859.7Biology (General)QH301-705.5ENBMC Bioinformatics, Vol 22, Iss S5, Pp 1-12 (2021) |
institution |
DOAJ |
collection |
DOAJ |
language |
EN |
topic |
Atrial fibrillation Electrocardiogram Artificial intelligence Ensemble learning Computer applications to medicine. Medical informatics R858-859.7 Biology (General) QH301-705.5 |
spellingShingle |
Atrial fibrillation Electrocardiogram Artificial intelligence Ensemble learning Computer applications to medicine. Medical informatics R858-859.7 Biology (General) QH301-705.5 Cai Wu Maxwell Hwang Tian-Hsiang Huang Yen-Ming J. Chen Yiu-Jen Chang Tsung-Han Ho Jian Huang Kao-Shing Hwang Wen-Hsien Ho Application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation |
description |
Abstract Background Atrial fibrillation is a paroxysmal heart disease without any obvious symptoms for most people during the onset. The electrocardiogram (ECG) at the time other than the onset of this disease is not significantly different from that of normal people, which makes it difficult to detect and diagnose. However, if atrial fibrillation is not detected and treated early, it tends to worsen the condition and increase the possibility of stroke. In this paper, P-wave morphology parameters and heart rate variability feature parameters were simultaneously extracted from the ECG. A total of 31 parameters were used as input variables to perform the modeling of artificial intelligence ensemble learning model. Results This paper applied three artificial intelligence ensemble learning methods, namely Bagging ensemble learning method, AdaBoost ensemble learning method, and Stacking ensemble learning method. The prediction results of these three artificial intelligence ensemble learning methods were compared. As a result of the comparison, the Stacking ensemble learning method combined with various models finally obtained the best prediction effect with the accuracy of 92%, sensitivity of 88%, specificity of 96%, positive predictive value of 95.7%, negative predictive value of 88.9%, F1 score of 0.9231 and area under receiver operating characteristic curve value of 0.911. Conclusion In feature extraction, this paper combined P-wave morphology parameters and heart rate variability parameters as input parameters for model training, and validated the value of the proposed parameters combination for the improvement of the model’s predicting effect. In the calculation of the P-wave morphology parameters, the hybrid Taguchi-genetic algorithm was used to obtain more accurate Gaussian function fitting parameters. The prediction model was trained using the Stacking ensemble learning method, so that the model accuracy had better results, which can further improve the early prediction of atrial fibrillation. |
format |
article |
author |
Cai Wu Maxwell Hwang Tian-Hsiang Huang Yen-Ming J. Chen Yiu-Jen Chang Tsung-Han Ho Jian Huang Kao-Shing Hwang Wen-Hsien Ho |
author_facet |
Cai Wu Maxwell Hwang Tian-Hsiang Huang Yen-Ming J. Chen Yiu-Jen Chang Tsung-Han Ho Jian Huang Kao-Shing Hwang Wen-Hsien Ho |
author_sort |
Cai Wu |
title |
Application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation |
title_short |
Application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation |
title_full |
Application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation |
title_fullStr |
Application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation |
title_full_unstemmed |
Application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation |
title_sort |
application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation |
publisher |
BMC |
publishDate |
2021 |
url |
https://doaj.org/article/dc335e28fd3c4e2a84533e1c01858bbd |
work_keys_str_mv |
AT caiwu applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation AT maxwellhwang applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation AT tianhsianghuang applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation AT yenmingjchen applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation AT yiujenchang applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation AT tsunghanho applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation AT jianhuang applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation AT kaoshinghwang applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation AT wenhsienho applicationofartificialintelligenceensemblelearningmodelinearlypredictionofatrialfibrillation |
_version_ |
1718429376638353408 |