Integrating EPSOSA-BP neural network algorithm for enhanced accuracy and robustness in optimizing coronary artery disease prediction

Sci Rep. 2024 Dec 28;14(1):30993. doi: 10.1038/s41598-024-82184-2.

Abstract

Coronary artery disease represents a formidable health threat to middle-aged and elderly populations worldwide. This research introduces an advanced BP neural network algorithm, EPSOSA-BP, which integrates particle swarm optimization, simulated annealing, and a particle elimination mechanism to elevate the precision of heart disease prediction models. To address prior limitations in feature selection, the study employs single-hot encoding and Principal Component Analysis, thereby enhancing the model's feature learning capability. The proposed method achieved remarkable accuracy rates of 93.22% and 95.20% on the UCI and Kaggle datasets, respectively, underscoring its exceptional performance even with small sample sizes. Ablation experiments further validated the efficacy of the data preprocessing and feature selection techniques employed. Notably, the EPSOSA algorithm surpassed classical optimization algorithms in terms of convergence speed, while also demonstrating improved sensitivity and specificity. This model holds significant potential for facilitating early identification of high-risk patients, which could ultimately save lives and optimize the utilization of medical resources. Despite implementation challenges, including technical integration and data standardization, the algorithm shows promise for use in emergency settings and community health services for regular cardiac risk monitoring.

Keywords: Cardiovascular disease (CVD); Elimination strategy; Feature engineering; Particle swarm optimization algorithm; Principal component analysis; Simulated annealing algorithm.

MeSH terms

  • Algorithms*
  • Coronary Artery Disease*
  • Humans
  • Neural Networks, Computer*
  • Principal Component Analysis